Skip to content

Commit d0322b7

Browse files
authored
Version grpc client (#121)
* add version client * Fix interface * Update mock interface
1 parent fe873c4 commit d0322b7

File tree

2 files changed

+11
-1
lines changed

2 files changed

+11
-1
lines changed

pkg/client/client.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ type Client interface {
2222
ProjectMember() v1.ProjectMemberServiceClient
2323
Tenant() v1.TenantServiceClient
2424
TenantMember() v1.TenantMemberServiceClient
25+
Version() v1.VersionServiceClient
2526
Close() error
2627
}
2728

@@ -128,3 +129,7 @@ func (c GRPCClient) Tenant() v1.TenantServiceClient {
128129
func (c GRPCClient) TenantMember() v1.TenantMemberServiceClient {
129130
return v1.NewTenantMemberServiceClient(c.conn)
130131
}
132+
133+
func (c GRPCClient) Version() v1.VersionServiceClient {
134+
return v1.NewVersionServiceClient(c.conn)
135+
}

pkg/client/mock.go

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,14 +9,16 @@ type MockClient struct {
99
tsc v1.TenantServiceClient
1010
pmsc v1.ProjectMemberServiceClient
1111
tmsc v1.TenantMemberServiceClient
12+
vsc v1.VersionServiceClient
1213
}
1314

14-
func NewMock(psc v1.ProjectServiceClient, tsc v1.TenantServiceClient, pmsc v1.ProjectMemberServiceClient, tmsc v1.TenantMemberServiceClient) *MockClient {
15+
func NewMock(psc v1.ProjectServiceClient, tsc v1.TenantServiceClient, pmsc v1.ProjectMemberServiceClient, tmsc v1.TenantMemberServiceClient, vsc v1.VersionServiceClient) *MockClient {
1516
return &MockClient{
1617
psc: psc,
1718
tsc: tsc,
1819
pmsc: pmsc,
1920
tmsc: tmsc,
21+
vsc: vsc,
2022
}
2123
}
2224

@@ -36,3 +38,6 @@ func (c *MockClient) Tenant() v1.TenantServiceClient {
3638
func (c *MockClient) TenantMember() v1.TenantMemberServiceClient {
3739
return c.tmsc
3840
}
41+
func (c *MockClient) Version() v1.VersionServiceClient {
42+
return c.vsc
43+
}

0 commit comments

Comments
 (0)