Merge pull request #12 from artshmelev/kvscanner-scan-fix-bytes

[*] fix kvScanner.Scan and test on bug

Author: bgaifullin

load.go

@@ -68,7 +68,14 @@ type kvScanner struct {
 }
 
 func (kv *kvScanner) Scan(v interface{}) error {
-	kv.m[kv.column] = v
+	if b, ok := v.([]byte); ok {
+		tmp := make([]byte, len(b))
+		copy(tmp, b)
+		kv.m[kv.column] = tmp
+	} else {
+		// int64, float64, bool, string, time.Time, nil
+		kv.m[kv.column] = v
+	}
 	return nil
 }
 

load_test.go

@@ -44,6 +44,21 @@ func TestLoad(t *testing.T) {
 	}
 }
 
+func TestLoadWithBytesValue(t *testing.T) {
+	var values []driver.Value
+	columns := []string{"fieldname"}
+	value := []byte("fieldvalue")
+	session, dbmock := newSessionMock()
+	values = append(values, value)
+	rows := sqlmock.NewRows(columns).AddRow(values...)
+	dbmock.ExpectQuery("SELECT .+").WillReturnRows(rows)
+	v := reflect.New(reflect.TypeOf(map[string]interface{}(nil))).Elem().Addr().Interface()
+	session.Select(columns...).From("table").Load(v)
+	value[0] = byte('a')
+	assert.Equal(t, map[string]interface{}{"fieldname": []byte("fieldvalue")},
+		reflect.Indirect(reflect.ValueOf(v)).Interface())
+}
+
 func BenchmarkLoad(b *testing.B) {
 	session, dbmock := newSessionMock()
 	rows := sqlmock.NewRows([]string{"a", "b", "c"})