@Test
public void shouldBeAbleToSaveBudgets() {
Budgets savedBudgets = budgetsRepository.save(budgets1);
assertThat(savedBudgets.getId()).isNotNull();
assertThat(savedBudgets.getId()).isGreaterThan(0);
}
@Before
public void setUp() {
budget1 = BudgetForTests.createWithDefaults().setStateToCreated();
budget2 = BudgetForTests.createWithDefaults().setStateToCreated();
budgets1 = Budgets.create();
budgets1.addBudget(budget1);
budgets1.addBudget(budget2);
}
@Test
public void shouldBeAbleToFindByUsername() {
dbSeeder.seedData(budget1, budget2, budgets1);
Budgets budgets = budgetsRepository.findByUsername(Budgets.DEFAULT_USERNAME);
assertThat(budgets.getBudgets()).containsOnly(budget1, budget2);
assertBudget(budgets.getSelectedBudget(), budget1);
assertBudget(budgets.getBudgets().get(0), budget1);
assertBudget(budgets.getBudgets().get(1), budget2);
}
