fixed

gophersbd · May 21, 2018 · eeb183a · eeb183a
1 parent 7acce77
commit eeb183a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/pkg/dialect/postgres/dialect_postgres.go b/pkg/dialect/postgres/dialect_postgres.go
@@ -54,7 +54,7 @@ func (s *postgres) ColumnSignatureOf(field *descriptor.Field) string {
 	options := field.Column.Options
 	d := options.GetDefault()
 	if d != "" {
-		additionalType = additionalType + fmt.Sprintf("DEFAULT %v", d)
+		additionalType = additionalType + " " + fmt.Sprintf("DEFAULT %v", d)
 	}
 
 	st := sqlType.Name

diff --git a/pkg/dialect/postgres/dialect_postgres_test.go b/pkg/dialect/postgres/dialect_postgres_test.go
@@ -64,4 +64,13 @@ func TestPostgres_ColumnSignatureOf(t *testing.T) {
 	f.Column.Options.Size = 65533
 	signature = d.ColumnSignatureOf(f)
 	assert.Equal(t, "TEXT PRIMARY KEY NOT NULL UNIQUE", signature)
+
+	f.Column.Options.Default = "TEST"
+	f.Column.Options.Unique = false
+	signature = d.ColumnSignatureOf(f)
+	assert.Equal(t, "TEXT PRIMARY KEY NOT NULL DEFAULT TEST", signature)
+
+	f.Column.Options = nil
+	signature = d.ColumnSignatureOf(f)
+	assert.Equal(t, "VARCHAR(255)", signature)
 }