81 lines
2.6 KiB
Go
81 lines
2.6 KiB
Go
// Copyright (c) Mainflux
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package mocks
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/golang/protobuf/ptypes/empty"
|
|
"github.com/mainflux/mainflux"
|
|
"github.com/mainflux/mainflux/pkg/errors"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
var _ mainflux.AuthServiceClient = (*authServiceMock)(nil)
|
|
|
|
type SubjectSet struct {
|
|
Object string
|
|
Relation string
|
|
}
|
|
|
|
type authServiceMock struct {
|
|
users map[string]string
|
|
authz map[string][]SubjectSet
|
|
}
|
|
|
|
func (svc authServiceMock) ListPolicies(ctx context.Context, in *mainflux.ListPoliciesReq, opts ...grpc.CallOption) (*mainflux.ListPoliciesRes, error) {
|
|
panic("not implemented")
|
|
}
|
|
|
|
// NewAuthService creates mock of users service.
|
|
func NewAuthService(users map[string]string, authzDB map[string][]SubjectSet) mainflux.AuthServiceClient {
|
|
return &authServiceMock{users, authzDB}
|
|
}
|
|
|
|
func (svc authServiceMock) Identify(ctx context.Context, in *mainflux.Token, opts ...grpc.CallOption) (*mainflux.UserIdentity, error) {
|
|
if id, ok := svc.users[in.Value]; ok {
|
|
return &mainflux.UserIdentity{Id: id, Email: id}, nil
|
|
}
|
|
return nil, errors.ErrAuthentication
|
|
}
|
|
|
|
func (svc authServiceMock) Issue(ctx context.Context, in *mainflux.IssueReq, opts ...grpc.CallOption) (*mainflux.Token, error) {
|
|
if id, ok := svc.users[in.GetEmail()]; ok {
|
|
switch in.Type {
|
|
default:
|
|
return &mainflux.Token{Value: id}, nil
|
|
}
|
|
}
|
|
return nil, errors.ErrAuthentication
|
|
}
|
|
|
|
func (svc authServiceMock) Authorize(ctx context.Context, req *mainflux.AuthorizeReq, _ ...grpc.CallOption) (r *mainflux.AuthorizeRes, err error) {
|
|
if sub, ok := svc.authz[req.GetSub()]; ok {
|
|
for _, v := range sub {
|
|
if v.Relation == req.GetAct() && v.Object == req.GetObj() {
|
|
return &mainflux.AuthorizeRes{Authorized: true}, nil
|
|
}
|
|
}
|
|
}
|
|
return &mainflux.AuthorizeRes{Authorized: false}, nil
|
|
}
|
|
|
|
func (svc authServiceMock) AddPolicy(ctx context.Context, in *mainflux.AddPolicyReq, opts ...grpc.CallOption) (*mainflux.AddPolicyRes, error) {
|
|
svc.authz[in.GetSub()] = append(svc.authz[in.GetSub()], SubjectSet{Object: in.GetObj(), Relation: in.GetAct()})
|
|
return &mainflux.AddPolicyRes{Authorized: true}, nil
|
|
}
|
|
|
|
func (svc authServiceMock) DeletePolicy(ctx context.Context, in *mainflux.DeletePolicyReq, opts ...grpc.CallOption) (*mainflux.DeletePolicyRes, error) {
|
|
// Not implemented
|
|
return &mainflux.DeletePolicyRes{Deleted: true}, nil
|
|
}
|
|
|
|
func (svc authServiceMock) Members(ctx context.Context, req *mainflux.MembersReq, _ ...grpc.CallOption) (r *mainflux.MembersRes, err error) {
|
|
panic("not implemented")
|
|
}
|
|
|
|
func (svc authServiceMock) Assign(ctx context.Context, req *mainflux.Assignment, _ ...grpc.CallOption) (r *empty.Empty, err error) {
|
|
panic("not implemented")
|
|
}
|