add small test

Signed-off-by: Future Outlier <[email protected]>
flyteorg · Dec 10, 2023 · 9f14c08 · 9f14c08
1 parent aa24087
commit 9f14c08
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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)