dat.Null* helpers. Fixes #12

Author: GeorgeErickson

types.go

@@ -58,6 +58,31 @@ type NullBool struct {
 	sql.NullBool
 }
 
+// NullStringFrom creates a valid NullString
+func NullStringFrom(v string) NullString {
+	return NullString{sql.NullString{v, true}}
+}
+
+// NullFloat64From creates a valid NullFloat64
+func NullFloat64From(v float64) NullFloat64 {
+	return NullFloat64{sql.NullFloat64{v, true}}
+}
+
+// NullInt64From creates a valid NullInt64
+func NullInt64From(v int64) NullInt64 {
+	return NullInt64{sql.NullInt64{v, true}}
+}
+
+// NullTimeFrom creates a valid NullTime
+func NullTimeFrom(v time.Time) NullTime {
+	return NullTime{pq.NullTime{v, true}}
+}
+
+// NullBoolFrom creates a valid NullBool
+func NullBoolFrom(v bool) NullBool {
+	return NullBool{sql.NullBool{v, true}}
+}
+
 var nullString = []byte("null")
 
 // MarshalJSON correctly serializes a NullString to JSON

types_test.go

@@ -0,0 +1,48 @@
+package dat
+
+import (
+	"testing"
+	"time"
+
+	"gopkg.in/stretchr/testify.v1/assert"
+)
+
+func TestNullStringFrom(t *testing.T) {
+	v := "foo"
+	n := NullStringFrom(v)
+
+	assert.True(t, n.Valid)
+	assert.Equal(t, n.String, v)
+}
+
+func TestNullFloat64From(t *testing.T) {
+	v := 42.2
+	n := NullFloat64From(v)
+
+	assert.True(t, n.Valid)
+	assert.Equal(t, n.Float64, v)
+}
+
+func TestNullInt64From(t *testing.T) {
+	v := int64(400)
+	n := NullInt64From(v)
+
+	assert.True(t, n.Valid)
+	assert.Equal(t, n.Int64, v)
+}
+
+func TestNullTimeFrom(t *testing.T) {
+	v := time.Now()
+	n := NullTimeFrom(v)
+
+	assert.True(t, n.Valid)
+	assert.Equal(t, n.Time, v)
+}
+
+func TestNullBoolFrom(t *testing.T) {
+	v := false
+	n := NullBoolFrom(v)
+
+	assert.True(t, n.Valid)
+	assert.Equal(t, n.Bool, v)
+}