test balances push, pushall

ledger/balances.go

@@ -12,17 +12,15 @@ type Balance map[Currency]float64
 func (balances Balances) PushAll(other Balances) {
 	for k, v := range other {
 		if _, ok := balances[k]; !ok {
-			balances[k] = v
+			balances[k] = make(Balance)
-		} else {
+		}
 		for k2, v2 := range v {
 			if _, ok := balances[k][k2]; !ok {
-					balances[k][k2] = v2
+				balances[k][k2] = 0
-				} else {
+			}
 			balances[k][k2] += v2
 		}
 	}
-		}
-	}
 }
 
 func (balances Balances) Push(d Delta) {

ledger/balances_test.go

@@ -16,4 +16,39 @@ func TestBalances(t *testing.T) {
 			t.Error(ba["z"])
 		}
 	})
+
+	t.Run("pushall", func(t *testing.T) {
+		a := make(Balances)
+		a.Push(Delta{Name: "a", Currency: USD, Value: 0.1})
+		a.Push(Delta{Name: "ab", Currency: USD, Value: 1.2})
+
+		b := make(Balances)
+		b.Push(Delta{Name: "b", Currency: USD, Value: 2.3})
+		b.Push(Delta{Name: "ab", Currency: USD, Value: 3.4})
+
+		b.PushAll(a)
+
+		if len(a) != 2 {
+			t.Error("modified original", len(a), a)
+		}
+		if a["a"][USD] != 0.1 {
+			t.Error("modified original a", a["a"])
+		}
+		if a["ab"][USD] != 1.2 {
+			t.Error("modified original ab", a["ab"])
+		}
+
+		if len(b) != 3 {
+			t.Error("didnt union names", len(b), b)
+		}
+		if b["a"][USD] != 0.1 {
+			t.Error("didnt pull other unique", b["a"])
+		}
+		if b["b"][USD] != 2.3 {
+			t.Error("didnt retain unique", b["b"])
+		}
+		if b["ab"][USD] != 4.6 {
+			t.Error("didnt sum other", b["ab"])
+		}
+	})
 }