github/paralus
1
0
Fork 0
mirror of https://github.com/paralus/paralus.git synced 2026-05-06 00:17:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add apikey tests

Browse Source
This commit is contained in:
Abin Simon
2022-06-13 15:25:32 +05:30
parent ae7db385ea
commit e6a02705a7
1 changed files with 122 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

122
pkg/service/apikey_test.go Normal file
View File

@@ -0,0 +1,122 @@
package service
import (
"context"
"testing"
"github.com/DATA-DOG/go-sqlmock"
"github.com/google/uuid"
userrpcv3 "github.com/paralus/paralus/proto/rpc/user"
)
func TestApiKeyCreate(t *testing.T) {
db, mock := getDB(t)
defer db.Close()
ak := NewApiKeyService(db, getLogger())
uuuid := uuid.NewString()
auuid := uuid.NewString()
req := &userrpcv3.ApiKeyRequest{Username: "user-" + uuuid, Id: uuuid}
// mocks
mock.ExpectQuery(`INSERT INTO "authsrv_apikey"`).
WithArgs().WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(auuid, "apikey-"+auuid))
resp, err := ak.Create(context.Background(), req)
if err != nil {
t.Error("unable to create apikey:", err)
}
if resp.ID != uuid.MustParse(auuid) {
t.Errorf("incorrect id for apikey; expected '%v', got '%v'", uuid.MustParse(auuid), resp.ID)
}
if resp.Name != "apikey-"+auuid {
t.Errorf("incorrect name for apikey; expected '%v', got '%v'", "apikey-"+auuid, resp.Name)
}
}
func TestApiKeyDelete(t *testing.T) {
db, mock := getDB(t)
defer db.Close()
ak := NewApiKeyService(db, getLogger())
uuuid := uuid.NewString()
req := &userrpcv3.ApiKeyRequest{Username: "user-" + uuuid, Id: uuuid}
// mocks
mock.ExpectExec(`UPDATE "authsrv_apikey" AS "apikey" SET trash = TRUE WHERE \(account_id = 'user-` + uuuid + `'\) AND \(key = '` + uuuid + `'\)`).
WillReturnResult(sqlmock.NewResult(1, 1))
_, err := ak.Delete(context.Background(), req)
if err != nil {
t.Error("unable to delete apikey:", err)
}
}
func TestApiKeyGet(t *testing.T) {
db, mock := getDB(t)
defer db.Close()
ak := NewApiKeyService(db, getLogger())
uuuid := uuid.NewString()
req := &userrpcv3.ApiKeyRequest{Username: "user-" + uuuid, Id: uuuid}
// mocks
mock.ExpectQuery(`SELECT "apikey"."id", "apikey"."name", .*FROM "authsrv_apikey" AS "apikey" WHERE \(name = 'user-` + uuuid + `'\) AND \(trash = FALSE\)`).
WithArgs().WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(uuuid, "user-"+uuuid))
resp, err := ak.Get(context.Background(), req)
if err != nil {
t.Error("unable to get apikey:", err)
}
if resp.ID != uuid.MustParse(uuuid) {
t.Errorf("incorrect id for apikey; expected '%v', got '%v'", uuid.MustParse(uuuid), resp.ID)
}
if resp.Name != "user-"+uuuid {
t.Errorf("incorrect name for apikey; expected '%v', got '%v'", "apikey-"+uuuid, resp.Name)
}
}
func TestApiKeyGetByKey(t *testing.T) {
db, mock := getDB(t)
defer db.Close()
ak := NewApiKeyService(db, getLogger())
uuuid := uuid.NewString()
req := &userrpcv3.ApiKeyRequest{Username: "user-" + uuuid, Id: uuuid}
// mocks
mock.ExpectQuery(`SELECT "apikey"."id", "apikey"."name", .*FROM "authsrv_apikey" AS "apikey" WHERE \(key = '` + uuuid + `'\) AND \(trash = FALSE\)`).
WithArgs().WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(uuuid, "user-"+uuuid))
resp, err := ak.GetByKey(context.Background(), req)
if err != nil {
t.Error("unable to get apikey:", err)
}
if resp.ID != uuid.MustParse(uuuid) {
t.Errorf("incorrect id for apikey; expected '%v', got '%v'", uuid.MustParse(uuuid), resp.ID)
}
if resp.Name != "user-"+uuuid {
t.Errorf("incorrect name for apikey; expected '%v', got '%v'", "apikey-"+uuuid, resp.Name)
}
}
func TestApiKeyList(t *testing.T) {
db, mock := getDB(t)
defer db.Close()
ak := NewApiKeyService(db, getLogger())
uuuid := uuid.NewString()
req := &userrpcv3.ApiKeyRequest{Username: "user-" + uuuid, Id: uuuid}
// mocks
mock.ExpectQuery(`SELECT "apikey"."id", "apikey"."name", .*FROM "authsrv_apikey" AS "apikey" WHERE \(account_id = 'user-` + uuuid + `'\) AND \(trash = FALSE\)`).
WithArgs().WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(uuuid, "user-"+uuuid))
resp, err := ak.List(context.Background(), req)
if err != nil {
t.Error("unable to list apikey:", err)
}
if resp.Items[0].Name != "user-"+uuuid {
t.Errorf("incorrect name for apikey; expected '%v', got '%v'", "apikey-"+uuuid, resp.Items[0].Name)
}
}