fix tests

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Jan 9, 2024 · 50b6ba0 · 50b6ba0
1 parent fbb619b
commit 50b6ba0
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/flyteidl/clients/go/admin/client.go b/flyteidl/clients/go/admin/client.go
@@ -6,6 +6,7 @@ import (
 	"crypto/x509"
 	"errors"
 	"fmt"
+
 	"github.com/flyteorg/flyte/flyteidl/gen/pb-go/flyteidl/artifact"
 
 	grpcRetry "github.com/grpc-ecosystem/go-grpc-middleware/retry"

diff --git a/flyteplugins/go/tasks/plugins/webapi/agent/integration_test.go b/flyteplugins/go/tasks/plugins/webapi/agent/integration_test.go
@@ -40,9 +40,25 @@ type MockPlugin struct {
 type MockAsyncTask struct {
 }
 
+func (m *MockAsyncTask) GetTaskMetrics(ctx context.Context, in *admin.GetTaskMetricsRequest, opts ...grpc.CallOption) (*admin.GetTaskMetricsResponse, error) {
+	panic("not implemented")
+}
+
+func (m *MockAsyncTask) GetTaskLogs(ctx context.Context, in *admin.GetTaskLogsRequest, opts ...grpc.CallOption) (*admin.GetTaskLogsResponse, error) {
+	panic("not implemented")
+}
+
 type MockSyncTask struct {
 }
 
+func (m *MockSyncTask) GetTaskMetrics(ctx context.Context, in *admin.GetTaskMetricsRequest, opts ...grpc.CallOption) (*admin.GetTaskMetricsResponse, error) {
+	panic("not implemented")
+}
+
+func (m *MockSyncTask) GetTaskLogs(ctx context.Context, in *admin.GetTaskLogsRequest, opts ...grpc.CallOption) (*admin.GetTaskLogsResponse, error) {
+	panic("not implemented")
+}
+
 func (m *MockAsyncTask) CreateTask(_ context.Context, createTaskRequest *admin.CreateTaskRequest, _ ...grpc.CallOption) (*admin.CreateTaskResponse, error) {
 	expectedArgs := []string{"pyflyte-fast-execute", "--output-prefix", "fake://bucket/prefix/nhv"}
 	if slices.Equal(createTaskRequest.Template.GetContainer().Args, expectedArgs) {