Skip to content

Commit 089b84e

Browse files
committed
fix: update tests to remove NOT NULL expectation
1 parent 8d8e425 commit 089b84e

1 file changed

Lines changed: 15 additions & 15 deletions

File tree

db_proto/sql/risingwave/dialect_test.go

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -84,14 +84,14 @@ func TestDialectRisingwave_CreateTableStaticSql(t *testing.T) {
8484
// Check _cursor_ table
8585
assert.Contains(t, sql, "_cursor_")
8686
assert.Contains(t, sql, "name varchar primary key")
87-
assert.Contains(t, sql, "cursor varchar not null")
87+
assert.Contains(t, sql, "cursor varchar")
8888
assert.Contains(t, sql, "on conflict overwrite", "RisingWave should use ON CONFLICT OVERWRITE")
8989

9090
// Check _blocks_ table
9191
assert.Contains(t, sql, "_blocks_")
9292
assert.Contains(t, sql, "number integer")
93-
assert.Contains(t, sql, "hash varchar not null")
94-
assert.Contains(t, sql, "timestamp timestamp with time zone not null")
93+
assert.Contains(t, sql, "hash varchar")
94+
assert.Contains(t, sql, "timestamp timestamp with time zone")
9595
}
9696

9797
func TestDialectRisingwave_CreateTable_SimpleTable(t *testing.T) {
@@ -117,9 +117,9 @@ func TestDialectRisingwave_CreateTable_SimpleTable(t *testing.T) {
117117
assert.Equal(t, 1, len(d.CreateTableSql))
118118

119119
sql := d.CreateTableSql["users"]
120-
assert.Contains(t, sql, "CREATE TABLE IF NOT EXISTS public.users")
121-
assert.Contains(t, sql, "block_number INTEGER NOT NULL")
122-
assert.Contains(t, sql, "block_timestamp TIMESTAMP WITH TIME ZONE NOT NULL")
120+
assert.Contains(t, sql, "CREATE TABLE IF NOT EXISTS public.users")
121+
assert.Contains(t, sql, "block_number INTEGER")
122+
assert.Contains(t, sql, "block_timestamp TIMESTAMP WITH TIME ZONE")
123123
assert.Contains(t, sql, `"name" CHARACTER VARYING`)
124124
assert.Contains(t, sql, `"age" INTEGER`)
125125

@@ -197,9 +197,9 @@ func TestDialectRisingwave_CreateTable_ChildTable(t *testing.T) {
197197

198198
sql := d.CreateTableSql["mints"]
199199
assert.Contains(t, sql, "CREATE TABLE IF NOT EXISTS public.mints")
200-
assert.Contains(t, sql, "block_number INTEGER NOT NULL")
201-
assert.Contains(t, sql, "block_timestamp TIMESTAMP WITH TIME ZONE NOT NULL")
202-
assert.Contains(t, sql, "instruction_id CHARACTER VARYING NOT NULL")
200+
assert.Contains(t, sql, "block_number INTEGER")
201+
assert.Contains(t, sql, "block_timestamp TIMESTAMP WITH TIME ZONE")
202+
assert.Contains(t, sql, "instruction_id CHARACTER VARYING")
203203
assert.Contains(t, sql, `"amount" NUMERIC`)
204204

205205
// Should not contain foreign key constraints
@@ -589,8 +589,8 @@ func TestDialectRisingwave_ComplexTableStructure(t *testing.T) {
589589
assert.Contains(t, sql, `"email" CHARACTER VARYING UNIQUE`)
590590
assert.Contains(t, sql, `"balance" NUMERIC`)
591591
assert.Contains(t, sql, `"active" BOOLEAN`)
592-
assert.Contains(t, sql, "block_number INTEGER NOT NULL")
593-
assert.Contains(t, sql, "block_timestamp TIMESTAMP WITH TIME ZONE NOT NULL")
592+
assert.Contains(t, sql, "block_number INTEGER")
593+
assert.Contains(t, sql, "block_timestamp TIMESTAMP WITH TIME ZONE")
594594

595595
// Ensure no foreign keys
596596
assert.NotContains(t, sql, "FOREIGN KEY")
@@ -654,18 +654,18 @@ func TestDialectRisingwave_MultipleChildTables(t *testing.T) {
654654

655655
// Check first child table
656656
transferSQL := d.CreateTableSql["transfers"]
657-
assert.Contains(t, transferSQL, "transaction_id CHARACTER VARYING NOT NULL")
657+
assert.Contains(t, transferSQL, "transaction_id CHARACTER VARYING")
658658
assert.Contains(t, transferSQL, `"amount" NUMERIC`)
659659

660660
// Check second child table
661661
logSQL := d.CreateTableSql["logs"]
662-
assert.Contains(t, logSQL, "transaction_id CHARACTER VARYING NOT NULL")
662+
assert.Contains(t, logSQL, "transaction_id CHARACTER VARYING")
663663
assert.Contains(t, logSQL, `"message" CHARACTER VARYING`)
664664

665665
// All should have block metadata
666666
for _, sql := range []string{parentSQL, transferSQL, logSQL} {
667-
assert.Contains(t, sql, "block_number INTEGER NOT NULL")
668-
assert.Contains(t, sql, "block_timestamp TIMESTAMP WITH TIME ZONE NOT NULL")
667+
assert.Contains(t, sql, "block_number INTEGER")
668+
assert.Contains(t, sql, "block_timestamp TIMESTAMP WITH TIME ZONE")
669669
}
670670
}
671671

0 commit comments

Comments
 (0)