D-Net

package unitest;

import eu.dnetlib.goldoa.domain.*;

import eu.dnetlib.goldoa.service.*;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.test.annotation.Rollback;

import org.springframework.test.annotation.SystemProfileValueSource;

import org.springframework.test.context.ContextConfiguration;

import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import javax.transaction.Transactional;

import java.util.ArrayList;

import java.util.List;

import static org.junit.Assert.assertEquals;

/**

 * Created by panagiotis on 23/1/2017.

 */

@RunWith( SpringJUnit4ClassRunner.class )

@ContextConfiguration(locations = "classpath:applicationContext-goldoa-service-test.xml")

public class BudgetTest {

    @Autowired

    BudgetManagerImpl bg;

    @Autowired

    UserManager um;

    @Autowired

    OrganizationManager om;

    @Test

    @Transactional

    @Rollback(true)

    public void newBudgetRequest() throws PersonManagerException, OrganizationManagerException {

        BankAccount bankAccount = new BankAccount();

        final Budget budget = new Budget();

        budget.setUser(um.getByEmail("pkanakakis@di.uoa.gr"));

        budget.setBankAccount(bankAccount);

        budget.setAmountRequested(Float.parseFloat("10000"));

        budget.getOrganizations().add(om.getById("corda::983580389"));

        bg.submitBudgetRequest(budget);

    }

    @Test

    @Transactional

    @Rollback(true)

    public void getExistingBudgetRequests() throws OrganizationManagerException {

        List<String> orgs = new ArrayList<>();

        orgs.add("corda::983580389");

        for(Budget b :bg.getBudgetsForOrganization(orgs))

            System.out.println(b.getId());

    }

    @Test

    @Transactional

    @Rollback(true)

    public void getBudgets() throws OrganizationManagerException {

         for(Budget b :bg.getBudgets())

            System.out.println(b.getId());

    }

    @Test

    @Transactional

    @Rollback(true)

    public void getBudgetsForUser() throws OrganizationManagerException {

        List<Budget> budgets = bg.getBudgetsForUser("pkanakakis@di.uoa.gr");

        for(Budget b :budgets)

            System.out.println(b.getId());

    }

    @Test

    @Transactional

    @Rollback(true)

    public void getBudgetsForOrganizations() throws OrganizationManagerException, PersonManagerException {

        User user = um.getByEmail("pkanakakis@di.uoa.gr");

        List<Budget> budgets = new ArrayList<>();

        List<String> orgs = new ArrayList<>();

        for(Affiliation a :user.getAffiliations())

            orgs.add(a.getOrganization().getId());

        budgets = bg.getBudgetsForOrganization(orgs);

        for(Budget b :budgets)

            System.out.println(b.getId());

    }

}