package se.leap.bitmaskclient.eip;

import android.preference.PreferenceManager;
import android.text.TextUtils;
import android.util.Log;

import com.tngtech.java.junit.dataprovider.DataProvider;
import com.tngtech.java.junit.dataprovider.DataProviderRunner;
import com.tngtech.java.junit.dataprovider.UseDataProvider;

import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import org.powermock.modules.junit4.PowerMockRunnerDelegate;

import java.io.IOException;
import java.util.ArrayList;

import de.blinkt.openvpn.core.ConfigParser;
import se.leap.bitmaskclient.base.models.Provider;
import se.leap.bitmaskclient.base.utils.ConfigHelper;
import se.leap.bitmaskclient.base.utils.PreferenceHelper;
import se.leap.bitmaskclient.testutils.MockSharedPreferences;

import static junit.framework.Assert.assertEquals;
import static junit.framework.Assert.assertTrue;
import static org.mockito.ArgumentMatchers.anyInt;
import static org.powermock.api.mockito.PowerMockito.mockStatic;
import static org.powermock.api.mockito.PowerMockito.when;
import static se.leap.bitmaskclient.base.models.Constants.PROVIDER_PRIVATE_KEY;
import static se.leap.bitmaskclient.base.models.Constants.PROVIDER_VPN_CERTIFICATE;
import static se.leap.bitmaskclient.testutils.MockHelper.mockTextUtils;
import static se.leap.bitmaskclient.testutils.TestSetupHelper.getInputAsString;

/**
 * Created by cyberta on 18.12.18.
 */
@RunWith(PowerMockRunner.class)
@PowerMockRunnerDelegate(DataProviderRunner.class)
@PrepareForTest({ConfigHelper.class, TextUtils.class})
public class GatewaySelectorTest {
    public static final String TAG = GatewaySelectorTest.class.getSimpleName();

    /**
     * locations": {
     "name": ""Frankfurt"",
     "timezone": "+1"
     },
     ""Seattle, WA"__wa": {
     "name": ""Seattle, WA", WA",
     "timezone": "-7"
     },
     ""Moscow"": {
     "country_code": "RU",
     "hemisphere": "N",
     "name": ""Moscow"",
     "timezone": "+3"
     },
     ""Manila"": {
     "country_code": "PH",
     "hemisphere": "N",
     "name": ""Manila"",
     "timezone": "+8"
     }
     },
     */


    PreferenceHelper preferenceHelper;
    GatewaySelector gatewaySelector;
    JSONObject eipDefinition;
    ArrayList<Gateway> gatewayList = new ArrayList<>();

    @Before
    public void setup() throws IOException, JSONException, ConfigParser.ConfigParseError {
        mockStatic(ConfigHelper.class);
        when(ConfigHelper.timezoneDistance(anyInt(), anyInt())).thenCallRealMethod();
        mockTextUtils();
        preferenceHelper = new PreferenceHelper(new MockSharedPreferences());
        eipDefinition = new JSONObject(getInputAsString(getClass().getClassLoader().getResourceAsStream("eip-service-four-gateways.json")));
        JSONArray gateways = eipDefinition.getJSONArray("gateways");
        for (int i = 0; i < gateways.length(); i++) {
            JSONObject gw = gateways.getJSONObject(i);
            JSONObject secrets = secretsConfiguration();
            Gateway aux = new Gateway(eipDefinition, secrets, gw, null);
            gatewayList.add(aux);
        }

    }

    private JSONObject secretsConfiguration() throws IOException, JSONException {
        JSONObject result = new JSONObject();
        result.put(Provider.CA_CERT, getInputAsString(getClass().getClassLoader().getResourceAsStream("riseup.net.pem")));
        result.put(PROVIDER_PRIVATE_KEY, getInputAsString(getClass().getClassLoader().getResourceAsStream("private_rsa_key.pem")));
        result.put(PROVIDER_VPN_CERTIFICATE, getInputAsString(getClass().getClassLoader().getResourceAsStream("riseup.net.vpn_cert.pem")));
        return result;
    }

    @DataProvider
    public static Object[][] dataProviderTimezones() {
        // @formatter:off
        return new Object[][] {
                //{ -12,  "Seattle, WA" }
                { -11,  "Seattle, WA" },
                { -10, "Seattle, WA" },
                { -9, "Seattle, WA" },
                { -8, "Seattle, WA" },
                { -7, "Seattle, WA" }, // <-"Seattle, WA"
                { -6, "Seattle, WA" },
                { -5, "Seattle, WA" },
                { -4, "Seattle, WA" },
               // { -3, "Seattle, WA" },
                { -2, "Frankfurt" },
                { -1, "Frankfurt" },
                { 0, "Frankfurt" },
                { 1, "Frankfurt" }, // <- "Frankfurt"
               // { 2, "Moscow" },
                { 3, "Moscow" }, // <- "Moscow"
                { 4, "Moscow" },
                { 5, "Moscow" },
                { 6, "Manila" },
                { 7, "Manila" },
                { 8, "Manila" }, // <- "Manila"
                { 9, "Manila" },
                { 10, "Manila" },
                { 11, "Manila" },
                { 12, "Manila" }
        };
        // @formatter:on
    }

    @DataProvider
    public static Object[][] dataProviderSameDistanceTimezones() {
        // @formatter:off
        return new Object[][] {
                { -12,  "Seattle, WA", "Manila" },
                { -3, "Seattle, WA", "Frankfurt" },
                { 2, "Moscow", "Frankfurt" },

        };
        // @formatter:on
    }

    @Test
    @UseDataProvider("dataProviderTimezones")
    public void testSelect(int timezone, String expected) {
        when(ConfigHelper.getCurrentTimezone()).thenReturn(timezone);
        gatewaySelector = new GatewaySelector(gatewayList);
        assertEquals(expected, gatewaySelector.select().getName());
    }

    @Test
    @UseDataProvider("dataProviderSameDistanceTimezones")
    public void testSelectSameTimezoneDistance(int timezone, String expected1, String expected2) {
        when(ConfigHelper.getCurrentTimezone()).thenReturn(timezone);
        gatewaySelector = new GatewaySelector(gatewayList);
        assertTrue(gatewaySelector.select().getName().equals(expected1) || gatewaySelector.select().getName().equals(expected2));
    }

    @Test
    @UseDataProvider("dataProviderSameDistanceTimezones")
    public void testNClostest_SameTimezoneDistance_chooseGatewayWithSameDistance(int timezone, String expected1, String expected2) {
        when(ConfigHelper.getCurrentTimezone()).thenReturn(timezone);
        gatewaySelector = new GatewaySelector(gatewayList);
        ArrayList<String> gateways = new ArrayList<>();
        gateways.add(gatewaySelector.select(0).getName());
        gateways.add(gatewaySelector.select(1).getName());

        assertTrue(gateways.contains(expected1) && gateways.contains(expected2));

    }

    @Test
    public void testNClostest_OneTimezonePerSet_choseSecondClosestTimezone() {
        when(ConfigHelper.getCurrentTimezone()).thenReturn(-4);
        gatewaySelector = new GatewaySelector(gatewayList);

        assertTrue("Frankfurt".equals(gatewaySelector.select(1).getName()));
    }
}