deal with google client mock

instill-ai · Oct 8, 2024 · 8dc4057 · 8dc4057
1 parent fb13661
commit 8dc4057
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 0 deletions.
diff --git a/mocks/bigquery_client_mock.go b/mocks/bigquery_client_mock.go
diff --git a/services/ bigquery_client.go b/services/ bigquery_client.go
@@ -0,0 +1,23 @@
+
+package bigqueryclient
+
+import (
+	"context"
+	"cloud.google.com/go/bigquery"
+)
+
+// BigQueryClient defines the interface for the BigQuery client
+type BigQueryClient interface {
+	CreateTable(ctx context.Context, datasetID string, tableID string, schema bigquery.Schema) error
+}
+
+// MyBigQueryClient implements the BigQueryClient interface
+type MyBigQueryClient struct {
+	client *bigquery.Client
+}
+
+// CreateTable creates a new table in BigQuery
+func (m *MyBigQueryClient) CreateTable(ctx context.Context, datasetID string, tableID string, schema bigquery.Schema) error {
+	table := m.client.Dataset(datasetID).Table(tableID)
+	return table.Create(ctx, &bigquery.TableMetadata{Schema: schema})
+}
diff --git a/services/myservice.go b/services/myservice.go
diff --git a/services/myservice_test.go b/services/myservice_test.go