int tests

Snowflake-Labs · Oct 12, 2023 · 5969b8e · 5969b8e
1 parent 261ab6f
commit 5969b8e
Show file tree

Hide file tree

Showing 3 changed files with 102 additions and 98 deletions.
diff --git a/pkg/sdk/testint/external_tables_integration_test.go b/pkg/sdk/testint/external_tables_integration_test.go
@@ -408,14 +408,3 @@ func TestInt_ExternalTables(t *testing.T) {
 		})
 	})
 }
-
-type AllowedValue struct {
-	Value string `ddl:"keyword,single_quotes"`
-}
-
-func TestT(t *testing.T) {
-	a := AllowedValue{Value: "123"}
-	b, err := structToSQL(a)
-	require.NoError(t, err)
-	assert.Equal(t, `'123'`, b)
-}
diff --git a/pkg/sdk/testint/helpers.go b/pkg/sdk/testint/helpers.go
@@ -279,19 +279,20 @@ func createTagWithOptions(t *testing.T, client *sdk.Client, database *sdk.Databa
 		}
 }
 
-func createStageWithName(t *testing.T, client *sdk.Client, name string) (*string, func()) {
+func createStageWithDirectory(t *testing.T, client *sdk.Client, database *sdk.Database, schema *sdk.Schema, name string) (*sdk.Stage, func()) {
 	t.Helper()
 	ctx := context.Background()
-	stageCleanup := func() {
-		_, err := client.ExecForTests(ctx, fmt.Sprintf("DROP STAGE %s", name))
-		require.NoError(t, err)
-	}
-	_, err := client.ExecForTests(ctx, fmt.Sprintf("CREATE STAGE %s", name))
-	if err != nil {
-		return nil, stageCleanup
-	}
+	_, err := client.ExecForTests(ctx, fmt.Sprintf(`CREATE STAGE "%s" DIRECTORY = (ENABLE = TRUE)`, name))
 	require.NoError(t, err)
-	return &name, stageCleanup
+
+	return &sdk.Stage{
+			DatabaseName: database.Name,
+			SchemaName:   schema.Name,
+			Name:         name,
+		}, func() {
+			_, err := client.ExecForTests(ctx, fmt.Sprintf(`DROP STAGE "%s"`, name))
+			require.NoError(t, err)
+		}
 }
 
 func createStage(t *testing.T, client *sdk.Client, database *sdk.Database, schema *sdk.Schema, name string) (*sdk.Stage, func()) {
@@ -689,3 +690,15 @@ func createFileFormatWithOptions(t *testing.T, client *sdk.Client, schema sdk.Da
 		require.NoError(t, err)
 	}
 }
+
+func createView(t *testing.T, client *sdk.Client, viewId sdk.SchemaObjectIdentifier, asQuery string) func() {
+	t.Helper()
+	ctx := context.Background()
+	_, err := client.ExecForTests(ctx, fmt.Sprintf(`CREATE VIEW %s AS %s`, viewId.FullyQualifiedName(), asQuery))
+	require.NoError(t, err)
+
+	return func() {
+		_, err := client.ExecForTests(ctx, fmt.Sprintf(`DROP VIEW %s`, viewId.FullyQualifiedName()))
+		require.NoError(t, err)
+	}
+}