diff options
Diffstat (limited to 'internal/db/mocks.go')
-rw-r--r-- | internal/db/mocks.go | 19 |
1 files changed, 17 insertions, 2 deletions
diff --git a/internal/db/mocks.go b/internal/db/mocks.go index 8cef2b96..12622a7c 100644 --- a/internal/db/mocks.go +++ b/internal/db/mocks.go @@ -15,14 +15,29 @@ import ( var _ AccessTokensStore = (*MockAccessTokensStore)(nil) type MockAccessTokensStore struct { - MockGetBySHA func(sha string) (*AccessToken, error) - MockSave func(t *AccessToken) error + MockCreate func(userID int64, name string) (*AccessToken, error) + MockDeleteByID func(userID, id int64) error + MockGetBySHA func(sha string) (*AccessToken, error) + MockList func(userID int64) ([]*AccessToken, error) + MockSave func(t *AccessToken) error +} + +func (m *MockAccessTokensStore) Create(userID int64, name string) (*AccessToken, error) { + return m.MockCreate(userID, name) +} + +func (m *MockAccessTokensStore) DeleteByID(userID, id int64) error { + return m.MockDeleteByID(userID, id) } func (m *MockAccessTokensStore) GetBySHA(sha string) (*AccessToken, error) { return m.MockGetBySHA(sha) } +func (m *MockAccessTokensStore) List(userID int64) ([]*AccessToken, error) { + return m.MockList(userID) +} + func (m *MockAccessTokensStore) Save(t *AccessToken) error { return m.MockSave(t) } |