adapted test for CachedLabelProvider

now that we use computeIfAbsent
eclipse-emf-parsley · Nov 17, 2023 · 320f523 · 320f523
1 parent 3637a8b
commit 320f523
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/...eclipse.emf.parsley.tests/src/org/eclipse/emf/parsley/tests/CachedLabelProviderTest.xtend b/...eclipse.emf.parsley.tests/src/org/eclipse/emf/parsley/tests/CachedLabelProviderTest.xtend
@@ -20,6 +20,7 @@ import org.junit.Test
 import static org.mockito.Mockito.*
 
 import static extension org.junit.Assert.*
+import org.eclipse.swt.graphics.Image
 
 class CachedLabelProviderTest extends AbstractImageBasedTest {
 
@@ -30,7 +31,9 @@ class CachedLabelProviderTest extends AbstractImageBasedTest {
 	@Before
 	def void setupLabelProvider() {
 		delegate = mock(ILabelProvider)
+		val Image mockImage = mock(Image)
 		when(delegate.getText(any)).thenReturn(TEST_VAL)
+		when(delegate.getImage(any)).thenReturn(mockImage)
 		labelProvider = new CachedLabelProvider(delegate)
 	}
 
@@ -48,8 +51,8 @@ class CachedLabelProviderTest extends AbstractImageBasedTest {
 
 	@Test
 	def void testGetImage() {
-		assertNull(labelProvider.getImage(classForControlsInstance))
-		assertNull(labelProvider.getImage(classForControlsInstance))
+		assertNotNull(labelProvider.getImage(classForControlsInstance))
+		assertNotNull(labelProvider.getImage(classForControlsInstance))
 		verify(delegate, times(1)).getImage(any)
 	}