From 9f14c085c4893448d22abc138a5692d6622369c9 Mon Sep 17 00:00:00 2001 From: Future Outlier Date: Sun, 10 Dec 2023 18:43:19 +0800 Subject: [PATCH] add small test Signed-off-by: Future Outlier --- flyteplugins/go/tasks/plugins/webapi/agent/plugin_test.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/flyteplugins/go/tasks/plugins/webapi/agent/plugin_test.go b/flyteplugins/go/tasks/plugins/webapi/agent/plugin_test.go index a12d21818e..446d3563d4 100644 --- a/flyteplugins/go/tasks/plugins/webapi/agent/plugin_test.go +++ b/flyteplugins/go/tasks/plugins/webapi/agent/plugin_test.go @@ -60,6 +60,12 @@ func TestPlugin(t *testing.T) { assert.NotNil(t, err) }) + t.Run("test getAgentMetadataClientFunc", func(t *testing.T) { + client, err := getAgentMetadataClientFunc(context.Background(), &Agent{Endpoint: "localhost:80"}, map[*Agent]*grpc.ClientConn{}) + assert.NoError(t, err) + assert.NotNil(t, client) + }) + t.Run("test getAsyncClientFunc", func(t *testing.T) { client, err := getAsyncClientFunc(context.Background(), &Agent{Endpoint: "localhost:80"}, map[*Agent]*grpc.ClientConn{}) assert.NoError(t, err)