test balances push

ledger/balances.go

@@ -9,6 +9,22 @@ type Balances map[string]Balance
 
 type Balance map[Currency]float64
 
+func (balances Balances) PushAll(other Balances) {
+	for k, v := range other {
+		if _, ok := balances[k]; !ok {
+			balances[k] = v
+		} else {
+			for k2, v2 := range v {
+				if _, ok := balances[k][k2]; !ok {
+					balances[k][k2] = v2
+				} else {
+					balances[k][k2] += v2
+				}
+			}
+		}
+	}
+}
+
 func (balances Balances) Push(d Delta) {
 	if _, ok := balances[d.Name]; !ok {
 		balances[d.Name] = make(Balance)