Adding test for select star.

util/sqlgen/fields.go → util/sqlgen/columns.go

@@ -9,13 +9,16 @@ type Columns struct {
 }
 
 func (self Columns) String() string {
-	out := make([]string, len(self.v))
+	if len(self.v) > 0 {
+		out := make([]string, len(self.v))
 
-	for i := range self.v {
-		out[i] = self.v[i].String()
-	}
+		for i := range self.v {
+			out[i] = self.v[i].String()
+		}
 
-	return strings.Join(out, sqlColumnComma)
+		return strings.Join(out, sqlColumnComma)
+	}
+	return ""
 }
 
 func (self Columns) Len() int {

util/sqlgen/main.go

@@ -18,7 +18,13 @@ const (
 		{{end}}
 	`
 	sqlSelectLayout = `
-		SELECT {{.Columns}}
+		SELECT
+
+			{{if .Columns}}
+				{{.Columns}}
+			{{else}}
+				*
+			{{end}}
 
 			FROM {{.Source}}
 
@@ -80,18 +86,14 @@ const (
 		DROP TABLE {{.Source}}
 	`
 
-	sqlTautology       = `1 = 1`
-	sqlAllFields       = `*`
 	sqlAndKeyword      = `AND`
 	sqlOrKeyword       = `OR`
-	sqlDefaultOperator = `=`
 	sqlDescKeyword     = `DESC`
 	sqlAscKeyword      = `ASC`
+	sqlDefaultOperator = `=`
 	sqlConditionGroup  = `({{.}})`
 
 	sqlColumnValue = `{{.Column}} {{.Operator}} {{.Value}}`
-
-	sqlFunction = `{{.Function}}({{.Value}})`
 )
 
 type Type uint

util/sqlgen/main_test.go

@@ -103,6 +103,23 @@ func TestSelectCountWhere(t *testing.T) {
 	}
 }
 
+func TestSelectStarFrom(t *testing.T) {
+	var s, e string
+	var stmt Statement
+
+	stmt = Statement{
+		Type:   SqlSelect,
+		Source: Source{"source name"},
+	}
+
+	s = trim(stmt.Compile())
+	e = `SELECT * FROM "source name"`
+
+	if s != e {
+		t.Fatalf("Got: %s, Expecting: %s", s, e)
+	}
+}
+
 func TestSelectFieldsFrom(t *testing.T) {
 	var s, e string
 	var stmt Statement