From d4a068ff06003fb8cce4b05661166f57a4d335f7 Mon Sep 17 00:00:00 2001
From: Lunny Xiao <xiaolunwen@gmail.com>
Date: Sat, 9 Sep 2017 16:52:57 +0800
Subject: [PATCH] fix bit type on postgres

---
 dialect_postgres.go   | 3 +++
 session_query_test.go | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/dialect_postgres.go b/dialect_postgres.go
index 1d4daa27..3f5c526f 100644
--- a/dialect_postgres.go
+++ b/dialect_postgres.go
@@ -781,6 +781,9 @@ func (db *postgres) SqlType(c *core.Column) string {
 	case core.TinyInt:
 		res = core.SmallInt
 		return res
+	case core.Bit:
+		res = core.Boolean
+		return res
 	case core.MediumInt, core.Int, core.Integer:
 		if c.IsAutoIncrement {
 			return core.Serial
diff --git a/session_query_test.go b/session_query_test.go
index 950ac70a..4bb4598b 100644
--- a/session_query_test.go
+++ b/session_query_test.go
@@ -65,7 +65,7 @@ func TestQueryString2(t *testing.T) {
 	assert.Equal(t, 1, len(records))
 	assert.Equal(t, 2, len(records[0]))
 	assert.Equal(t, "1", records[0]["id"])
-	assert.Equal(t, "0", records[0]["msg"])
+	assert.True(t, "0" == records[0]["msg"] || "false" == records[0]["msg"])
 }
 
 func toString(i interface{}) string {