@@ -52,9 +52,12 @@ public class ControllerTest {
5252 private Model model ;
5353 private DateProvider mockedDateProvider ;
5454
55+ private WorkRepository mockedWorkRepository ;
56+
5557 @ Before
5658 public void beforeTest () {
57- model = new Model (Mockito .mock (ProjectRepository .class ), Mockito .mock (WorkRepository .class ),
59+ mockedWorkRepository = Mockito .mock (WorkRepository .class );
60+ model = new Model (Mockito .mock (ProjectRepository .class ), mockedWorkRepository ,
5861 Mockito .mock (SettingsRepository .class ));
5962 mockedDateProvider = Mockito .mock (DateProvider .class );
6063 testee = new Controller (model , mockedDateProvider );
@@ -348,14 +351,14 @@ public void shouldUpdateWorkItemPersistentlyWhenWorkItemIsEdited() {
348351 testee .editWork (originalWork , newWork );
349352
350353 final Work testWork = model .getPastWorkItems ().get (0 );
351- assertThat ("edited StartTime is not equal " , newWork .getStartTime (), equalTo (testWork .getStartTime ()));
352- assertThat ("edited EndTime is not equal " , newWork .getEndTime (), equalTo (testWork .getEndTime ()));
353- assertThat ("edited CreationDate is not equal " , newWork .getCreationDate (), equalTo (testWork .getCreationDate ()));
354- assertThat ("edited Notes are not equal " , newWork .getNotes (), equalTo (testWork .getNotes ()));
355- assertThat ("edited Project is not equal " , newWork .getProject (), equalTo (testWork .getProject ()));
354+ assertThat ("start time is not updated " , newWork .getStartTime (), equalTo (testWork .getStartTime ()));
355+ assertThat ("end time is not updated " , newWork .getEndTime (), equalTo (testWork .getEndTime ()));
356+ assertThat ("creation date is not updated " , newWork .getCreationDate (), equalTo (testWork .getCreationDate ()));
357+ assertThat ("notes are not updated " , newWork .getNotes (), equalTo (testWork .getNotes ()));
358+ assertThat ("project is not updated " , newWork .getProject (), equalTo (testWork .getProject ()));
356359
357360 final ArgumentCaptor <Work > argument = ArgumentCaptor .forClass (Work .class );
358- Mockito .verify (model . getWorkRepository () , Mockito .times (1 )).save (argument .capture ());
361+ Mockito .verify (mockedWorkRepository , Mockito .times (1 )).save (argument .capture ());
359362 assertThat ("not saved Persistent" , originalWork , is (argument .getValue ()));
360363
361364 }
@@ -388,8 +391,9 @@ public void shouldNotUpdateOthersWhenWorkItemIsEdited() {
388391 not (contains (newWork )));
389392
390393 final ArgumentCaptor <Work > argument = ArgumentCaptor .forClass (Work .class );
391- Mockito .verify (model .getWorkRepository (), Mockito .times (1 )).save (argument .capture ());
392- assertThat ("saved other Work persistently than what should be updated" , argument .getValue (), is (originalWork ));
394+ Mockito .verify (mockedWorkRepository , Mockito .times (1 )).save (argument .capture ());
395+ assertThat ("saved other Work persistently than what should be updated" , argument .getValue (),
396+ not (is (notToBeUpdatedWork )));
393397
394398 }
395399
0 commit comments