Merge pull request go-mysql-org#790 from dveeden/test_attrs

Add test for attributes

Author: lance6716

client/conn_test.go

@@ -21,6 +21,7 @@ func (s *connTestSuite) SetUpSuite(c *C) {
 s.c, err = Connect(addr, *testUser, *testPassword, "", func(c *Conn) {
 // required for the ExecuteMultiple test
 c.SetCapability(mysql.CLIENT_MULTI_STATEMENTS)
+c.SetAttributes(map[string]string{"attrtest": "attrvalue"})
 })
 if err != nil {
 c.Fatal(err)
@@ -175,3 +176,9 @@ func (s *connTestSuite) TestExecuteSelectStreaming(c *C) {
 // Check perResultCallback call count
 c.Assert(perResultCallbackCalledTimes, Equals, 1)
 }
+
+func (s *connTestSuite) TestAttributes(c *C) {
+// Test that both custom attributes and library set attributes are visible
+c.Assert(s.c.attributes["_client_name"], Equals, "go-mysql")
+c.Assert(s.c.attributes["attrtest"], Equals, "attrvalue")
+}