mirror of
https://github.com/absmach/supermq.git
synced 2026-06-23 04:20:17 +00:00
dusanb94
118 lines
3.9 KiB
Go
118 lines
3.9 KiB
Go
// Copyright (c) Magistrala
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package mocks
|
|
|
|
import (
|
|
context "context"
|
|
|
|
mainflux "github.com/absmach/magistrala"
|
|
"github.com/absmach/magistrala/pkg/errors"
|
|
"github.com/stretchr/testify/mock"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
const InvalidValue = "invalid"
|
|
|
|
var _ mainflux.AuthServiceClient = (*Service)(nil)
|
|
|
|
type Service struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (m *Service) Issue(ctx context.Context, in *mainflux.IssueReq, opts ...grpc.CallOption) (*mainflux.Token, error) {
|
|
ret := m.Called(ctx, in)
|
|
if in.GetId() == InvalidValue || in.GetId() == "" {
|
|
return &mainflux.Token{}, errors.ErrAuthentication
|
|
}
|
|
|
|
return ret.Get(0).(*mainflux.Token), ret.Error(1)
|
|
}
|
|
|
|
func (m *Service) Login(ctx context.Context, in *mainflux.LoginReq, opts ...grpc.CallOption) (*mainflux.Token, error) {
|
|
ret := m.Called(ctx, in)
|
|
if in.GetId() == InvalidValue || in.GetId() == "" {
|
|
return &mainflux.Token{}, errors.ErrAuthentication
|
|
}
|
|
|
|
return ret.Get(0).(*mainflux.Token), ret.Error(1)
|
|
}
|
|
|
|
func (m *Service) Refresh(ctx context.Context, in *mainflux.RefreshReq, opts ...grpc.CallOption) (*mainflux.Token, error) {
|
|
ret := m.Called(ctx, in)
|
|
if in.GetValue() == InvalidValue || in.GetValue() == "" {
|
|
return &mainflux.Token{}, errors.ErrAuthentication
|
|
}
|
|
|
|
return ret.Get(0).(*mainflux.Token), ret.Error(1)
|
|
}
|
|
|
|
func (m *Service) Identify(ctx context.Context, in *mainflux.IdentityReq, opts ...grpc.CallOption) (*mainflux.IdentityRes, error) {
|
|
ret := m.Called(ctx, in)
|
|
if in.GetToken() == InvalidValue || in.GetToken() == "" {
|
|
return &mainflux.IdentityRes{}, errors.ErrAuthentication
|
|
}
|
|
|
|
return ret.Get(0).(*mainflux.IdentityRes), ret.Error(1)
|
|
}
|
|
|
|
func (m *Service) Authorize(ctx context.Context, in *mainflux.AuthorizeReq, opts ...grpc.CallOption) (*mainflux.AuthorizeRes, error) {
|
|
ret := m.Called(ctx, in)
|
|
if in.GetSubject() == InvalidValue || in.GetSubject() == "" {
|
|
return &mainflux.AuthorizeRes{Authorized: false}, errors.ErrAuthorization
|
|
}
|
|
if in.GetObject() == InvalidValue || in.GetObject() == "" {
|
|
return &mainflux.AuthorizeRes{Authorized: false}, errors.ErrAuthorization
|
|
}
|
|
|
|
return ret.Get(0).(*mainflux.AuthorizeRes), ret.Error(1)
|
|
}
|
|
|
|
func (m *Service) AddPolicy(ctx context.Context, in *mainflux.AddPolicyReq, opts ...grpc.CallOption) (*mainflux.AddPolicyRes, error) {
|
|
ret := m.Called(ctx, in)
|
|
|
|
return ret.Get(0).(*mainflux.AddPolicyRes), ret.Error(1)
|
|
}
|
|
|
|
func (m *Service) DeletePolicy(ctx context.Context, in *mainflux.DeletePolicyReq, opts ...grpc.CallOption) (*mainflux.DeletePolicyRes, error) {
|
|
ret := m.Called(ctx, in)
|
|
|
|
return ret.Get(0).(*mainflux.DeletePolicyRes), ret.Error(1)
|
|
}
|
|
|
|
func (m *Service) ListObjects(ctx context.Context, in *mainflux.ListObjectsReq, opts ...grpc.CallOption) (*mainflux.ListObjectsRes, error) {
|
|
ret := m.Called(ctx, in)
|
|
|
|
return ret.Get(0).(*mainflux.ListObjectsRes), ret.Error(1)
|
|
}
|
|
|
|
func (m *Service) ListAllObjects(ctx context.Context, in *mainflux.ListObjectsReq, opts ...grpc.CallOption) (*mainflux.ListObjectsRes, error) {
|
|
ret := m.Called(ctx, in)
|
|
|
|
return ret.Get(0).(*mainflux.ListObjectsRes), ret.Error(1)
|
|
}
|
|
|
|
func (m *Service) CountObjects(ctx context.Context, in *mainflux.CountObjectsReq, opts ...grpc.CallOption) (*mainflux.CountObjectsRes, error) {
|
|
ret := m.Called(ctx, in)
|
|
|
|
return ret.Get(0).(*mainflux.CountObjectsRes), ret.Error(1)
|
|
}
|
|
|
|
func (m *Service) ListSubjects(ctx context.Context, in *mainflux.ListSubjectsReq, opts ...grpc.CallOption) (*mainflux.ListSubjectsRes, error) {
|
|
ret := m.Called(ctx, in)
|
|
|
|
return ret.Get(0).(*mainflux.ListSubjectsRes), ret.Error(1)
|
|
}
|
|
|
|
func (m *Service) ListAllSubjects(ctx context.Context, in *mainflux.ListSubjectsReq, opts ...grpc.CallOption) (*mainflux.ListSubjectsRes, error) {
|
|
ret := m.Called(ctx, in)
|
|
|
|
return ret.Get(0).(*mainflux.ListSubjectsRes), ret.Error(1)
|
|
}
|
|
|
|
func (m *Service) CountSubjects(ctx context.Context, in *mainflux.CountSubjectsReq, opts ...grpc.CallOption) (*mainflux.CountSubjectsRes, error) {
|
|
ret := m.Called(ctx, in)
|
|
|
|
return ret.Get(0).(*mainflux.CountSubjectsRes), ret.Error(1)
|
|
}
|