Fix issue in unit testsfeature/SDL-0243-DisplayCap

1 files changed, 2 insertions, 15 deletions

diff --git a/android/sdl_android/src/androidTest/java/com/smartdevicelink/managers/screen/choiceset/PreloadChoicesOperationTests.java b/android/sdl_android/src/androidTest/java/com/smartdevicelink/managers/screen/choiceset/PreloadChoicesOperationTests.java
index 96671298d..b29863110 100644
--- a/android/sdl_android/src/androidTest/java/com/smartdevicelink/managers/screen/choiceset/PreloadChoicesOperationTests.java
+++ b/android/sdl_android/src/androidTest/java/com/smartdevicelink/managers/screen/choiceset/PreloadChoicesOperationTests.java
@@ -39,13 +39,13 @@ import com.smartdevicelink.AndroidTestCase2;
 import com.smartdevicelink.managers.file.FileManager;
 import com.smartdevicelink.managers.file.filetypes.SdlArtwork;
 import com.smartdevicelink.proxy.interfaces.ISdl;
-import com.smartdevicelink.proxy.rpc.DisplayCapabilities;
 import com.smartdevicelink.proxy.rpc.ImageField;
 import com.smartdevicelink.proxy.rpc.TextField;
 import com.smartdevicelink.proxy.rpc.WindowCapability;
 import com.smartdevicelink.proxy.rpc.enums.CharacterSet;
 import com.smartdevicelink.proxy.rpc.enums.FileType;
 import com.smartdevicelink.proxy.rpc.enums.ImageFieldName;
+import com.smartdevicelink.proxy.rpc.enums.ImageType;
 import com.smartdevicelink.proxy.rpc.enums.TextFieldName;
 import com.smartdevicelink.test.Test;
 
@@ -76,6 +76,7 @@ public class PreloadChoicesOperationTests extends AndroidTestCase2 {
 
 		WindowCapability windowCapability = new WindowCapability();
 		windowCapability.setImageFields(Collections.singletonList(imageField));
+		windowCapability.setImageTypeSupported(Arrays.asList(ImageType.STATIC, ImageType.DYNAMIC));
 		windowCapability.setTextFields(Collections.singletonList(textField));
 
 		ISdl internalInterface = mock(ISdl.class);
@@ -89,13 +90,6 @@ public class PreloadChoicesOperationTests extends AndroidTestCase2 {
 	}
 
 	public void testHasTextFieldOfName(){
-
-		TextField textField = new TextField();
-		textField.setName(TextFieldName.secondaryText);
-
-		DisplayCapabilities capabilities = new DisplayCapabilities();
-		capabilities.setTextFields(Collections.singletonList(textField));
-
 		boolean test = preloadChoicesOperation.hasTextFieldOfName(TextFieldName.secondaryText);
 		assertFalse(test);
 
@@ -104,13 +98,6 @@ public class PreloadChoicesOperationTests extends AndroidTestCase2 {
 	}
 
 	public void testHasImageFieldOfName(){
-
-		ImageField imageField = new ImageField();
-		imageField.setName(ImageFieldName.choiceImage);
-
-		DisplayCapabilities capabilities = new DisplayCapabilities();
-		capabilities.setImageFields(Collections.singletonList(imageField));
-
 		boolean test = preloadChoicesOperation.hasImageFieldOfName(ImageFieldName.choiceImage);
 		assertTrue(test);