tdd les go

ledger/file_test.go

@@ -6,21 +6,11 @@ import (
 )
 
 func TestFileDeltas(t *testing.T) {
-	f, err := NewFile("./testdata/happy.dat")
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	deltas, err := f.Deltas()
-	if err != nil {
-		t.Fatal(err)
-	}
-
 	d := func(s string) time.Time {
 		v, _ := time.Parse("2006-01-02", s)
 		return v
 	}
-	want := []Delta{
+	happy := []Delta{
 		{
 			Date:     d("2022-12-12"),
 			Account:  "AssetAccount:Cash:Fidelity76",
@@ -47,12 +37,33 @@ func TestFileDeltas(t *testing.T) {
 		},
 	}
 
-	if len(deltas) != len(want) {
-		t.Error(len(deltas))
+	cases := map[string][]Delta{
+		"empty": nil,
+		"one":   happy[:2],
+		"happy": happy[:],
 	}
-	for i := range want {
-		if want[i] != deltas[i] {
-			t.Errorf("[%d] \n\twant=%+v, \n\t got=%+v", i, want[i], deltas[i])
-		}
+
+	for name, d := range cases {
+		want := d
+		t.Run(name, func(t *testing.T) {
+			f, err := NewFile("./testdata/" + name + ".dat")
+			if err != nil {
+				t.Fatal(err)
+			}
+
+			deltas, err := f.Deltas()
+			if err != nil {
+				t.Fatal(err)
+			}
+
+			if len(deltas) != len(want) {
+				t.Error(len(deltas))
+			}
+			for i := range want {
+				if want[i] != deltas[i] {
+					t.Errorf("[%d] \n\twant=%+v, \n\t got=%+v", i, want[i], deltas[i])
+				}
+			}
+		})
 	}
 }