EmployeeBuilderTest.java

package org.softlang.company.tests;

import static org.junit.Assert.assertEquals;

import org.junit.Test;
import org.softlang.company.features.CompanyBuilder;
import org.softlang.company.model.Company;
import org.softlang.company.model.Department;
import org.softlang.company.model.Employee;

public class EmployeeBuilderTest {

	private final String NAME_COMP = "abc2";
	private final String NAME_DEP = "aggs2";
	
	private final String NAME_EMPL = "Hugo";
	private final Double SALARY = 1.43;
	private final String ADDRESS = "Berlin";
	
	@Test
	public void testSingleEmployee() {
		
		Company cqCompany = new Company();
		cqCompany.setName(NAME_COMP);
		
		Department dep = createSampleDepartment();
		dep.getEmployees().add(createSampleEmployee());
		cqCompany.getDepts().add(dep);
		
		CompanyBuilder builder = new CompanyBuilder();
		Company mcCompany = builder
				.company(NAME_COMP)
					.department(NAME_DEP)
						.employee(NAME_EMPL)
							.salary(SALARY)
							.address(ADDRESS)
				.endCompany();
		assertEquals(cqCompany, mcCompany);
	}

	private Department createSampleDepartment() {
		Department dep = new Department();
		dep.setName(NAME_DEP);
		return dep;
	}

	private Employee createSampleEmployee() {
		Employee empl = new Employee();
		empl.setName(NAME_EMPL);
		empl.setAddress(ADDRESS);
		empl.setSalary(SALARY);
		return empl;
	}
	
}