postgresql tests

pkg/spec/postgresql_test.go

@@ -1,8 +1,8 @@
 package spec
 
 import (
-	"bytes"
 	"encoding/json"
+	"errors"
 	"reflect"
 	"testing"
 	"time"
@@ -104,7 +104,31 @@ var wrongMaintenanceWindows = [][]byte{
 	[]byte(`"Mon:00:00"`),
 }
 
-var cl = []byte(`{
+var unmarshalCluster = []struct {
+	in  []byte
+	out Postgresql
+}{{
+	[]byte(`{
+  "kind": "Postgresql","apiVersion": "acid.zalan.do/v1",
+  "metadata": {"name": "acid-testcluster1"}, "spec": {"teamId": 100}}`),
+	Postgresql{
+		TypeMeta: unversioned.TypeMeta{
+			Kind:       "Postgresql",
+			APIVersion: "acid.zalan.do/v1",
+		},
+		Metadata: v1.ObjectMeta{
+			Name: "acid-testcluster1",
+		},
+		Status: ClusterStatusInvalid,
+		Error: &json.UnmarshalTypeError{
+			Value:  "number",
+			Type:   reflect.TypeOf(""),
+			Offset: 126,
+			Struct: "PostgresSpec",
+			Field:  "teamId",
+		},
+	}},
+	{[]byte(`{
   "kind": "Postgresql",
   "apiVersion": "acid.zalan.do/v1",
   "metadata": {
@@ -164,9 +188,8 @@ var cl = []byte(`{
       "Sat:00:00-Sat:04:00"
     ]
   }
-}`)
-
-var clExp = Postgresql{
+}`),
+		Postgresql{
 			TypeMeta: unversioned.TypeMeta{
 				Kind:       "Postgresql",
 				APIVersion: "acid.zalan.do/v1",
@@ -222,10 +245,31 @@ var clExp = Postgresql{
 				},
 				ClusterName: "testcluster1",
 			},
-	Status: "",
 			Error: nil,
+		}},
+	{
+		[]byte(`{"kind": "Postgresql","apiVersion": "acid.zalan.do/v1","metadata": {"name": "teapot-testcluster1"}, "spec": {"teamId": "acid"}}`),
+		Postgresql{
+			TypeMeta: unversioned.TypeMeta{
+				Kind:       "Postgresql",
+				APIVersion: "acid.zalan.do/v1",
+			},
+			Metadata: v1.ObjectMeta{
+				Name: "teapot-testcluster1",
+			},
+			Spec:   PostgresSpec{TeamID: "acid"},
+			Status: ClusterStatusInvalid,
+			Error:  errors.New("name must match {TEAM}-{NAME} format"),
+		}},
 }
 
+var invalidClusterSpec = []struct {
+	in  []byte
+	err error
+}{{[]byte(`{"kind": "Postgresql","apiVersion": "acid.zalan.do/v1"`),
+	errors.New("unexpected end of JSON input"),
+}}
+
 func mustParseTime(s string) time.Time {
 	v, err := time.Parse("15:04", s)
 	if err != nil {
@@ -323,25 +367,30 @@ func TestUnmarshalMaintWindowsErrs(t *testing.T) {
 	}
 }
 
-func TestPostgresqlUnmarshal(t *testing.T) {
+func TestPostgresUnmarshal(t *testing.T) {
+	for _, tt := range unmarshalCluster {
 		var cluster Postgresql
-	err := cluster.UnmarshalJSON(cl)
+		err := cluster.UnmarshalJSON(tt.in)
 		if err != nil {
 			t.Errorf("Unmarshal Error: %v", err)
 		}
 
-	if !reflect.DeepEqual(cluster, clExp) {
-		t.Errorf("Expected Postgresql: %#v, got %#v", clExp, cluster)
+		if !reflect.DeepEqual(cluster, tt.out) {
+			t.Errorf("Expected Postgresql: %#v, got %#v", tt.out, cluster)
+		}
 	}
 }
 
-func TestPostgresqlMarshal(t *testing.T) {
-	m, err := json.Marshal(clExp)
-	if err != nil {
-		t.Errorf("Marshal Error: %v", err)
+func TestInvalidPostgresUnmarshal(t *testing.T) {
+	for _, tt := range invalidClusterSpec {
+		var cluster Postgresql
+		err := cluster.UnmarshalJSON(tt.in)
+		if err == nil {
+			t.Errorf("Error expected for %s", string(tt.in))
 		}
 
-	if bytes.Compare(m, cl) != 0 {
-		t.Errorf("Expected postgresql marshal: %s, got %s", string(cl), string(m))
+		if err.Error() != tt.err.Error() {
+			t.Errorf("Unmarshal error expected: %v, got: %v", tt.err, err)
+		}
 	}
 }