Function TestLineageE2E_DebitFlow_ToLineageEnabledBalance

lineage_test.go:879–955

(t *testing.T)

}

func TestLineageE2E_DebitFlow_ToLineageEnabledBalance(t *testing.T) {
	ctx := context.Background()
	mockDS := new(mocks.MockDataSource)

	aliceIdentity := &model.Identity{
		IdentityID: "idt_alice_123",
		FirstName:  "Alice",
		LastName:   "Smith",
	}

	bobIdentity := &model.Identity{
		IdentityID: "idt_bob_456",
		FirstName:  "Bob",
		LastName:   "Jones",
	}

	aliceMainBalance := &model.Balance{
		BalanceID:          "bln_alice_main",
		IdentityID:         "idt_alice_123",
		Currency:           "USD",
		Balance:            big.NewInt(10000),
		TrackFundLineage:   true,
		AllocationStrategy: "FIFO",
	}

	bobMainBalance := &model.Balance{
		BalanceID:          "bln_bob_main",
		IdentityID:         "idt_bob_456",
		Currency:           "USD",
		Balance:            big.NewInt(0),
		TrackFundLineage:   true,
		AllocationStrategy: "FIFO",
	}

	aliceStripeMapping := model.LineageMapping{
		BalanceID:          "bln_alice_main",
		Provider:           "stripe",
		ShadowBalanceID:    "bln_stripe_alice_shadow",
		AggregateBalanceID: "bln_alice_aggregate",
		CreatedAt:          time.Date(2024, 1, 1, 10, 0, 0, 0, time.UTC),
	}

	mockDS.On("GetBalanceByID", "bln_alice_main", mock.Anything, mock.Anything).Return(aliceMainBalance, nil)
	mockDS.On("GetBalanceByID", "bln_bob_main", mock.Anything, mock.Anything).Return(bobMainBalance, nil)
	mockDS.On("GetIdentityByID", "idt_alice_123").Return(aliceIdentity, nil)
	mockDS.On("GetIdentityByID", "idt_bob_456").Return(bobIdentity, nil)
	mockDS.On("GetLineageMappings", mock.Anything, "bln_alice_main").Return([]model.LineageMapping{aliceStripeMapping}, nil)
	mockDS.On("GetBalanceByIDLite", mock.Anything).Return(&model.Balance{
		DebitBalance:  big.NewInt(5000),
		CreditBalance: big.NewInt(0),
	}, nil)
	mockDS.On("GetBalanceByIndicator", mock.Anything, mock.Anything).Return(&model.Balance{
		BalanceID: "bln_new_shadow",
	}, nil)
	mockDS.On("UpsertLineageMapping", mock.Anything, mock.Anything).Return(nil)

	ledgerforgeInstance := &LedgerForge{datasource: mockDS}