Skip to content

Commit ae685c9

Browse files
committed
Merge branch 'main' of https://github.com/angie-3/code-differently-24-q4 into angielesson15
2 parents ff83a81 + 43b0c16 commit ae685c9

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

lesson_15/tdd/tdd_app/src/main/java/com/codedifferently/lesson15/EmployeeManager.java

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,8 @@
44
import java.util.Map;
55

66
public class EmployeeManager {
7-
private Map<Integer, Employee> employees;
7+
8+
private final Map<Integer, Employee> employees;
89

910
public EmployeeManager() {
1011
employees = new HashMap<>();
@@ -15,17 +16,27 @@ public void addEmployee(Employee employee) {
1516
}
1617

1718
public Employee getEmployee(int id) {
19+
assertEmployeeInCollection(id);
1820
return employees.get(id);
1921
}
2022

2123
public void updateEmployee(Employee employee) {
24+
assertEmployeeInCollection(employee.getId());
2225
employees.put(employee.getId(), employee);
2326
}
2427

2528
public void removeEmployee(int id) {
29+
assertEmployeeInCollection(id);
2630
employees.remove(id);
2731
}
2832

33+
private void assertEmployeeInCollection(int id) {
34+
if (this.employees.containsKey(id)) {
35+
return;
36+
}
37+
throw new IllegalArgumentException("Employee does not in collection with id " + id);
38+
}
39+
2940
public int getEmployeeCount() {
3041
return employees.size();
3142
}

0 commit comments

Comments
 (0)