|
1 | 1 | package com.codedifferently.lesson15;
|
2 | 2 |
|
3 |
| -import static org.assertj.core.api.Assertions.assertThat; |
| 3 | +import static org.junit.jupiter.api.Assertions.assertEquals; |
4 | 4 |
|
| 5 | +import org.junit.jupiter.api.BeforeEach; |
5 | 6 | import org.junit.jupiter.api.Test;
|
6 | 7 |
|
7 | 8 | class Lesson15Test {
|
8 | 9 |
|
| 10 | + private EmployeeManager employeeManager; |
| 11 | + private Employee employee; |
| 12 | + |
| 13 | + @BeforeEach |
| 14 | + public void setUp() { |
| 15 | + employeeManager = new EmployeeManager(); |
| 16 | + employee = new Employee(1, "Wall E", "Inbound", 30000.00); |
| 17 | + } |
| 18 | + |
| 19 | + @Test |
| 20 | + public void testAddEmployee() { |
| 21 | + employeeManager.addEmployee(employee); |
| 22 | + assertEquals(1, employeeManager.getEmployeeCount()); |
| 23 | + assertEquals(employee, employeeManager.getEmployee(1)); |
| 24 | + } |
| 25 | + |
| 26 | + @Test |
| 27 | + public void testGetEmployee() { |
| 28 | + employeeManager.addEmployee(employee); |
| 29 | + Employee retrieved = employeeManager.getEmployee(1); |
| 30 | + assertEquals(employee, retrieved); |
| 31 | + } |
| 32 | + |
| 33 | + @Test |
| 34 | + public void testUpdateEmployee() { |
| 35 | + employeeManager.addEmployee(employee); |
| 36 | + Employee updatedEmployee = new Employee(1, "EVE", "Inbound", 80000.00); |
| 37 | + employeeManager.updateEmployee(updatedEmployee); |
| 38 | + assertEquals(updatedEmployee, employeeManager.getEmployee(1)); |
| 39 | + } |
| 40 | + |
9 | 41 | @Test
|
10 |
| - public void testLesson15() { |
11 |
| - assertThat(new Lesson15()).isNotNull(); |
| 42 | + public void testRemoveEmployee() { |
| 43 | + employeeManager.addEmployee(employee); |
| 44 | + employeeManager.removeEmployee(1); |
| 45 | + assertEquals(0, employeeManager.getEmployeeCount()); |
12 | 46 | }
|
13 | 47 |
|
14 | 48 | @Test
|
15 |
| - public void testGetGreeting() { |
16 |
| - // Act |
17 |
| - Lesson15.main(null); |
| 49 | + public void testGetEmployeeCount() { |
| 50 | + assertEquals(0, employeeManager.getEmployeeCount()); |
| 51 | + employeeManager.addEmployee(employee); |
| 52 | + assertEquals(1, employeeManager.getEmployeeCount()); |
18 | 53 | }
|
19 | 54 | }
|
0 commit comments