From a2eee3eceb8794e3ab7d533c5a1be85f40e10dad Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Carlos=20Nieto?= <jose.carlos@menteslibres.net>
Date: Tue, 29 Jul 2014 21:35:11 -0500
Subject: [PATCH] Adding tests for ordering by function. Will help closing #32.

---
 util/sqlgen/main_test.go | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/util/sqlgen/main_test.go b/util/sqlgen/main_test.go
index 0cfe5c80..53b8669f 100644
--- a/util/sqlgen/main_test.go
+++ b/util/sqlgen/main_test.go
@@ -419,6 +419,30 @@ func TestSelectFieldsFromWithOrderBy(t *testing.T) {
 	if s != e {
 		t.Fatalf("Got: %s, Expecting: %s", s, e)
 	}
+
+	// ORDER BY function
+	stmt = Statement{
+		Type: SqlSelect,
+		Columns: Columns{
+			{"foo"},
+			{"bar"},
+			{"baz"},
+		},
+		OrderBy: OrderBy{
+			SortColumns{
+				{Column{Raw{"FOO()"}}, SqlSortDesc},
+				{Column{Raw{"BAR()"}}, SqlSortAsc},
+			},
+		},
+		Table: Table{"table_name"},
+	}
+
+	s = trim(stmt.Compile(defaultTemplate))
+	e = `SELECT "foo", "bar", "baz" FROM "table_name" ORDER BY FOO() DESC, BAR() ASC`
+
+	if s != e {
+		t.Fatalf("Got: %s, Expecting: %s", s, e)
+	}
 }
 
 func TestSelectFieldsFromWhere(t *testing.T) {
-- 
GitLab