Add test for db schema with a UNIQUE constraint

xataio · Jan 18, 2024 · a0bf665 · a0bf665
1 parent 5b7a2f3
commit a0bf665
Showing 1 changed file with 47 additions and 0 deletions.
diff --git a/pkg/state/state_test.go b/pkg/state/state_test.go
@@ -107,6 +107,12 @@ func TestReadSchema(t *testing.T) {
 									Name: "id_unique",
 								},
 							},
+							UniqueConstraints: map[string]schema.UniqueConstraint{
+								"id_unique": {
+									Name:    "id_unique",
+									Columns: []string{"id"},
+								},
+							},
 						},
 					},
 				},
@@ -193,6 +199,47 @@ func TestReadSchema(t *testing.T) {
 					},
 				},
 			},
+			{
+				name:       "unique constraint",
+				createStmt: "CREATE TABLE public.table1 (id int PRIMARY KEY, name TEXT, CONSTRAINT name_unique UNIQUE(name) );",
+				wantSchema: &schema.Schema{
+					Name: "public",
+					Tables: map[string]schema.Table{
+						"table1": {
+							Name: "table1",
+							Columns: map[string]schema.Column{
+								"id": {
+									Name:     "id",
+									Type:     "integer",
+									Nullable: false,
+									Unique:   true,
+								},
+								"name": {
+									Name:     "name",
+									Type:     "text",
+									Unique:   true,
+									Nullable: true,
+								},
+							},
+							PrimaryKey: []string{"id"},
+							Indexes: map[string]schema.Index{
+								"table1_pkey": {
+									Name: "table1_pkey",
+								},
+								"name_unique": {
+									Name: "name_unique",
+								},
+							},
+							UniqueConstraints: map[string]schema.UniqueConstraint{
+								"name_unique": {
+									Name:    "name_unique",
+									Columns: []string{"name"},
+								},
+							},
+						},
+					},
+				},
+			},
 		}
 
 		// init the state