diff options
Diffstat (limited to 'tests/src/se/leap/bitmaskclient/test/testProviderAPI.java')
| -rw-r--r-- | tests/src/se/leap/bitmaskclient/test/testProviderAPI.java | 219 |
1 files changed, 219 insertions, 0 deletions
diff --git a/tests/src/se/leap/bitmaskclient/test/testProviderAPI.java b/tests/src/se/leap/bitmaskclient/test/testProviderAPI.java new file mode 100644 index 00000000..0df59f9b --- /dev/null +++ b/tests/src/se/leap/bitmaskclient/test/testProviderAPI.java @@ -0,0 +1,219 @@ +/** + * + */ +package se.leap.bitmaskclient.test; + +import java.io.IOException; +import java.io.InputStreamReader; +import java.math.BigInteger; +import java.util.Map; +import java.util.Scanner; +import java.util.concurrent.CountDownLatch; + +import org.json.JSONException; +import org.json.JSONObject; + +import se.leap.bitmaskclient.ConfigHelper; +import se.leap.bitmaskclient.ConfigurationWizard; +import se.leap.bitmaskclient.Dashboard; +import se.leap.bitmaskclient.LeapSRPSession; +import se.leap.bitmaskclient.LogInDialog; +import se.leap.bitmaskclient.Provider; +import se.leap.bitmaskclient.ProviderAPI; +import android.app.Application; +import android.content.Intent; +import android.content.res.AssetManager; +import android.os.Bundle; +import android.os.ResultReceiver; +import android.test.ServiceTestCase; + +/** + * @author parmegv + * + */ +public class testProviderAPI extends ServiceTestCase<ProviderAPI> { + + /** + * @param providerAPI_class + */ + public testProviderAPI() { + super(ProviderAPI.class); + } + + /* (non-Javadoc) + * @see android.test.ServiceTestCase#setUp() + */ + protected void setUp() throws Exception { + super.setUp(); + + ConfigHelper.setSharedPreferences(getContext().getSharedPreferences(Dashboard.SHARED_PREFERENCES, Application.MODE_PRIVATE)); + } + + /* (non-Javadoc) + * @see android.test.ServiceTestCase#tearDown() + */ + protected void tearDown() throws Exception { + super.tearDown(); + } + + public void testGetHexString() throws InterruptedException { + String expected_hex_representation = "44eba0239ddfcc5a488d208df32a89eb00e93e6576b22ba2e4410085a413cf64e9c2f08ebc36a788a0761391150ad4a0507ca43f9ca659e2734f0457a85358c0bb39fa87183c9d3f9f8a3b148dab6303a4e796294f3e956472ba0e2ea5697382acd93c8b8f1b3a7a9d8517eebffd6301bfc8de7f7b701f0878a71faae1e25ad4"; + byte[] byte_representation = new BigInteger(expected_hex_representation, 16).toByteArray(); + + String hex_representation = new BigInteger(byte_representation).toString(16); + assertEquals(expected_hex_representation, hex_representation); + } + + public void testAuthenticateBySRP() throws InterruptedException, IOException, JSONException { + int number_of_trials_per_login = 5; + int trial = 0; + int total_different_logins = 5; + String username = "", password = "", api_url = ""; + AssetManager assets_manager = this.getContext().getAssets(); + String tests_string = new Scanner(new InputStreamReader(assets_manager.open("testAuthenticateBySRP_data"))).useDelimiter("\\A").next(); + JSONObject all_tests = new JSONObject(tests_string); + JSONObject current_test; + for(int current_login = 3; current_login < current_login + total_different_logins ; current_login++) + { + current_test = all_tests.getJSONObject(String.valueOf(current_login+1)); + username = current_test.getString("username"); + password = current_test.getString("password"); + api_url = current_test.getString("api-url"); + for(trial = 0; trial < number_of_trials_per_login; trial++) + { + System.out.println("Login " + current_login + ". Trial " + trial); + + CountDownLatch done_signal = new CountDownLatch(1); + JSONObject result = new JSONObject(); + new Thread(new RunAuthenticateBySRP(done_signal, result, username, password, api_url)).run(); + done_signal.await(); + assertEquals(ProviderAPI.SRP_AUTHENTICATION_SUCCESSFUL, result.getInt(ProviderAPI.RESULT_KEY)); + + result = new JSONObject(); + done_signal = new CountDownLatch(1); + new Thread(new RunLogOut(done_signal, result, api_url)).run(); + done_signal.await(); + assertEquals(ProviderAPI.LOGOUT_SUCCESSFUL, result.getInt(ProviderAPI.RESULT_KEY)); + } + } + } + + public class RunGetHexString implements Runnable { + + private CountDownLatch done_signal; + private Map<String, Integer> result; + + public RunGetHexString(CountDownLatch cdl, Map<String, Integer> result, byte[] byte_representation) { + this.done_signal = cdl; + this.result = result; + } + + @Override + public void run() { + Intent provider_API_command = new Intent(); + + Bundle parameters = new Bundle(); + parameters.putString("cert", "https://bitmask.net/ca.crt"); + parameters.putString("eip", "https://api.bitmask.net/1/config/eip-service.json"); + + provider_API_command.putExtra(ProviderAPI.DOWNLOAD_JSON_FILES_BUNDLE_EXTRA, parameters); + provider_API_command.putExtra(ProviderAPI.PARAMETERS, parameters); + ResultReceiver result_receiver = new ResultReceiver(null) { + @Override + protected void onReceiveResult(int result_code, Bundle resultData) { + result.put("result", result_code); + done_signal.countDown(); + } + }; + + provider_API_command.putExtra("receiver", result_receiver); + startService(provider_API_command); + } + } + + public class RunAuthenticateBySRP implements Runnable { + + private CountDownLatch done_signal; + private JSONObject result; + private String username; + private String password; + private String api_url; + + public RunAuthenticateBySRP(CountDownLatch cdl, JSONObject result, String username, String password, String api_url) { + this.done_signal = cdl; + this.result = result; + this.username = username; + this.password = password; + this.api_url = api_url; + } + + @Override + public void run() { + Intent provider_API_command = new Intent(); + + Bundle parameters = new Bundle(); + parameters.putString(LogInDialog.USERNAME, username); + parameters.putString(LogInDialog.PASSWORD, password); + parameters.putString(Provider.API_URL, api_url); + + provider_API_command.setAction(ProviderAPI.SRP_AUTH); + provider_API_command.putExtra(ProviderAPI.PARAMETERS, parameters); + ResultReceiver result_receiver = new ResultReceiver(null) { + @Override + protected void onReceiveResult(int result_code, Bundle resultData) { + try { + result.put("result", result_code); + result.put("session_id_cookie_key", resultData.getString("session_id_cookie_key")); + result.put("session_id", resultData.getString("session_id")); + } catch (JSONException e) { + // TODO Auto-generated catch block + e.printStackTrace(); + } + done_signal.countDown(); + } + }; + + provider_API_command.putExtra(ProviderAPI.RECEIVER_KEY, result_receiver); + startService(provider_API_command); + } + } + + public class RunLogOut implements Runnable { + + private CountDownLatch done_signal; + private JSONObject result; + private String api_url; + + public RunLogOut(CountDownLatch cdl, JSONObject result, String api_url) { + this.done_signal = cdl; + this.result = result; + this.api_url = api_url; + } + + @Override + public void run() { + Intent provider_API_command = new Intent(); + + Bundle parameters = new Bundle(); + parameters.putString(Provider.API_URL, api_url); + + provider_API_command.setAction(ProviderAPI.LOG_OUT); + provider_API_command.putExtra(ProviderAPI.PARAMETERS, parameters); + ResultReceiver result_receiver = new ResultReceiver(null) { + @Override + protected void onReceiveResult(int result_code, Bundle resultData) { + try { + result.put("result", result_code); + } catch (JSONException e) { + // TODO Auto-generated catch block + e.printStackTrace(); + } + done_signal.countDown(); + } + }; + provider_API_command.putExtra(ProviderAPI.RECEIVER_KEY, result_receiver); + + startService(provider_API_command); + } + } +} |