Function TestWriteSchemaWithClusteringKeys

core/schema_test.go:325–355

(t *testing.T)

}

func TestWriteSchemaWithClusteringKeys(t *testing.T) {
	var buf bytes.Buffer

	di := &sdata.DBInfo{
		Type:    "snowflake",
		Version: 1,
		Schema:  "main",
		Tables: []sdata.DBTable{
			{
				Name:   "events",
				Schema: "main",
				Columns: []sdata.DBColumn{
					{Name: "id", Type: "bigint", PrimaryKey: true, NotNull: true, Schema: "main", Table: "events"},
					{Name: "created_at", Type: "timestamp", NotNull: true, Schema: "main", Table: "events"},
					{Name: "region", Type: "varchar", NotNull: true, Schema: "main", Table: "events"},
				},
				ClusteringKeys: []string{"created_at", "region"},
			},
		},
	}

	if err := writeSchema(di, &buf); err != nil {
		t.Fatal(err)
	}

	output := buf.String()

	if !strings.Contains(output, `@cluster(columns: ["created_at", "region"])`) {
		t.Errorf("expected @cluster directive in output, got:\n%s", output)
	}
}

func TestSchemaRoundTrip_ClusteringKeys(t *testing.T) {
	schema := []byte(`# dbinfo:snowflake,1,main