Allow multiple selects

Adjust tests to various cases
go-rel · Apr 18, 2024 · 76069f8 · 76069f8
1 parent 08ae153
commit 76069f8
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/builder/query_test.go b/builder/query_test.go
@@ -106,9 +106,17 @@ func TestQuery_Build(t *testing.T) {
 			query:  rel.From("users").Lock("FOR UPDATE"),
 		},
 		{
-			result: "SELECT `id`,`name` FROM `contacts`;",
+			result: "SELECT `contacts`.`id`,`contacts`.`name` FROM `contacts`;",
+			query:  rel.Select("id", "name").From("contacts"),
+		},
+		{
+			result: "SELECT `contacts`.`id`,`contacts`.`name` FROM `contacts`;",
 			query:  rel.Select("id").Select("name").From("contacts"),
 		},
+		{
+			result: "SELECT `contacts`.`id`,`contacts`.`name` FROM `contacts`;",
+			query:  rel.From("contacts").Select("id").Select("name"),
+		},
 	}
 
 	for _, test := range tests {
@@ -190,9 +198,17 @@ func TestQuery_Build_ordinal(t *testing.T) {
 			query:  rel.From("users").Lock("FOR UPDATE"),
 		},
 		{
-			result: "SELECT \"id\",\"name\" FROM \"contacts\";",
+			result: "SELECT \"contacts\".\"id\",\"contacts\".\"name\" FROM \"contacts\";",
 			query:  rel.Select("id", "name").From("contacts"),
 		},
+		{
+			result: "SELECT \"contacts\".\"id\",\"contacts\".\"name\" FROM \"contacts\";",
+			query:  rel.Select("id").Select("name").From("contacts"),
+		},
+		{
+			result: "SELECT \"contacts\".\"id\",\"contacts\".\"name\" FROM \"contacts\";",
+			query:  rel.From("contacts").Select("id").Select("name"),
+		},
 	}
 
 	for _, test := range tests {