mirror of
https://github.com/matrix-org/dendrite.git
synced 2025-12-12 09:23:09 -06:00
add login tests and structure to test /profile
Signed-off-by: MohitKS5 <mohitkumarsingh907@gmail.com>
This commit is contained in:
MohitKS5
parent
653f20995d
commit
a8b1daafae
|
|
@ -0,0 +1,169 @@
|
|||
// Copyright 2017 Vector Creations Ltd
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package routing
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"encoding/json"
|
||||
"github.com/matrix-org/dendrite/clientapi/auth/authtypes"
|
||||
"github.com/matrix-org/dendrite/clientapi/jsonerror"
|
||||
"github.com/matrix-org/dendrite/common/config"
|
||||
)
|
||||
|
||||
type MockAccountsDB struct{}
|
||||
type MockDevicesDB struct{}
|
||||
|
||||
func (MockAccountsDB) GetAccountByPassword(context.Context, string, string) (*authtypes.Account, error) {
|
||||
var fakeAccount = authtypes.Account{
|
||||
"reguser:matrix.org",
|
||||
"reguser",
|
||||
"matrix.org",
|
||||
&authtypes.Profile{},
|
||||
"",
|
||||
}
|
||||
return &fakeAccount, nil
|
||||
}
|
||||
|
||||
func (MockDevicesDB) CreateDevice(context.Context, string, *string, string, *string) (*authtypes.Device, error) {
|
||||
var fakeDevice = authtypes.Device{
|
||||
"fakedevices",
|
||||
"requser:matrix.org",
|
||||
"asikdhfkhdsafkjsdjkfasdf",
|
||||
}
|
||||
return &fakeDevice, nil
|
||||
}
|
||||
|
||||
type InvalidAccountDB struct{}
|
||||
type InvalidDeviceDB struct{}
|
||||
type fakeErr struct{}
|
||||
|
||||
func (fakeErr) Error() string { return "forced error" }
|
||||
|
||||
func (InvalidAccountDB) GetAccountByPassword(context.Context, string, string) (*authtypes.Account, error) {
|
||||
var err fakeErr
|
||||
return nil, err
|
||||
}
|
||||
|
||||
func (InvalidDeviceDB) CreateDevice(context.Context, string, *string, string, *string) (*authtypes.Device, error) {
|
||||
var err fakeErr
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var (
|
||||
accountDB = MockAccountsDB{}
|
||||
deviceDB = MockDevicesDB{}
|
||||
BadAccountDB = InvalidAccountDB{}
|
||||
BadDeviceDB = InvalidDeviceDB{}
|
||||
cfg = config.Dendrite{}
|
||||
)
|
||||
|
||||
type loginRequest struct {
|
||||
User string
|
||||
Password string
|
||||
}
|
||||
|
||||
// not sending the user at all
|
||||
func TestNoUserLogin(t *testing.T) {
|
||||
u := loginRequest{"", "blahblah"}
|
||||
b := new(bytes.Buffer)
|
||||
json.NewEncoder(b).Encode(u)
|
||||
req := httptest.NewRequest("POST", "https://api.com", b)
|
||||
err := Login(req, accountDB, deviceDB, cfg).JSON.(*jsonerror.MatrixError).ErrCode
|
||||
if err != "M_BAD_JSON" {
|
||||
t.Error("No usern verification expected error_code 'M_BAD_JSON' got: ", err)
|
||||
}
|
||||
}
|
||||
|
||||
// sending invalid username
|
||||
func TestInvalidUsername(t *testing.T) {
|
||||
cfg.Matrix.ServerName = "matrix.org"
|
||||
u := loginRequest{"@minions", "blahblah"}
|
||||
b := new(bytes.Buffer)
|
||||
json.NewEncoder(b).Encode(u)
|
||||
req := httptest.NewRequest("POST", "https://api.com", b)
|
||||
err := Login(req, accountDB, deviceDB, cfg).JSON.(*jsonerror.MatrixError).ErrCode
|
||||
if err != "M_INVALID_USERNAME" {
|
||||
t.Error("invalid username verification, expected error_code 'M_INVALID_USERNAME' got: ", err)
|
||||
}
|
||||
}
|
||||
|
||||
// sending wrong servername in ID
|
||||
func TestInvalidServerName(t *testing.T) {
|
||||
cfg.Matrix.ServerName = "matrix.org"
|
||||
u := loginRequest{"@minions:notmatrix.org", "blahblah"}
|
||||
b := new(bytes.Buffer)
|
||||
json.NewEncoder(b).Encode(u)
|
||||
req := httptest.NewRequest("POST", "https://api.com", b)
|
||||
err := Login(req, accountDB, deviceDB, cfg).JSON.(*jsonerror.MatrixError).ErrCode
|
||||
if err != "M_INVALID_USERNAME" {
|
||||
t.Error("invalid servername verification, expected error_code 'M_INVALID_USERNAME' got: ", err)
|
||||
}
|
||||
}
|
||||
|
||||
// wrong username password
|
||||
func TestWrongUsernamePassword(t *testing.T) {
|
||||
cfg.Matrix.ServerName = "matrix.org"
|
||||
u := loginRequest{"@minions:matrix.org", "blahblah"}
|
||||
b := new(bytes.Buffer)
|
||||
json.NewEncoder(b).Encode(u)
|
||||
req := httptest.NewRequest("POST", "https://api.com", b)
|
||||
err := Login(req, BadAccountDB, deviceDB, cfg).JSON.(*jsonerror.MatrixError).ErrCode
|
||||
if err != "M_FORBIDDEN" {
|
||||
t.Error("wrong username password verification, expected error_code 'M_FORBIDDEN' got: ", err)
|
||||
}
|
||||
}
|
||||
|
||||
// devices database failure
|
||||
func TestDeviceNotCreated(t *testing.T) {
|
||||
cfg.Matrix.ServerName = "matrix.org"
|
||||
u := loginRequest{"minions", "blahblah"}
|
||||
b := new(bytes.Buffer)
|
||||
json.NewEncoder(b).Encode(u)
|
||||
req := httptest.NewRequest("POST", "https://api.com", b)
|
||||
err := Login(req, accountDB, BadDeviceDB, cfg).JSON.(*jsonerror.MatrixError).ErrCode
|
||||
if err != "M_UNKNOWN" {
|
||||
t.Error("device database failure verification, expected error_code 'M_UNKNOWN' got: ", err)
|
||||
}
|
||||
}
|
||||
|
||||
// login with userID
|
||||
func TestValidLoginviaUserID(t *testing.T) {
|
||||
cfg.Matrix.ServerName = "matrix.org"
|
||||
u := loginRequest{"@minions:matrix.org", "blahblah"}
|
||||
b := new(bytes.Buffer)
|
||||
json.NewEncoder(b).Encode(u)
|
||||
req := httptest.NewRequest("POST", "https://api.com", b)
|
||||
err := Login(req, accountDB, deviceDB, cfg).Code
|
||||
if err != 200 {
|
||||
t.Error("valid login via userID verification, expected http response code '200' got: ", err)
|
||||
}
|
||||
}
|
||||
|
||||
// login with username
|
||||
func TestValidLoginViaUsername(t *testing.T) {
|
||||
cfg.Matrix.ServerName = "matrix.org"
|
||||
u := loginRequest{"minions", "blahblah"}
|
||||
b := new(bytes.Buffer)
|
||||
json.NewEncoder(b).Encode(u)
|
||||
req := httptest.NewRequest("POST", "https://api.com", b)
|
||||
err := Login(req, accountDB, deviceDB, cfg).Code
|
||||
if err != 200 {
|
||||
t.Error("valid login verification via username, expected http response code '200' got: ", err)
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,102 @@
|
|||
package routing
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/matrix-org/dendrite/clientapi/auth/authtypes"
|
||||
"github.com/matrix-org/dendrite/clientapi/jsonerror"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func (MockAccountsDB) GetProfileByLocalpart(context.Context, string) (*authtypes.Profile, error) {
|
||||
return &authtypes.Profile{
|
||||
"minions",
|
||||
"minions",
|
||||
"http://minion.org",
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (MockAccountsDB) SetAvatarURL(context.Context, string, string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (MockAccountsDB) GetMembershipsByLocalpart(context.Context, string) ([]authtypes.Membership, error) {
|
||||
return []authtypes.Membership{
|
||||
{
|
||||
"minions",
|
||||
"despicable_me",
|
||||
"mission_1",
|
||||
},
|
||||
{
|
||||
"minions",
|
||||
"despicable_me_2",
|
||||
"mission_2",
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (MockAccountsDB) SetDisplayName(context.Context, string, string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (InvalidAccountDB) GetProfileByLocalpart(context.Context, string) (*authtypes.Profile, error) {
|
||||
var err fakeErr
|
||||
return nil, err
|
||||
}
|
||||
|
||||
func (InvalidAccountDB) SetAvatarURL(context.Context, string, string) error {
|
||||
var err fakeErr
|
||||
return err
|
||||
}
|
||||
|
||||
func (InvalidAccountDB) GetMembershipsByLocalpart(context.Context, string) ([]authtypes.Membership, error) {
|
||||
var err fakeErr
|
||||
return nil, err
|
||||
}
|
||||
|
||||
func (InvalidAccountDB) SetDisplayName(context.Context, string, string) error {
|
||||
var err fakeErr
|
||||
return err
|
||||
}
|
||||
|
||||
// Post method
|
||||
func TestPostMethodProfile(t *testing.T) {
|
||||
const userID = "@minions:matrix.org"
|
||||
req := httptest.NewRequest("POST", "https://api.com", nil)
|
||||
err := GetProfile(req, accountDB, userID).JSON.(*jsonerror.MatrixError).ErrCode
|
||||
if err != "M_NOT_FOUND" {
|
||||
t.Error("POST method verification expected error_code 'M_NOT_FOUND' got: ", err)
|
||||
}
|
||||
}
|
||||
|
||||
// invalid userID
|
||||
func TestInvalidUserID(t *testing.T) {
|
||||
const userID = "@minimatrix"
|
||||
req := httptest.NewRequest("GET", "https://api.com", nil)
|
||||
err := GetProfile(req, accountDB, userID).Code
|
||||
if err != 500 {
|
||||
t.Error("invalid userID verification expected error_code '500' got: ", err)
|
||||
}
|
||||
}
|
||||
|
||||
// Database Error
|
||||
func TestDatabaseError(t *testing.T) {
|
||||
const userID = "@minions:matrix.org"
|
||||
req := httptest.NewRequest("GET", "https://api.com", nil)
|
||||
err := GetProfile(req, BadAccountDB, userID).Code
|
||||
if err != 500 {
|
||||
t.Error("Database Error verification expected error_code '500' got: ", err)
|
||||
}
|
||||
}
|
||||
|
||||
// Valid Request
|
||||
func TestValidGetProfileRequest(t *testing.T) {
|
||||
const userID = "@minions:matrix.org"
|
||||
req := httptest.NewRequest("GET", "https://api.com", nil)
|
||||
err := GetProfile(req, accountDB, userID).Code
|
||||
if err != 200 {
|
||||
t.Error("Valid GetProfile verification expected code '200' got: ", err)
|
||||
}
|
||||
}
|
||||
|
||||
// Todo write similar tests for other get and put functions
|
||||
Loading…
Reference in a new issue