Add tests for writing locks

Author: sdboyer

lock_test.go

@@ -5,39 +5,46 @@
 package main
 
 import (
+	"bytes"
 	"encoding/hex"
+	"encoding/json"
 	"reflect"
 	"strings"
 	"testing"
 
 	"github.com/sdboyer/gps"
 )
 
-func TestReadLock(t *testing.T) {
-	const le = `{
-    "memo": "2252a285ab27944a4d7adcba8dbd03980f59ba652f12db39fa93b927c345593e",
-    "projects": [
-        {
-            "name": "github.com/sdboyer/gps",
-            "branch": "master",
+const le = `{
+	"memo": "2252a285ab27944a4d7adcba8dbd03980f59ba652f12db39fa93b927c345593e",
+	"projects": [
+		{
+			"name": "github.com/sdboyer/gps",
+			"branch": "master",
 			"version": "v0.12.0",
-            "revision": "d05d5aca9f895d19e9265839bffeadd74a2d2ecb",
-            "packages": ["."]
-        }
-    ]
+			"revision": "d05d5aca9f895d19e9265839bffeadd74a2d2ecb",
+			"packages": [
+				"."
+			]
+		}
+	]
 }`
-	const lg = `{
-    "memo": "2252a285ab27944a4d7adcba8dbd03980f59ba652f12db39fa93b927c345593e",
-    "projects": [
-        {
-            "name": "github.com/sdboyer/gps",
-            "branch": "master",
-            "revision": "d05d5aca9f895d19e9265839bffeadd74a2d2ecb",
-            "packages": ["."]
-        }
-    ]
+
+const lg = `{
+	"memo": "2252a285ab27944a4d7adcba8dbd03980f59ba652f12db39fa93b927c345593e",
+	"projects": [
+		{
+			"name": "github.com/sdboyer/gps",
+			"branch": "master",
+			"revision": "d05d5aca9f895d19e9265839bffeadd74a2d2ecb",
+			"packages": [
+				"."
+			]
+		}
+	]
 }`
 
+func TestReadLock(t *testing.T) {
 	_, err := readLock(strings.NewReader(le))
 	if err == nil {
 		t.Error("Reading lock with invalid props should have caused error, but did not")
@@ -66,3 +73,30 @@ func TestReadLock(t *testing.T) {
 		t.Error("Valid lock did not parse as expected")
 	}
 }
+
+func TestWriteLock(t *testing.T) {
+	memo, _ := hex.DecodeString("2252a285ab27944a4d7adcba8dbd03980f59ba652f12db39fa93b927c345593e")
+	l := &lock{
+		Memo: memo,
+		P: []gps.LockedProject{
+			gps.NewLockedProject(
+				gps.ProjectIdentifier{ProjectRoot: gps.ProjectRoot("github.com/sdboyer/gps")},
+				gps.NewBranch("master").Is(gps.Revision("d05d5aca9f895d19e9265839bffeadd74a2d2ecb")),
+				[]string{"."},
+			),
+		},
+	}
+
+	b, err := json.Marshal(l)
+	if err != nil {
+		t.Fatalf("Error while marshaling valid lock to JSON: %q", err)
+	}
+
+	var out bytes.Buffer
+	json.Indent(&out, b, "", "\t")
+
+	s := out.String()
+	if s != lg {
+		t.Errorf("Valid lock did not marshal to JSON as expected:\n\t(GOT): %s\n\t(WNT): %s", s, lg)
+	}
+}