Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions pkg/client/client.go
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@ type Client interface {
ProjectMember() v1.ProjectMemberServiceClient
Tenant() v1.TenantServiceClient
TenantMember() v1.TenantMemberServiceClient
Version() v1.VersionServiceClient
Close() error
}

Expand Down Expand Up @@ -128,3 +129,7 @@ func (c GRPCClient) Tenant() v1.TenantServiceClient {
func (c GRPCClient) TenantMember() v1.TenantMemberServiceClient {
return v1.NewTenantMemberServiceClient(c.conn)
}

func (c GRPCClient) Version() v1.VersionServiceClient {
return v1.NewVersionServiceClient(c.conn)
}
7 changes: 6 additions & 1 deletion pkg/client/mock.go
Original file line number Diff line number Diff line change
Expand Up @@ -9,14 +9,16 @@ type MockClient struct {
tsc v1.TenantServiceClient
pmsc v1.ProjectMemberServiceClient
tmsc v1.TenantMemberServiceClient
vsc v1.VersionServiceClient
}

func NewMock(psc v1.ProjectServiceClient, tsc v1.TenantServiceClient, pmsc v1.ProjectMemberServiceClient, tmsc v1.TenantMemberServiceClient) *MockClient {
func NewMock(psc v1.ProjectServiceClient, tsc v1.TenantServiceClient, pmsc v1.ProjectMemberServiceClient, tmsc v1.TenantMemberServiceClient, vsc v1.VersionServiceClient) *MockClient {
return &MockClient{
psc: psc,
tsc: tsc,
pmsc: pmsc,
tmsc: tmsc,
vsc: vsc,
}
}

Expand All @@ -36,3 +38,6 @@ func (c *MockClient) Tenant() v1.TenantServiceClient {
func (c *MockClient) TenantMember() v1.TenantMemberServiceClient {
return c.tmsc
}
func (c *MockClient) Version() v1.VersionServiceClient {
return c.vsc
}