From cc71e8f4e7da1718cd0a749f3a4e18f021c4a63d Mon Sep 17 00:00:00 2001
From: siddontang <siddontang@gmail.com>
Date: Wed, 30 Sep 2015 08:28:23 +0800
Subject: [PATCH] *: add test for enum type

---
 column/column_test.go          |  3 +++
 stmt/stmts/stmt_helper_test.go | 23 +++++++++++++++++++++++
 2 files changed, 26 insertions(+)

diff --git a/column/column_test.go b/column/column_test.go
index 5ff94d0c68..11baace5d6 100644
--- a/column/column_test.go
+++ b/column/column_test.go
@@ -52,6 +52,9 @@ func (s *testColumnSuite) TestString(c *C) {
 	col.Elems = []string{"a", "b"}
 
 	c.Assert(col.GetTypeDesc(), Equals, "enum('a','b')")
+
+	col.Elems = []string{"'a'", "b"}
+	c.Assert(col.GetTypeDesc(), Equals, "enum('''a''','b')")
 }
 
 func (s *testColumnSuite) TestFind(c *C) {
diff --git a/stmt/stmts/stmt_helper_test.go b/stmt/stmts/stmt_helper_test.go
index 1ce496120d..a9d50737d2 100644
--- a/stmt/stmts/stmt_helper_test.go
+++ b/stmt/stmts/stmt_helper_test.go
@@ -53,4 +53,27 @@ func (s *testStmtSuite) TestGetColDefaultValue(c *C) {
 
 	testSQL = " insert helper_test (c1) values (1);"
 	mustExec(c, s.testDB, testSQL)
+
+	testSQL = `drop table if exists helper_test;
+    create table helper_test (c1 enum("a"), c2 enum("b", "e") not null, c3 enum("c") default "c", c4 enum("d") default "d" not null);`
+	mustExec(c, s.testDB, testSQL)
+
+	testSQL = "insert into helper_test values();"
+	mustExec(c, s.testDB, testSQL)
+
+	row := s.testDB.QueryRow("select * from helper_test")
+	var (
+		v1 interface{}
+		v2 interface{}
+		v3 interface{}
+		v4 interface{}
+	)
+
+	err = row.Scan(&v1, &v2, &v3, &v4)
+	c.Assert(err, IsNil)
+	c.Assert(v1, IsNil)
+	c.Assert(v2, Equals, "b")
+	c.Assert(v3, Equals, "c")
+	c.Assert(v4, Equals, "d")
+
 }