|
1 | 1 | import static org.junit.jupiter.api.Assertions.assertEquals; |
| 2 | +import org.junit.jupiter.api.BeforeEach; |
| 3 | +import org.junit.jupiter.api.Test; |
2 | 4 |
|
3 | 5 | import com.codedifferently.lesson15.Employee; |
4 | 6 | import com.codedifferently.lesson15.EmployeeManager; |
5 | | -import java.util.HashMap; |
6 | | -import java.util.Map; |
7 | | -import org.junit.jupiter.api.BeforeEach; |
8 | | -import org.junit.jupiter.api.Test; |
9 | 7 |
|
10 | 8 | public class EmployeeManagerTest { |
11 | 9 | Employee employee; |
12 | 10 | EmployeeManager employeeManager; |
13 | | - Map<Integer, Employee> employees; |
14 | 11 | Employee employeeNew; |
15 | 12 |
|
16 | 13 | @BeforeEach |
17 | 14 | void setUp() { |
18 | 15 | employee = new Employee(15, "Pablo", "Emperor", 100); |
19 | 16 | employeeManager = new EmployeeManager(); |
20 | | - employees = new HashMap<>(); |
21 | 17 | } |
22 | 18 |
|
23 | 19 | @Test |
24 | 20 | void testAddEmployee() { |
25 | | - employees.put(employee.getId(), employee); |
26 | | - |
27 | 21 | employeeManager.addEmployee(employee); |
28 | 22 |
|
29 | | - assertEquals(employees.get(employee.getId()), employeeManager.getEmployee(employee.getId())); |
| 23 | + assertEquals(1, employeeManager.getEmployeeCount()); |
30 | 24 | } |
31 | 25 |
|
32 | 26 | @Test |
33 | 27 | void testGetEmployee() { |
34 | | - employees.put(employee.getId(), employee); |
| 28 | + employeeManager.addEmployee(employee); |
35 | 29 |
|
36 | | - assertEquals(employee, employees.get(employee.getId())); |
| 30 | + assertEquals(employee, employeeManager.getEmployee(employee.getId())); |
37 | 31 | } |
38 | 32 |
|
39 | 33 | @Test |
40 | 34 | void testUpdateEmployee() { |
41 | | - employees.put(employee.getId(), employee); |
42 | | - |
43 | 35 | employeeNew = new Employee(15, "Paul", "President", 200); |
44 | 36 |
|
45 | | - employees.put(employeeNew.getId(), employeeNew); |
46 | | - |
47 | 37 | employeeManager.addEmployee(employee); |
48 | 38 |
|
49 | 39 | employeeManager.updateEmployee(employeeNew); |
50 | 40 |
|
51 | | - assertEquals( |
52 | | - employeeManager.getEmployee(employeeNew.getId()), employees.get(employeeNew.getId())); |
| 41 | + assertEquals(employeeManager.getEmployee(employeeNew.getId()), employeeNew); |
53 | 42 | } |
54 | 43 |
|
55 | 44 | @Test |
56 | 45 | void testRemoveEmployee() { |
57 | | - employees.put(employee.getId(), employee); |
58 | | - |
59 | | - employees.remove(employee.getId(), employee); |
60 | | - |
61 | 46 | employeeManager.addEmployee(employee); |
62 | 47 |
|
63 | 48 | employeeManager.removeEmployee(employee.getId()); |
64 | 49 |
|
65 | 50 | assertEquals(0, employeeManager.getEmployeeCount()); |
66 | 51 | } |
67 | 52 |
|
68 | | - @Test |
69 | | - void testAssertEmployeeInCollection() throws IllegalArgumentException {} |
70 | | - |
71 | 53 | @Test |
72 | 54 | void testGetEmployeeCount() { |
73 | | - employees.put(employee.getId(), employee); |
74 | | - |
75 | 55 | employeeNew = new Employee(16, "Paul", "President", 200); |
76 | 56 |
|
77 | | - employees.put(employeeNew.getId(), employeeNew); |
78 | | - |
79 | 57 | employeeManager.addEmployee(employee); |
80 | 58 |
|
81 | 59 | employeeManager.addEmployee(employeeNew); |
|
0 commit comments