p53841790
/
wheat-cache
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

test(client): add storage client

This commit is contained in:
bandl 2021-10-28 11:59:46 +08:00
parent dce8739514
commit 200623bd29
1 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
client/client_test.go
View File

@ -13,8 +13,10 @@ func TestClient(t *testing.T) {
cli, err := NewWheatClient("127.0.0.1:5891", middle.WithUnaryColonyClient) cli, err := NewWheatClient("127.0.0.1:5891", middle.WithUnaryColonyClient)
require.NoError(t, err) require.NoError(t, err)
ctx := context.Background() ctx := context.Background()
bKey := &proto.BaseKey{ bKey := &proto.BaseKey{
Key: "apple", Key: "apple",
Ttl: 10,
} }
resp, err := cli.Set(ctx, &proto.SetRequest{ resp, err := cli.Set(ctx, &proto.SetRequest{
Key: bKey, Key: bKey,
@ -29,5 +31,20 @@ func TestClient(t *testing.T) {
}) })
require.NoError(t, err) require.NoError(t, err)
require.Equal(t, getResp.Result, "yyyy") require.Equal(t, getResp.Result, "yyyy")
}
func TestClientGet(t *testing.T) {
cli, err := NewWheatClient("127.0.0.1:5891", middle.WithUnaryColonyClient)
require.NoError(t, err)
ctx := context.Background()
bKey := &proto.BaseKey{
Key: "apple",
}
getResp, err := cli.Get(ctx, &proto.GetRequest{
Key: bKey,
})
require.NoError(t, err)
require.Equal(t, getResp.Result, "yyyy")
} }