adapt tests in logistic_budget to new logic

The test failure was difficult to understand. Maybe if that happens
again I'll consider using newids instead of mocks here. The ideal would
be to find the way to decouple the tests even more, so that that
logistic_budget really tests only budgets.