@Before
public void setup() {
user = new User("test-user", "pass");
user.setEmail("test@localhost");
user = userRepository.save(user);
organization = new Organization("test-organization", user);
organization = organizationRepository.save(organization);
category = new Category();
category.setOrganization(organization);
category.setName("Category 1");
category.setDescription("Category description");
category = categoryRepository.save(category);
topic = new Topic();
topic.setCategory(category);
topic.setName("Topic 1");
topic.setDescription("Topic description");
topic = topicRepository.save(topic);
cardDetails1 = new CardDetails();
cardDetails1.setCategory(category);
cardDetails1.setCreator(user);
cardDetails1.setText("Card 1");
cardDetails1 = cardDetailsRepository.save(cardDetails1);
cardDetails2 = new CardDetails();
cardDetails2.setCategory(category);
cardDetails2.setCreator(user);
cardDetails2.setText("Card 2");
cardDetails2 = cardDetailsRepository.save(cardDetails2);
cardDetails3 = new CardDetails();
cardDetails3.setCategory(category);
cardDetails3.setCreator(user);
cardDetails3.setText("Card 3");
cardDetails3 = cardDetailsRepository.save(cardDetails3);
cardDetails4 = new CardDetails();
cardDetails4.setCategory(category);
cardDetails4.setCreator(user);
cardDetails4.setText("Card 4");
cardDetails4 = cardDetailsRepository.save(cardDetails4);
cardDetails5 = new CardDetails();
cardDetails5.setCategory(category);
cardDetails5.setCreator(user);
cardDetails5.setText("Card 5");
cardDetails5 = cardDetailsRepository.save(cardDetails5);
category.setCards(new HashSet<>());
category.getCards().add(cardDetails1);
category.getCards().add(cardDetails2);
category.getCards().add(cardDetails3);
category.getCards().add(cardDetails4);
category.getCards().add(cardDetails5);
category = categoryRepository.save(category);
}
@Test
public void initCategory() throws Exception {
CategoryRepository categoryRepository = new JdbcCategoryRepository("client1");
List<Category> categories = categoryRepository.getCategories();
List<DBObject> list = new ArrayList<>();
for (Category category : categories) {
list.add((DBObject) JSON.parse(Json.toJson(category).toString()));
}
dataBase.getCollection(DemoCategoryRepository.DEMO_CATEGORIES_COLLECTION).insert(list);
}
@Test
public void testAddCategory() {
Category category = categoryRepository.save(new Category("test"));
assertEquals(1, categoryRepository.count());
}
/** Stores to persistence layer initial data. */
@PostConstruct
public void init() {
if (roleRepository.findByName("ROLE_ADMIN") == null) {
Role roleUser = new Role();
roleUser.setName("ROLE_USER");
roleRepository.save(roleUser);
Role roleAdmin = new Role();
roleAdmin.setName("ROLE_ADMIN");
roleRepository.save(roleAdmin);
User user = new User();
user.setEnabled(true);
user.setEmail("admin@admin");
BCryptPasswordEncoder encoder = new BCryptPasswordEncoder();
user.setPassword(encoder.encode("admin"));
List<Role> roles = new ArrayList<>();
roles.add(roleAdmin);
roles.add(roleUser);
user.setRoles(roles);
userRepository.save(user);
// Create account for wallet
Account walletAccount = new Account();
walletAccount.setName(context.getMessage("Name.default.account", null, Locale.ENGLISH));
walletAccount.setUser(user);
walletAccount.setAmount(new BigDecimal(0));
walletAccount.setCurrency(Currency.getInstance("UAH"));
accountRepository.save(walletAccount);
Account bankAccount = new Account();
bankAccount.setName("Bank");
bankAccount.setUser(user);
bankAccount.setAmount(new BigDecimal(500));
bankAccount.setCurrency(Currency.getInstance("UAH"));
accountRepository.save(bankAccount);
// Create categories for expenses
for (int i = 1; i < 6; i++) {
Category category = new Category();
category.setName(
context.getMessage("Name" + i + ".default.category", null, Locale.ENGLISH));
category.setType(Operation.EXPENSE);
category.setUser(user);
categoryRepository.save(category);
}
// Create categories for incomes
for (int i = 6; i < 8; i++) {
Category category = new Category();
category.setName(
context.getMessage("Name" + i + ".default.category", null, Locale.ENGLISH));
category.setType(Operation.INCOME);
category.setUser(user);
categoryRepository.save(category);
}
Transaction transaction1 = new Transaction();
transaction1.setDate(new Date());
transaction1.setAccount(walletAccount);
transaction1.setAmount(new BigDecimal(50));
transaction1.setCurrency(Currency.getInstance("UAH"));
transaction1.setCategory(categoryRepository.findOne(3));
transaction1.setType(Operation.EXPENSE);
transaction1.setComment("McDonalds");
transaction1.setUser(user);
transactionRepository.save(transaction1);
Transaction transaction2 = new Transaction();
Calendar calendar = new GregorianCalendar();
calendar.set(Calendar.MONTH, calendar.get(Calendar.MONTH) - 1);
transaction2.setDate(calendar.getTime());
transaction2.setAccount(bankAccount);
transaction2.setAmount(new BigDecimal(45));
transaction2.setCurrency(Currency.getInstance("UAH"));
transaction2.setCategory(categoryRepository.findOne(7));
transaction2.setType(Operation.INCOME);
transaction2.setComment("Festo");
transaction2.setUser(user);
transactionRepository.save(transaction2);
List<Transaction> transactions = new ArrayList<>();
transactions.add(transaction1);
user.setTransactions(transactions);
}
}
