unit tests for gateway manager's gateway selection

1 files changed, 150 insertions, 39 deletions

diff --git a/app/src/test/java/se/leap/bitmaskclient/eip/GatewaysManagerTest.java b/app/src/test/java/se/leap/bitmaskclient/eip/GatewaysManagerTest.java
index 12e95a47..47db3e09 100644
--- a/app/src/test/java/se/leap/bitmaskclient/eip/GatewaysManagerTest.java
+++ b/app/src/test/java/se/leap/bitmaskclient/eip/GatewaysManagerTest.java
@@ -2,6 +2,7 @@ package se.leap.bitmaskclient.eip;
 
 import android.content.Context;
 import android.content.SharedPreferences;
+import android.util.Log;
 
 import org.json.JSONException;
 import org.json.JSONObject;
@@ -10,95 +11,205 @@ import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.Answers;
 import org.mockito.Mock;
-import org.mockito.junit.MockitoJUnitRunner;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.modules.junit4.PowerMockRunner;
 
 import java.io.IOException;
 
+import de.blinkt.openvpn.VpnProfile;
+import de.blinkt.openvpn.core.ConfigParser;
 import de.blinkt.openvpn.core.connection.Connection;
-import se.leap.bitmaskclient.Constants;
 import se.leap.bitmaskclient.Provider;
+import se.leap.bitmaskclient.ProviderObservable;
+import se.leap.bitmaskclient.testutils.MockHelper;
+import se.leap.bitmaskclient.testutils.MockSharedPreferences;
 import se.leap.bitmaskclient.testutils.TestSetupHelper;
+import se.leap.bitmaskclient.utils.ConfigHelper;
+import se.leap.bitmaskclient.utils.PreferenceHelper;
 
+import static de.blinkt.openvpn.core.connection.Connection.TransportType.OBFS4;
+import static de.blinkt.openvpn.core.connection.Connection.TransportType.OPENVPN;
 import static junit.framework.Assert.assertEquals;
 import static junit.framework.Assert.assertNotNull;
 import static junit.framework.Assert.assertNull;
+import static org.junit.Assert.assertNotEquals;
+import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.anyString;
-import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
+import static org.powermock.api.mockito.PowerMockito.mockStatic;
+import static se.leap.bitmaskclient.Constants.GATEWAYS;
+import static se.leap.bitmaskclient.Constants.PROVIDER_EIP_DEFINITION;
+import static se.leap.bitmaskclient.Constants.PROVIDER_PRIVATE_KEY;
+import static se.leap.bitmaskclient.Constants.PROVIDER_VPN_CERTIFICATE;
+import static se.leap.bitmaskclient.Provider.CA_CERT;
+import static se.leap.bitmaskclient.testutils.TestSetupHelper.getProvider;
 
 /**
  * Created by cyberta on 09.10.17.
  */
-@RunWith(MockitoJUnitRunner.class)
+@RunWith(PowerMockRunner.class)
+@PrepareForTest({ProviderObservable.class, Log.class, PreferenceHelper.class, ConfigHelper.class})
 public class GatewaysManagerTest {
 
     private GatewaysManager gatewaysManager;
 
     @Mock(answer = Answers.RETURNS_DEEP_STUBS)
     private Context mockContext;
-    @Mock(answer = Answers.RETURNS_DEEP_STUBS)
+
     private SharedPreferences sharedPreferences;
+    private JSONObject secrets;
+
 
     @Before
     public void setUp() throws IOException, JSONException {
+        mockStatic(Log.class);
+        mockStatic(ConfigHelper.class);
+        when(ConfigHelper.getCurrentTimezone()).thenReturn(-1);
+        when(ConfigHelper.stringEqual(anyString(), anyString())).thenCallRealMethod();
+        secrets = new JSONObject(getJsonStringFor("secrets.json"));
+        sharedPreferences = new MockSharedPreferences();
+        sharedPreferences.edit().
+                putString(PROVIDER_PRIVATE_KEY, secrets.getString(PROVIDER_PRIVATE_KEY)).
+                putString(CA_CERT, secrets.getString(CA_CERT)).
+                putString(PROVIDER_VPN_CERTIFICATE, secrets.getString(PROVIDER_VPN_CERTIFICATE))
+                .commit();
+    }
 
+    @Test
+    public void testFromEipServiceJson_emptyJson() throws Exception {
+        gatewaysManager = new GatewaysManager(mockContext, sharedPreferences);
+        assertEquals(0, gatewaysManager.size());
+    }
 
-        JSONObject secrets = new JSONObject(getJsonStringFor("secrets.json"));
-
-        when(sharedPreferences.getString(eq(Constants.PROVIDER_PRIVATE_KEY), anyString())).thenReturn(secrets.getString(Constants.PROVIDER_PRIVATE_KEY));
-        when(sharedPreferences.getString(eq(Provider.CA_CERT), anyString())).thenReturn(secrets.getString(Provider.CA_CERT));
-        when(sharedPreferences.getString(eq(Constants.PROVIDER_VPN_CERTIFICATE), anyString())).thenReturn(secrets.getString(Constants.PROVIDER_VPN_CERTIFICATE));
-
+    @Test
+    public void testFromEipServiceJson_ignoreGatewaysWithMisconfiguredTransportsWhileAddingValidOnes() throws Exception {
+        updateEipServiceJson("ptdemo_misconfigured_mixed_gateways.json");
+        gatewaysManager = new GatewaysManager(mockContext, sharedPreferences);
+        assertEquals(1, gatewaysManager.size());
+        assertNull(gatewaysManager.select(0).getProfile(OBFS4));
+        assertNotNull(gatewaysManager.select(0).getProfile(Connection.TransportType.OPENVPN));
+    }
 
+    @Test
+    public void testClearGatewaysAndProfiles_resetGateways() throws Exception {
+        updateEipServiceJson("eip-service-two-gateways.json");
         gatewaysManager = new GatewaysManager(mockContext, sharedPreferences);
+        assertEquals(2, gatewaysManager.size());
+        gatewaysManager.clearGateways();
+        assertEquals(0, gatewaysManager.size());
     }
 
     @Test
-    public void testFromEipServiceJson_emptyJson() throws Exception {
-            gatewaysManager.fromEipServiceJson(new JSONObject());
-            assertEquals(0, gatewaysManager.size());
+    public void testGatewayManagerFromCurrentProvider_noProvider_noGateways() {
+        gatewaysManager = new GatewaysManager(mockContext);
+        assertEquals(0, gatewaysManager.size());
     }
 
     @Test
-    public void testFromEipServiceJson_ignoreDuplicateGateways_apiv3() throws Exception {
-        String eipServiceJson = TestSetupHelper.getInputAsString(getClass().getClassLoader().getResourceAsStream("ptdemo_three_mixed_gateways.json"));
-        gatewaysManager.fromEipServiceJson(new JSONObject(eipServiceJson));
-        assertEquals(3, gatewaysManager.size());
-        eipServiceJson = TestSetupHelper.getInputAsString(getClass().getClassLoader().getResourceAsStream("ptdemo.bitmask.eip-service.json"));
-        gatewaysManager.fromEipServiceJson(new JSONObject(eipServiceJson));
+    public void testGatewayManagerFromCurrentProvider_misconfiguredProvider_noGateways() throws IOException, NullPointerException {
+        Provider provider = getProvider(null, null, null, "ptdemo_misconfigured_gateway.json");
+        MockHelper.mockProviderObserver(provider);
+        gatewaysManager = new GatewaysManager(mockContext);
+        assertEquals(0, gatewaysManager.size());
+    }
+
+    @Test
+    public void testGatewayManagerFromCurrentProvider_threeGateways() {
+        Provider provider = getProvider(null, null, null, "ptdemo_three_mixed_gateways.json");
+        MockHelper.mockProviderObserver(provider);
+        gatewaysManager = new GatewaysManager(mockContext);
         assertEquals(3, gatewaysManager.size());
     }
 
     @Test
-    public void testFromEipServiceJson_ignoreGatewaysWithMisconfiguredTransportsWhileAddingValidOnes() throws Exception {
-        String eipServiceJson = TestSetupHelper.getInputAsString(getClass().getClassLoader().getResourceAsStream("ptdemo_misconfigured_mixed_gateways.json"));
-        gatewaysManager.fromEipServiceJson(new JSONObject(eipServiceJson));
-        assertEquals(1, gatewaysManager.size());
-        assertNull(gatewaysManager.select(0).getProfile(Connection.TransportType.OBFS4));
-        assertNotNull(gatewaysManager.select(0).getProfile(Connection.TransportType.OPENVPN));
+    public void TestGetPosition_VpnProfileExtistingObfs4_returnPositionZero() throws JSONException, ConfigParser.ConfigParseError, IOException {
+        Provider provider = getProvider(null, null, null, "ptdemo_three_mixed_gateways.json");
+        JSONObject eipServiceJson = provider.getEipServiceJson();
+        JSONObject gateway1 = eipServiceJson.getJSONArray(GATEWAYS).getJSONObject(0);
+        MockHelper.mockProviderObserver(provider);
+        mockStatic(PreferenceHelper.class);
+        when(PreferenceHelper.getUsePluggableTransports(any(Context.class))).thenReturn(true);
+        gatewaysManager = new GatewaysManager(mockContext);
+
+        VpnConfigGenerator configGenerator = new VpnConfigGenerator(provider.getDefinition(), secrets, gateway1, 3);
+        VpnProfile profile = configGenerator.createProfile(OBFS4);
+        profile.mGatewayIp = "37.218.247.60";
+
+        assertEquals(0, gatewaysManager.getPosition(profile));
     }
 
     @Test
-    public void testFromEipServiceJson_ignoreDuplicateGateways() throws Exception {
-        String eipServiceJson = TestSetupHelper.getInputAsString(getClass().getClassLoader().getResourceAsStream("eip-service-two-gateways.json"));
-        gatewaysManager.fromEipServiceJson(new JSONObject(eipServiceJson));
-        assertEquals(2, gatewaysManager.size());
-        eipServiceJson = TestSetupHelper.getInputAsString(getClass().getClassLoader().getResourceAsStream("eip-service-one-gateway.json"));
-        gatewaysManager.fromEipServiceJson(new JSONObject(eipServiceJson));
-        assertEquals(2, gatewaysManager.size());
+    public void TestGetPosition_VpnProfileExtistingOpenvpn_returnPositionZero() throws JSONException, ConfigParser.ConfigParseError, IOException {
+        Provider provider = getProvider(null, null, null, "ptdemo_three_mixed_gateways.json");
+        JSONObject eipServiceJson = provider.getEipServiceJson();
+        JSONObject gateway1 = eipServiceJson.getJSONArray(GATEWAYS).getJSONObject(0);
+        MockHelper.mockProviderObserver(provider);
+        mockStatic(PreferenceHelper.class);
+        when(PreferenceHelper.getUsePluggableTransports(any(Context.class))).thenReturn(false);
+        gatewaysManager = new GatewaysManager(mockContext);
+
+        VpnConfigGenerator configGenerator = new VpnConfigGenerator(provider.getDefinition(), secrets, gateway1, 3);
+        VpnProfile profile = configGenerator.createProfile(OPENVPN);
+        profile.mGatewayIp = "37.218.247.60";
+
+        assertEquals(0, gatewaysManager.getPosition(profile));
     }
 
     @Test
-    public void testClearGatewaysAndProfiles_resetGateways() throws Exception {
-        String eipServiceJson = TestSetupHelper.getInputAsString(getClass().getClassLoader().getResourceAsStream("eip-service-two-gateways.json"));
-        gatewaysManager.fromEipServiceJson(new JSONObject(eipServiceJson));
-        assertEquals(2, gatewaysManager.size());
-        gatewaysManager.clearGateways();
-        assertEquals(0, gatewaysManager.size());
+    public void TestGetPosition_VpnProfileDifferentIp_returnMinusOne() throws JSONException, ConfigParser.ConfigParseError, IOException {
+        Provider provider = getProvider(null, null, null, "ptdemo_three_mixed_gateways.json");
+        JSONObject eipServiceJson = provider.getEipServiceJson();
+        JSONObject gateway1 = eipServiceJson.getJSONArray(GATEWAYS).getJSONObject(0);
+        MockHelper.mockProviderObserver(provider);
+        mockStatic(PreferenceHelper.class);
+        when(PreferenceHelper.getUsePluggableTransports(any(Context.class))).thenReturn(true);
+        gatewaysManager = new GatewaysManager(mockContext);
+
+        VpnConfigGenerator configGenerator = new VpnConfigGenerator(provider.getDefinition(), secrets, gateway1, 3);
+        VpnProfile profile = configGenerator.createProfile(OBFS4);
+        profile.mGatewayIp = "37.218.247.61";
+
+        assertEquals(-1, gatewaysManager.getPosition(profile));
+    }
+
+    @Test
+    public void TestGetPosition_VpnProfileMoscow_returnOne() throws JSONException, ConfigParser.ConfigParseError, IOException {
+        Provider provider = getProvider(null, null, null, "ptdemo_three_mixed_gateways.json");
+        JSONObject eipServiceJson = provider.getEipServiceJson();
+        JSONObject gateway1 = eipServiceJson.getJSONArray(GATEWAYS).getJSONObject(1);
+        MockHelper.mockProviderObserver(provider);
+        mockStatic(PreferenceHelper.class);
+        when(PreferenceHelper.getUsePluggableTransports(any(Context.class))).thenReturn(true);
+        gatewaysManager = new GatewaysManager(mockContext);
+
+        VpnConfigGenerator configGenerator = new VpnConfigGenerator(provider.getDefinition(), secrets, gateway1, 3);
+        VpnProfile profile = configGenerator.createProfile(OBFS4);
+        profile.mGatewayIp = "3.21.247.89";
+
+        assertEquals(1, gatewaysManager.getPosition(profile));
+    }
+
+    @Test
+    public void TestSelectN_selectFirstObfs4Connection_returnThirdGateway() throws JSONException, ConfigParser.ConfigParseError, IOException {
+        Provider provider = getProvider(null, null, null, "ptdemo_two_openvpn_one_pt_gateways.json");
+        JSONObject eipServiceJson = provider.getEipServiceJson();
+        JSONObject gateway3 = eipServiceJson.getJSONArray(GATEWAYS).getJSONObject(2);
+
+        MockHelper.mockProviderObserver(provider);
+        mockStatic(PreferenceHelper.class);
+        when(PreferenceHelper.getUsePluggableTransports(any(Context.class))).thenReturn(true);
+        gatewaysManager = new GatewaysManager(mockContext);
+
+        assertEquals("37.12.247.10", gatewaysManager.select(0).getRemoteIP());
     }
 
     private String getJsonStringFor(String filename) throws IOException {
         return TestSetupHelper.getInputAsString(getClass().getClassLoader().getResourceAsStream(filename));
     }
 
+    private void updateEipServiceJson(String filename) throws IOException {
+        String eipServiceJson = getJsonStringFor(filename);
+        sharedPreferences.edit().putString(PROVIDER_EIP_DEFINITION, eipServiceJson).commit();
+    }
 }
\ No newline at end of file