@Test public void getRequirementsNull() { RequirementsProvider requirementsProvider = new RequirementsProvider(3, 4); RuntimeAnnotatedExecutionTrace executionTrace = Mockito.mock(RuntimeAnnotatedExecutionTrace.class); when(executionTrace.getAnnotation(Requirements.class)).thenReturn(null); Assert.assertEquals( requirementsProvider.getRequirements(executionTrace), new ActualRequirements(3, 4)); }
@Test public void getRequirementsNonNull() { RequirementsProvider requirementsProvider = new RequirementsProvider(3, 4); Requirements requirements = Mockito.mock(Requirements.class); when(requirements.cpuCores()).thenReturn(5); when(requirements.memoryGB()).thenReturn(6); RuntimeAnnotatedExecutionTrace executionTrace = Mockito.mock(RuntimeAnnotatedExecutionTrace.class); when(executionTrace.getAnnotation(Requirements.class)).thenReturn(requirements); Assert.assertEquals( requirementsProvider.getRequirements(executionTrace), new ActualRequirements(5, 6)); }