provider) {\n assert provider != null;\n this.provider = provider;\n }\n\n @SuppressWarnings(\"unchecked\") // cast only happens when result comes from the provider\n @Override\n public T get() {\n Object result = instance;\n if (result == UNINITIALIZED) {\n synchronized (this) {\n result = instance;\n if (result == UNINITIALIZED) {\n result = provider.get();\n instance = reentrantCheck(instance, result);\n /* Null out the reference to the provider. We are never going to need it again, so we\n * can make it eligible for GC. */\n provider = null;\n }\n }\n }\n return (T) result;\n }\n\n /**\n * Checks to see if creating the new instance has resulted in a recursive call. If it has, and the\n * new instance is the same as the current instance, return the instance. However, if the new\n * instance differs from the current instance, an {@link IllegalStateException} is thrown.\n */\n public static Object reentrantCheck(Object currentInstance, Object newInstance) {\n boolean isReentrant = !(currentInstance == UNINITIALIZED);\n\n if (isReentrant && currentInstance != newInstance) {\n throw new IllegalStateException(\"Scoped provider was invoked recursively returning \"\n + \"different results: \" + currentInstance + \" & \" + newInstance + \". This is likely \"\n + \"due to a circular dependency.\");\n }\n return newInstance;\n }\n\n /** Returns a {@link Provider} that caches the value from the given delegate provider. */\n public static , T> Provider provider(P delegate) {\n if (delegate == null) {\n throw new IllegalArgumentException(\"delegate cannot be null\");\n }\n if (delegate instanceof DoubleCheck) {\n /* This should be a rare case, but if we have a scoped @Binds that delegates to a scoped\n * binding, we shouldn't cache the value again. */\n return delegate;\n }\n return new DoubleCheck(delegate);\n }\n}\n"
},
{
"path": "src/main/java/com/amplitude/util/Provider.java",
"content": "package com.amplitude.util;\n\npublic interface Provider {\n T get();\n}"
},
{
"path": "src/main/res/drawable/amp_button_bg.xml",
"content": "\n\n \n"
},
{
"path": "src/main/res/layout/amp_activity_eventexplorer_info.xml",
"content": "\n\n\n \n \n \n \n \n\n \n\n \n\n \n\n \n\n \n\n \n \n\n \n\n \n\n \n\n \n \n\n \n"
},
{
"path": "src/main/res/layout/amp_bubble_view.xml",
"content": "\n\n\n \n\n"
},
{
"path": "src/main/res/values/colors.xml",
"content": "\n\n #00000000\n #E5E5E5\n #C6D0D9\n #60758B\n #0C2A4B\n #005488\n\n"
},
{
"path": "src/main/res/values/strings.xml",
"content": "\n Copy\n Copied To Clipboard\n User Information\n DEVICE ID\n USER ID\n N/A\n\n"
},
{
"path": "src/test/java/com/amplitude/api/AmplitudeClientTest.java",
"content": "package com.amplitude.api;\n\nimport androidx.test.ext.junit.runners.AndroidJUnit4;\n\nimport org.json.JSONArray;\nimport org.json.JSONException;\nimport org.json.JSONObject;\nimport org.junit.After;\nimport org.junit.Assert;\nimport org.junit.Before;\nimport org.junit.Test;\nimport org.junit.runner.RunWith;\nimport org.mockito.Matchers;\nimport org.powermock.api.mockito.PowerMockito;\nimport org.powermock.core.classloader.annotations.PrepareForTest;\nimport org.robolectric.Robolectric;\nimport org.robolectric.Shadows;\nimport org.robolectric.annotation.Config;\nimport org.robolectric.shadows.ShadowLooper;\n\nimport java.io.IOException;\nimport java.util.Arrays;\nimport java.util.HashMap;\nimport java.util.List;\nimport java.util.Map;\nimport java.util.UUID;\n\nimport okhttp3.Call;\nimport okhttp3.OkHttpClient;\nimport okhttp3.Request;\nimport okhttp3.mockwebserver.MockResponse;\nimport okhttp3.mockwebserver.RecordedRequest;\n\nimport static java.util.concurrent.TimeUnit.SECONDS;\nimport static org.junit.Assert.assertEquals;\nimport static org.junit.Assert.assertFalse;\nimport static org.junit.Assert.assertNotEquals;\nimport static org.junit.Assert.assertNotNull;\nimport static org.junit.Assert.assertNull;\nimport static org.junit.Assert.assertTrue;\nimport static org.junit.Assert.fail;\nimport static org.robolectric.Shadows.shadowOf;\n\n@RunWith(AndroidJUnit4.class)\n@Config(manifest = Config.NONE)\npublic class AmplitudeClientTest extends BaseTest {\n\n private String generateStringWithLength(int length, char c) {\n if (length < 0) return \"\";\n char [] array = new char[length];\n Arrays.fill(array, c);\n return new String(array);\n }\n\n @Before\n public void setUp() throws Exception {\n super.setUp();\n amplitude.initialize(context, apiKey);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runOneTask();\n }\n\n @After\n public void tearDown() throws Exception {\n super.tearDown();\n }\n\n @Test\n public void testConstructor() {\n // verify that the constructor lowercases the instance name\n AmplitudeClient a = new AmplitudeClient(\"APP1\");\n AmplitudeClient b = new AmplitudeClient(\"New_App_2\");\n\n assertEquals(a.instanceName, \"app1\");\n assertEquals(b.instanceName, \"new_app_2\");\n }\n\n @Test\n public void testSetUserId() {\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n String userId = \"user_id\";\n amplitude.setUserId(userId);\n looper.runToEndOfTasks();\n assertEquals(userId, dbHelper.getValue(AmplitudeClient.USER_ID_KEY));\n assertEquals(userId, amplitude.getUserId());\n\n // try setting to null\n amplitude.setUserId(null);\n looper.runToEndOfTasks();\n assertNull(dbHelper.getValue(AmplitudeClient.USER_ID_KEY));\n assertNull(amplitude.getUserId());\n }\n\n @Test\n public void testSetUserIdTwice() {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n String userId1 = \"user_id1\";\n String userId2 = \"user_id2\";\n\n amplitude.setUserId(userId1);\n looper.runToEndOfTasks();\n assertEquals(amplitude.getUserId(), userId1);\n amplitude.logEvent(\"event1\");\n looper.runToEndOfTasks();\n\n JSONObject event1 = getLastUnsentEvent();\n assertEquals(event1.optString(\"event_type\"), \"event1\");\n assertEquals(event1.optString(\"user_id\"), userId1);\n\n amplitude.setUserId(userId2);\n looper.runToEndOfTasks();\n assertEquals(amplitude.getUserId(), userId2);\n amplitude.logEvent(\"event2\");\n looper.runToEndOfTasks();\n\n JSONObject event2 = getLastUnsentEvent();\n assertEquals(event2.optString(\"event_type\"), \"event2\");\n assertEquals(event2.optString(\"user_id\"), userId2);\n }\n\n @Test\n public void testSetDeviceId() {\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n\n String deviceId = amplitude.getDeviceId(); // Randomly generated device ID\n assertNotNull(deviceId);\n assertEquals(deviceId.length(), 36 + 1); // 36 for UUID, + 1 for appended R\n assertEquals(deviceId.charAt(36), 'R');\n assertEquals(dbHelper.getValue(amplitude.DEVICE_ID_KEY), deviceId);\n\n\n // test setting invalid device ids\n amplitude.setDeviceId(null);\n looper.runToEndOfTasks();\n assertEquals(amplitude.getDeviceId(), deviceId);\n assertEquals(dbHelper.getValue(amplitude.DEVICE_ID_KEY), deviceId);\n\n amplitude.setDeviceId(\"\");\n looper.runToEndOfTasks();\n assertEquals(amplitude.getDeviceId(), deviceId);\n assertEquals(dbHelper.getValue(amplitude.DEVICE_ID_KEY), deviceId);\n\n amplitude.setDeviceId(\"9774d56d682e549c\");\n looper.runToEndOfTasks();\n assertEquals(amplitude.getDeviceId(), deviceId);\n assertEquals(dbHelper.getValue(amplitude.DEVICE_ID_KEY), deviceId);\n\n amplitude.setDeviceId(\"unknown\");\n looper.runToEndOfTasks();\n assertEquals(amplitude.getDeviceId(), deviceId);\n assertEquals(dbHelper.getValue(amplitude.DEVICE_ID_KEY), deviceId);\n\n amplitude.setDeviceId(\"000000000000000\");\n looper.runToEndOfTasks();\n assertEquals(amplitude.getDeviceId(), deviceId);\n assertEquals(dbHelper.getValue(amplitude.DEVICE_ID_KEY), deviceId);\n\n amplitude.setDeviceId(\"Android\");\n looper.runToEndOfTasks();\n assertEquals(amplitude.getDeviceId(), deviceId);\n assertEquals(dbHelper.getValue(amplitude.DEVICE_ID_KEY), deviceId);\n\n amplitude.setDeviceId(\"DEFACE\");\n looper.runToEndOfTasks();\n assertEquals(amplitude.getDeviceId(), deviceId);\n assertEquals(dbHelper.getValue(amplitude.DEVICE_ID_KEY), deviceId);\n\n amplitude.setDeviceId(\"00000000-0000-0000-0000-000000000000\");\n assertEquals(amplitude.getDeviceId(), deviceId);\n assertEquals(dbHelper.getValue(amplitude.DEVICE_ID_KEY), deviceId);\n\n // set valid device id\n String newDeviceId = UUID.randomUUID().toString();\n amplitude.setDeviceId(newDeviceId);\n looper.runToEndOfTasks();\n assertEquals(amplitude.getDeviceId(), newDeviceId);\n assertEquals(dbHelper.getValue(amplitude.DEVICE_ID_KEY), newDeviceId);\n\n amplitude.logEvent(\"test\");\n looper.runToEndOfTasks();\n JSONObject event = getLastUnsentEvent();\n assertEquals(event.optString(\"event_type\"), \"test\");\n assertEquals(event.optString(\"device_id\"), newDeviceId);\n }\n\n @Test\n public void testSetUserProperties() throws JSONException {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n\n // setting null or empty user properties does nothing\n amplitude.setUserProperties(null);\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 0);\n amplitude.setUserProperties(new JSONObject());\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 0);\n\n JSONObject userProperties = new JSONObject().put(\"key1\", \"value1\").put(\"key2\", \"value2\");\n amplitude.setUserProperties(userProperties);\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 1);\n JSONObject event = getLastUnsentIdentify();\n assertEquals(Constants.IDENTIFY_EVENT, event.optString(\"event_type\"));\n assertTrue(Utils.compareJSONObjects(\n event.optJSONObject(\"event_properties\"), new JSONObject()\n ));\n\n JSONObject userPropertiesOperations = event.optJSONObject(\"user_properties\");\n assertEquals(userPropertiesOperations.length(), 1);\n assertTrue(userPropertiesOperations.has(Constants.AMP_OP_SET));\n\n JSONObject setOperations = userPropertiesOperations.optJSONObject(Constants.AMP_OP_SET);\n assertTrue(Utils.compareJSONObjects(userProperties, setOperations));\n }\n\n @Test\n public void testSetCustomLibrary() {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n amplitude.setLibraryName(\"amplitude-unity\");\n amplitude.setLibraryVersion(\"1.0.0\");\n amplitude.logEvent(\"test\");\n looper.runToEndOfTasks();\n\n JSONObject event = getLastEvent();\n assertNotNull(event);\n try {\n JSONObject library = event.getJSONObject(\"library\");\n String libName = library.getString(\"name\");\n String libVersion = library.getString(\"version\");\n assertEquals(libName, \"amplitude-unity\");\n assertEquals(libVersion, \"1.0.0\");\n } catch (Exception e) {\n Assert.fail();\n }\n }\n\n @Test\n public void testSetCustomLibraryWithNullValues() {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n amplitude.setLibraryName(null);\n amplitude.setLibraryVersion(null);\n amplitude.logEvent(\"test\");\n looper.runToEndOfTasks();\n\n JSONObject event = getLastEvent();\n assertNotNull(event);\n try {\n JSONObject library = event.getJSONObject(\"library\");\n String libName = library.getString(\"name\");\n String libVersion = library.getString(\"version\");\n assertEquals(libName, \"unknown-library\");\n assertEquals(libVersion, \"unknown-version\");\n } catch (Exception e) {\n Assert.fail();\n }\n }\n\n @Test\n public void testIdentifyMultipleOperations() throws JSONException {\n String property1 = \"string value\";\n String value1 = \"testValue\";\n\n String property2 = \"double value\";\n double value2 = 0.123;\n\n String property3 = \"boolean value\";\n boolean value3 = true;\n\n String property4 = \"json value\";\n\n Identify identify = new Identify().setOnce(property1, value1).add(property2, value2);\n identify.set(property3, value3).unset(property4);\n\n // identify should ignore this since duplicate key\n identify.set(property4, value3);\n\n amplitude.identify(identify);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n assertEquals(getUnsentIdentifyCount(), 1);\n assertEquals(getUnsentEventCount(), 0);\n JSONObject event = getLastUnsentIdentify();\n assertEquals(Constants.IDENTIFY_EVENT, event.optString(\"event_type\"));\n\n JSONObject userProperties = event.optJSONObject(\"user_properties\");\n JSONObject expected = new JSONObject();\n expected.put(Constants.AMP_OP_SET_ONCE, new JSONObject().put(property1, value1));\n expected.put(Constants.AMP_OP_ADD, new JSONObject().put(property2, value2));\n expected.put(Constants.AMP_OP_SET, new JSONObject().put(property3, value3));\n expected.put(Constants.AMP_OP_UNSET, new JSONObject().put(property4, \"-\"));\n assertTrue(Utils.compareJSONObjects(userProperties, expected));\n }\n\n @Test\n public void testOptOut() {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n ShadowLooper httplooper = Shadows.shadowOf(amplitude.httpThread.getLooper());\n\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n assertFalse(amplitude.isOptedOut());\n assertNull(dbHelper.getLongValue(AmplitudeClient.OPT_OUT_KEY));\n\n amplitude.setOptOut(true);\n looper.runToEndOfTasks();\n assertTrue(amplitude.isOptedOut());\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.OPT_OUT_KEY), 1L);\n RecordedRequest request = sendEvent(amplitude, \"test_opt_out\", null);\n assertNull(request);\n\n // Event shouldn't be sent event once opt out is turned off.\n amplitude.setOptOut(false);\n looper.runToEndOfTasks();\n assertFalse(amplitude.isOptedOut());\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.OPT_OUT_KEY), 0L);\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n httplooper.runToEndOfTasks();\n assertNull(request);\n\n request = sendEvent(amplitude, \"test_opt_out\", null);\n assertNotNull(request);\n }\n\n @Test\n public void testOffline() {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n ShadowLooper httplooper = Shadows.shadowOf(amplitude.httpThread.getLooper());\n\n amplitude.setOffline(true);\n RecordedRequest request = sendEvent(amplitude, \"test_offline\", null);\n assertNull(request);\n\n // Events should be sent after offline is turned off.\n amplitude.setOffline(false);\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n httplooper.runToEndOfTasks();\n\n try {\n request = server.takeRequest(1, SECONDS);\n } catch (InterruptedException e) {\n }\n assertNotNull(request);\n }\n\n @Test\n public void testLogEvent() {\n RecordedRequest request = sendEvent(amplitude, \"test_event\", null);\n assertNotNull(request);\n }\n\n @Test\n public void testIdentify() throws JSONException {\n long [] timestamps = {1000, 1001};\n clock.setTimestamps(timestamps);\n\n shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n amplitude.identify(new Identify().set(\"key\", \"value\"));\n shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n JSONObject identifyIntercepted = getLastIdentifyInterceptor();\n\n JSONObject expected = new JSONObject();\n expected.put(\"key\", \"value\");\n assertTrue(Utils.compareJSONObjects(identifyIntercepted.getJSONObject(\"user_properties\")\n .getJSONObject(Constants.AMP_OP_SET), expected));\n\n // verify db state\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n assertNull(dbHelper.getValue(AmplitudeClient.USER_ID_KEY));\n assertNull(dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY));\n assertNull(dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_ID_KEY));\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY), 1L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY), timestamps[0]);\n\n shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n assertEquals(events.length(), 1);\n JSONObject identify = events.getJSONObject(0);\n assertEquals(identify.getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(identify.getLong(\"event_id\"), 1);\n assertEquals(identify.getLong(\"timestamp\"), timestamps[0]);\n assertEquals(identify.getLong(\"sequence_number\"), 1);\n JSONObject userProperties = identify.getJSONObject(\"user_properties\");\n assertEquals(userProperties.length(), 1);\n assertTrue(userProperties.has(Constants.AMP_OP_SET));\n assertTrue(Utils.compareJSONObjects(userProperties.getJSONObject(Constants.AMP_OP_SET), expected));\n\n // verify db state\n assertNull(dbHelper.getValue(AmplitudeClient.USER_ID_KEY));\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY), 1L);\n assertNull(dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_ID_KEY));\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY), 1L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY), timestamps[0]);\n assertEquals((long)dbHelper.getLastIdentifyInterceptorId(), -1L);\n }\n\n @Test\n public void testNullIdentify() {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 0);\n\n amplitude.identify(null);\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 0);\n }\n\n @Test\n public void testLog3Events() throws InterruptedException {\n long [] timestamps = {1, 2, 3, 4, 5, 6, 7};\n clock.setTimestamps(timestamps);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n\n\n amplitude.logEvent(\"test_event1\");\n amplitude.logEvent(\"test_event2\");\n amplitude.logEvent(\"test_event3\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n assertEquals(getUnsentEventCount(), 3);\n assertEquals(getUnsentIdentifyCount(), 0);\n JSONArray events = getUnsentEvents(3);\n for (int i = 0; i < 3; i++) {\n assertEquals(events.optJSONObject(i).optString(\"event_type\"), \"test_event\" + (i+1));\n assertEquals(events.optJSONObject(i).optLong(\"timestamp\"), timestamps[i]);\n assertEquals(events.optJSONObject(i).optLong(\"sequence_number\"), i+1);\n }\n\n // send response and check that remove events works properly\n runRequest(amplitude);\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 0);\n }\n\n @Test\n public void testLog3Identifys() throws JSONException {\n long [] timestamps = {1, 2, 3, 4, 5, 6, 7};\n clock.setTimestamps(timestamps);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n amplitude.identify(new Identify().set(\"photo_count\", 1));\n amplitude.identify(new Identify().add(\"karma\", 2));\n amplitude.identify(new Identify().unset(\"gender\"));\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 3);\n JSONArray events = getUnsentIdentifys(3);\n\n JSONObject expectedIdentify1 = new JSONObject();\n expectedIdentify1.put(Constants.AMP_OP_SET, new JSONObject().put(\"photo_count\", 1));\n JSONObject expectedIdentify2 = new JSONObject();\n expectedIdentify2.put(Constants.AMP_OP_ADD, new JSONObject().put(\"karma\", 2));\n JSONObject expectedIdentify3 = new JSONObject();\n expectedIdentify3.put(Constants.AMP_OP_UNSET, new JSONObject().put(\"gender\", \"-\"));\n\n assertEquals(events.optJSONObject(0).optString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(events.optJSONObject(0).optLong(\"timestamp\"), timestamps[0]);\n assertEquals(events.optJSONObject(0).optLong(\"sequence_number\"), 1);\n assertTrue(Utils.compareJSONObjects(\n events.optJSONObject(0).optJSONObject(\"user_properties\"), expectedIdentify1\n ));\n assertEquals(events.optJSONObject(1).optString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(events.optJSONObject(1).optLong(\"timestamp\"), timestamps[1]);\n assertEquals(events.optJSONObject(1).optLong(\"sequence_number\"), 2);\n assertTrue(Utils.compareJSONObjects(\n events.optJSONObject(1).optJSONObject(\"user_properties\"), expectedIdentify2\n ));\n assertEquals(events.optJSONObject(2).optString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(events.optJSONObject(2).optLong(\"timestamp\"), timestamps[2]);\n assertEquals(events.optJSONObject(2).optLong(\"sequence_number\"), 3);\n assertTrue(Utils.compareJSONObjects(\n events.optJSONObject(2).optJSONObject(\"user_properties\"), expectedIdentify3\n ));\n\n // send response and check that remove events works properly\n runRequest(amplitude);\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 0);\n }\n\n @Test\n public void testLogEventAndIdentify() throws JSONException {\n long [] timestamps = {1, 1, 2};\n clock.setTimestamps(timestamps);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n amplitude.logEvent(\"test_event\");\n amplitude.identify(new Identify().add(\"photo_count\", 1));\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n // verify some internal counters\n assertEquals(getUnsentEventCount(), 1);\n assertEquals(amplitude.lastEventId, 1);\n assertEquals(getUnsentIdentifyCount(), 1);\n assertEquals(amplitude.lastIdentifyId, 1);\n\n JSONArray unsentEvents = getUnsentEvents(1);\n assertEquals(unsentEvents.optJSONObject(0).optString(\"event_type\"), \"test_event\");\n assertEquals(unsentEvents.optJSONObject(0).optLong(\"sequence_number\"), 1);\n\n JSONObject expectedIdentify = new JSONObject();\n expectedIdentify.put(Constants.AMP_OP_ADD, new JSONObject().put(\"photo_count\", 1));\n\n JSONArray unsentIdentifys = getUnsentIdentifys(1);\n assertEquals(unsentIdentifys.optJSONObject(0).optString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(unsentIdentifys.optJSONObject(0).optLong(\"sequence_number\"), 2);\n assertTrue(Utils.compareJSONObjects(\n unsentIdentifys.optJSONObject(0).optJSONObject(\"user_properties\"), expectedIdentify\n ));\n\n // send response and check that remove events works properly\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n assertEquals(events.length(), 2);\n assertEquals(events.optJSONObject(0).optString(\"event_type\"), \"test_event\");\n assertEquals(events.optJSONObject(1).optString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertTrue(Utils.compareJSONObjects(\n events.optJSONObject(1).optJSONObject(\"user_properties\"), expectedIdentify\n ));\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 0);\n }\n\n @Test\n public void testMergeEventsAndIdentifys() throws JSONException {\n long [] timestamps = {1, 2, 3, 4, 5, 5, 6, 7, 8, 9, 10};\n clock.setTimestamps(timestamps);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n\n amplitude.logEvent(\"test_event1\");\n amplitude.identify(new Identify().add(\"photo_count\", 1));\n amplitude.logEvent(\"test_event2\");\n amplitude.logEvent(\"test_event3\");\n amplitude.logEvent(\"test_event4\");\n amplitude.identify(new Identify().set(\"gender\", \"male\"));\n amplitude.identify(new Identify().unset(\"karma\"));\n\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n // verify some internal counters\n assertEquals(getUnsentEventCount(), 4);\n assertEquals(amplitude.lastEventId, 4);\n assertEquals(getUnsentIdentifyCount(), 3);\n assertEquals(amplitude.lastIdentifyId, 3);\n\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n assertEquals(events.length(), 7);\n\n JSONObject expectedIdentify1 = new JSONObject();\n expectedIdentify1.put(Constants.AMP_OP_ADD, new JSONObject().put(\"photo_count\", 1));\n JSONObject expectedIdentify2 = new JSONObject();\n expectedIdentify2.put(Constants.AMP_OP_SET, new JSONObject().put(\"gender\", \"male\"));\n JSONObject expectedIdentify3 = new JSONObject();\n expectedIdentify3.put(Constants.AMP_OP_UNSET, new JSONObject().put(\"karma\", \"-\"));\n\n assertEquals(events.getJSONObject(0).getString(\"event_type\"), \"test_event1\");\n assertEquals(events.getJSONObject(0).getLong(\"event_id\"), 1);\n assertEquals(events.getJSONObject(0).getLong(\"timestamp\"), timestamps[0]);\n assertEquals(events.getJSONObject(0).getLong(\"sequence_number\"), 1);\n\n assertEquals(events.getJSONObject(1).getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(events.getJSONObject(1).getLong(\"event_id\"), 1);\n assertEquals(events.getJSONObject(1).getLong(\"timestamp\"), timestamps[1]);\n assertEquals(events.getJSONObject(1).getLong(\"sequence_number\"), 2);\n assertTrue(Utils.compareJSONObjects(\n events.getJSONObject(1).getJSONObject(\"user_properties\"), expectedIdentify1\n ));\n\n assertEquals(events.getJSONObject(2).getString(\"event_type\"), \"test_event2\");\n assertEquals(events.getJSONObject(2).getLong(\"event_id\"), 2);\n assertEquals(events.getJSONObject(2).getLong(\"timestamp\"), timestamps[2]);\n assertEquals(events.getJSONObject(2).getLong(\"sequence_number\"), 3);\n\n assertEquals(events.getJSONObject(3).getString(\"event_type\"), \"test_event3\");\n assertEquals(events.getJSONObject(3).getLong(\"event_id\"), 3);\n assertEquals(events.getJSONObject(3).getLong(\"timestamp\"), timestamps[3]);\n assertEquals(events.getJSONObject(3).getLong(\"sequence_number\"), 4);\n\n // sequence number guarantees strict ordering regardless of timestamp\n assertEquals(events.getJSONObject(4).getString(\"event_type\"), \"test_event4\");\n assertEquals(events.getJSONObject(4).getLong(\"event_id\"), 4);\n assertEquals(events.getJSONObject(4).getLong(\"timestamp\"), timestamps[4]);\n assertEquals(events.getJSONObject(4).getLong(\"sequence_number\"), 5);\n\n assertEquals(events.getJSONObject(5).getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(events.getJSONObject(5).getLong(\"event_id\"), 2);\n assertEquals(events.getJSONObject(5).getLong(\"timestamp\"), timestamps[5]);\n assertEquals(events.getJSONObject(5).getLong(\"sequence_number\"), 6);\n assertTrue(Utils.compareJSONObjects(\n events.getJSONObject(5).getJSONObject(\"user_properties\"), expectedIdentify2\n ));\n\n assertEquals(events.getJSONObject(6).getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(events.getJSONObject(6).getLong(\"event_id\"), 3);\n assertEquals(events.getJSONObject(6).getLong(\"timestamp\"), timestamps[6]);\n assertEquals(events.getJSONObject(6).getLong(\"sequence_number\"), 7);\n assertTrue(Utils.compareJSONObjects(\n events.getJSONObject(6).getJSONObject(\"user_properties\"), expectedIdentify3\n ));\n\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 0);\n\n // verify db state\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n assertNull(dbHelper.getValue(AmplitudeClient.USER_ID_KEY));\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY), 3L);\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_ID_KEY), 4L);\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY), 7L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY), timestamps[6]);\n }\n\n @Test\n public void testMergeEventBackwardsCompatible() throws JSONException {\n amplitude.setEventUploadThreshold(4);\n // eventst logged before v2.1.0 won't have a sequence number, should get priority\n long [] timestamps = {1, 1, 2, 3};\n clock.setTimestamps(timestamps);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n\n amplitude.uploadingCurrently.set(true);\n amplitude.identify(new Identify().add(\"photo_count\", 1));\n amplitude.logEvent(\"test_event1\");\n amplitude.identify(new Identify().add(\"photo_count\", 2));\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n // need to delete sequence number from test event\n JSONObject event = getUnsentEvents(1).getJSONObject(0);\n assertEquals(event.getLong(\"event_id\"), 1);\n event.remove(\"sequence_number\");\n event.remove(\"event_id\");\n // delete event from db and reinsert modified event\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n dbHelper.removeEvent(1);\n dbHelper.addEvent(event.toString());\n amplitude.uploadingCurrently.set(false);\n\n // log another event to trigger upload\n amplitude.logEvent(\"test_event2\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n // verify some internal counters\n assertEquals(getUnsentEventCount(), 2);\n assertEquals(amplitude.lastEventId, 3);\n assertEquals(getUnsentIdentifyCount(), 2);\n assertEquals(amplitude.lastIdentifyId, 2);\n\n JSONObject expectedIdentify1 = new JSONObject();\n expectedIdentify1.put(Constants.AMP_OP_ADD, new JSONObject().put(\"photo_count\", 1));\n JSONObject expectedIdentify2 = new JSONObject();\n expectedIdentify2.put(Constants.AMP_OP_ADD, new JSONObject().put(\"photo_count\", 2));\n\n // send response and check that merging events correctly ordered events\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n assertEquals(events.length(), 4);\n assertEquals(events.optJSONObject(0).optString(\"event_type\"), \"test_event1\");\n assertFalse(events.optJSONObject(0).has(\"sequence_number\"));\n assertEquals(events.optJSONObject(1).optString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(events.optJSONObject(1).optLong(\"sequence_number\"), 1);\n assertTrue(Utils.compareJSONObjects(\n events.optJSONObject(1).optJSONObject(\"user_properties\"), expectedIdentify1\n ));\n assertEquals(events.optJSONObject(2).optString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(events.optJSONObject(2).optLong(\"sequence_number\"), 3);\n assertTrue(Utils.compareJSONObjects(\n events.optJSONObject(2).optJSONObject(\"user_properties\"), expectedIdentify2\n ));\n assertEquals(events.optJSONObject(3).optString(\"event_type\"), \"test_event2\");\n assertEquals(events.optJSONObject(3).optLong(\"sequence_number\"), 4);\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 0);\n }\n\n @Test\n public void testRemoveAfterSuccessfulUpload() throws JSONException {\n long [] timestamps = new long[Constants.EVENT_UPLOAD_MAX_BATCH_SIZE + 4];\n for (int i = 0; i < timestamps.length; i++) timestamps[i] = i;\n clock.setTimestamps(timestamps);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n\n for (int i = 0; i < Constants.EVENT_UPLOAD_THRESHOLD; i++) {\n amplitude.logEvent(\"test_event\" + i);\n }\n amplitude.identify(new Identify().add(\"photo_count\", 1));\n amplitude.identify(new Identify().add(\"photo_count\", 2));\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n assertEquals(getUnsentEventCount(), Constants.EVENT_UPLOAD_THRESHOLD);\n assertEquals(getUnsentIdentifyCount(), 2);\n\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n for (int i = 0; i < events.length(); i++) {\n assertEquals(events.optJSONObject(i).optString(\"event_type\"), \"test_event\" + i);\n }\n\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 2); // should have 2 identifys left\n }\n\n @Test\n public void testLogEventHasUUID() {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n amplitude.logEvent(\"test_event\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n JSONObject event = getLastUnsentEvent();\n assertTrue(event.has(\"uuid\"));\n assertNotNull(event.optString(\"uuid\"));\n assertTrue(event.optString(\"uuid\").length() > 0);\n }\n\n @Test\n public void testLogRevenue() {\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n\n JSONObject event, apiProps;\n\n amplitude.logRevenue(10.99);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n\n event = getLastUnsentEvent();\n apiProps = event.optJSONObject(\"api_properties\");\n assertEquals(Constants.AMP_REVENUE_EVENT, event.optString(\"event_type\"));\n assertEquals(Constants.AMP_REVENUE_EVENT, apiProps.optString(\"special\"));\n assertEquals(1, apiProps.optInt(\"quantity\"));\n assertNull(apiProps.optString(\"productId\", null));\n assertEquals(10.99, apiProps.optDouble(\"price\"), .01);\n assertNull(apiProps.optString(\"receipt\", null));\n assertNull(apiProps.optString(\"receiptSig\", null));\n\n amplitude.logRevenue(\"ID1\", 2, 9.99);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n\n event = getLastUnsentEvent();\n apiProps = event.optJSONObject(\"api_properties\");;\n assertEquals(Constants.AMP_REVENUE_EVENT, event.optString(\"event_type\"));\n assertEquals(Constants.AMP_REVENUE_EVENT, apiProps.optString(\"special\"));\n assertEquals(2, apiProps.optInt(\"quantity\"));\n assertEquals(\"ID1\", apiProps.optString(\"productId\"));\n assertEquals(9.99, apiProps.optDouble(\"price\"), .01);\n assertNull(apiProps.optString(\"receipt\", null));\n assertNull(apiProps.optString(\"receiptSig\", null));\n\n amplitude.logRevenue(\"ID2\", 3, 8.99, \"RECEIPT\", \"SIG\");\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n\n event = getLastUnsentEvent();\n apiProps = event.optJSONObject(\"api_properties\");\n assertEquals(Constants.AMP_REVENUE_EVENT, event.optString(\"event_type\"));\n assertEquals(Constants.AMP_REVENUE_EVENT, apiProps.optString(\"special\"));\n assertEquals(3, apiProps.optInt(\"quantity\"));\n assertEquals(\"ID2\", apiProps.optString(\"productId\"));\n assertEquals(8.99, apiProps.optDouble(\"price\"), .01);\n assertEquals(\"RECEIPT\", apiProps.optString(\"receipt\"));\n assertEquals(\"SIG\", apiProps.optString(\"receiptSig\"));\n\n assertNotNull(runRequest(amplitude));\n }\n\n @Test\n public void testLogRevenueV2() throws JSONException {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n\n // ignore invalid revenue objects\n amplitude.logRevenueV2(null);\n looper.runToEndOfTasks();\n amplitude.logRevenueV2(new Revenue());\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n\n // log valid revenue object\n double price = 10.99;\n int quantity = 15;\n String productId = \"testProductId\";\n String receipt = \"testReceipt\";\n String receiptSig = \"testReceiptSig\";\n String revenueType = \"testRevenueType\";\n JSONObject props = new JSONObject().put(\"city\", \"Boston\");\n\n Revenue revenue = new Revenue().setProductId(productId).setPrice(price);\n revenue.setQuantity(quantity).setReceipt(receipt, receiptSig);\n revenue.setRevenueType(revenueType).setRevenueProperties(props);\n\n amplitude.logRevenueV2(revenue);\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 1);\n\n JSONObject event = getLastUnsentEvent();\n assertEquals(event.optString(\"event_type\"), \"revenue_amount\");\n\n JSONObject obj = event.optJSONObject(\"event_properties\");\n assertEquals(obj.optDouble(\"$price\"), price, 0);\n assertEquals(obj.optInt(\"$quantity\"), 15);\n assertEquals(obj.optString(\"$productId\"), productId);\n assertEquals(obj.optString(\"$receipt\"), receipt);\n assertEquals(obj.optString(\"$receiptSig\"), receiptSig);\n assertEquals(obj.optString(\"$revenueType\"), revenueType);\n assertEquals(obj.optString(\"city\"), \"Boston\");\n\n // user properties should be empty\n assertTrue(Utils.compareJSONObjects(\n event.optJSONObject(\"user_properties\"), new JSONObject()\n ));\n\n // api properties should not have any revenue info\n JSONObject apiProps = event.optJSONObject(\"api_properties\");\n assertTrue(apiProps.length() > 0);\n assertFalse(apiProps.has(\"special\"));\n assertFalse(apiProps.has(\"productId\"));\n assertFalse(apiProps.has(\"quantity\"));\n assertFalse(apiProps.has(\"price\"));\n assertFalse(apiProps.has(\"receipt\"));\n assertFalse(apiProps.has(\"receiptSig\"));\n }\n\n @Test\n public void testLogEventSync() {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n\n amplitude.logEventSync(\"test_event_sync\", null);\n\n // Event should be in the database synchronously.\n JSONObject event = getLastEvent();\n assertEquals(\"test_event_sync\", event.optString(\"event_type\"));\n\n looper.runToEndOfTasks();\n\n server.enqueue(new MockResponse().setBody(\"success\"));\n ShadowLooper httplooper = Shadows.shadowOf(amplitude.httpThread.getLooper());\n httplooper.runToEndOfTasks();\n\n try {\n assertNotNull(server.takeRequest(1, SECONDS));\n } catch (InterruptedException e) {\n fail(e.toString());\n }\n }\n\n /**\n * Test for not excepting on empty event properties.\n * See https://github.com/amplitude/Amplitude-Android/issues/35\n */\n @Test\n public void testEmptyEventProps() {\n RecordedRequest request = sendEvent(amplitude, \"test_event\", new JSONObject());\n assertNotNull(request);\n }\n\n /**\n * Test that resend failed events only occurs every 30 events.\n */\n @Test\n public void testSaveEventLogic() {\n amplitude.trackSessionEvents(true);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n\n for (int i = 0; i < Constants.EVENT_UPLOAD_THRESHOLD; i++) {\n amplitude.logEvent(\"test\");\n }\n looper.runToEndOfTasks();\n // unsent events will be threshold (+1 for start session)\n assertEquals(getUnsentEventCount(), Constants.EVENT_UPLOAD_THRESHOLD + 1);\n\n server.enqueue(new MockResponse().setResponseCode(400).setBody(\"invalid_api_key\"));\n server.enqueue(new MockResponse().setResponseCode(400).setBody(\"bad_checksum\"));\n ShadowLooper httpLooper = Shadows.shadowOf(amplitude.httpThread.getLooper());\n httpLooper.runToEndOfTasks();\n\n // no events sent, queue should be same size\n assertEquals(getUnsentEventCount(), Constants.EVENT_UPLOAD_THRESHOLD + 1);\n\n for (int i = 0; i < Constants.EVENT_UPLOAD_THRESHOLD; i++) {\n amplitude.logEvent(\"test\");\n }\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), Constants.EVENT_UPLOAD_THRESHOLD * 2 + 1);\n httpLooper.runToEndOfTasks();\n\n // sent 61 events, should have only made 2 requests\n assertEquals(server.getRequestCount(), 2);\n }\n\n @Test\n public void testRequestTooLargeBackoffLogic() {\n amplitude.trackSessionEvents(true);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n // verify event queue empty\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n\n // 413 error force backoff with 2 events --> new upload limit will be 1\n amplitude.logEvent(\"test\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 2); // 2 events: start session + test\n server.enqueue(new MockResponse().setResponseCode(413));\n ShadowLooper httpLooper = Shadows.shadowOf(amplitude.httpThread.getLooper());\n httpLooper.runToEndOfTasks();\n\n // 413 error with upload limit 1 will remove the top (start session) event\n amplitude.logEvent(\"test\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 3);\n server.enqueue(new MockResponse().setResponseCode(413));\n httpLooper.runToEndOfTasks();\n\n // verify only start session event removed\n assertEquals(getUnsentEventCount(), 2);\n JSONArray events = getUnsentEvents(2);\n assertEquals(events.optJSONObject(0).optString(\"event_type\"), \"test\");\n assertEquals(events.optJSONObject(1).optString(\"event_type\"), \"test\");\n\n // upload limit persists until event count below threshold\n server.enqueue(new MockResponse().setBody(\"success\"));\n looper.runToEndOfTasks(); // retry uploading after removing large event\n httpLooper.runToEndOfTasks(); // send success --> 1 event sent\n looper.runToEndOfTasks(); // event count below threshold --> disable backoff\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 1);\n\n // verify backoff disabled - queue 2 more events, see that all get uploaded\n amplitude.logEvent(\"test\");\n amplitude.logEvent(\"test\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 3);\n server.enqueue(new MockResponse().setBody(\"success\"));\n httpLooper.runToEndOfTasks();\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n }\n\n @Test\n public void testUploadRemainingEvents() {\n long [] timestamps = {1, 2, 3, 4, 5, 6, 7};\n clock.setTimestamps(timestamps);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 0);\n\n amplitude.setEventUploadMaxBatchSize(2);\n amplitude.setEventUploadThreshold(2);\n amplitude.uploadingCurrently.set(true); // block uploading until we queue up enough events\n for (int i = 0; i < 6; i++) {\n amplitude.logEvent(String.format(\"test%d\", i));\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(dbHelper.getTotalEventCount(), i+1);\n }\n amplitude.uploadingCurrently.set(false);\n\n // allow event uploads\n // 7 events in queue, should upload 2, and then 2, and then 2, and then 2\n amplitude.logEvent(\"test7\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(dbHelper.getEventCount(), 7);\n assertEquals(dbHelper.getIdentifyCount(), 0);\n assertEquals(dbHelper.getTotalEventCount(), 7);\n\n // server response\n server.enqueue(new MockResponse().setBody(\"success\"));\n ShadowLooper httpLooper = Shadows.shadowOf(amplitude.httpThread.getLooper());\n httpLooper.runToEndOfTasks();\n\n // when receive success response, continue uploading\n looper.runToEndOfTasks();\n looper.runToEndOfTasks(); // remove uploaded events\n assertEquals(dbHelper.getEventCount(), 5);\n assertEquals(dbHelper.getIdentifyCount(), 0);\n assertEquals(dbHelper.getTotalEventCount(), 5);\n\n // 2nd server response\n server.enqueue(new MockResponse().setBody(\"success\"));\n httpLooper.runToEndOfTasks();\n looper.runToEndOfTasks(); // remove uploaded events\n assertEquals(dbHelper.getEventCount(), 3);\n assertEquals(dbHelper.getIdentifyCount(), 0);\n assertEquals(dbHelper.getTotalEventCount(), 3);\n\n // 3rd server response\n server.enqueue(new MockResponse().setBody(\"success\"));\n httpLooper.runToEndOfTasks();\n looper.runToEndOfTasks(); // remove uploaded events\n looper.runToEndOfTasks();\n assertEquals(dbHelper.getEventCount(), 1);\n assertEquals(dbHelper.getIdentifyCount(), 0);\n assertEquals(dbHelper.getTotalEventCount(), 1);\n }\n\n @Test\n public void testBackoffRemoveIdentify() {\n long [] timestamps = {1, 1, 2, 3, 4, 5};\n clock.setTimestamps(timestamps);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 0);\n\n // 413 error force backoff with 2 events --> new upload limit will be 1\n amplitude.identify(new Identify().add(\"photo_count\", 1));\n amplitude.logEvent(\"test1\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n assertEquals(getUnsentIdentifyCount(), 1);\n assertEquals(getUnsentEventCount(), 1);\n\n server.enqueue(new MockResponse().setResponseCode(413));\n ShadowLooper httpLooper = Shadows.shadowOf(amplitude.httpThread.getLooper());\n httpLooper.runToEndOfTasks();\n\n // 413 error with upload limit 1 will remove the top identify\n amplitude.logEvent(\"test2\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 2);\n assertEquals(getUnsentIdentifyCount(), 1);\n server.enqueue(new MockResponse().setResponseCode(413));\n httpLooper.runToEndOfTasks();\n\n // verify only identify removed\n assertEquals(getUnsentEventCount(), 2);\n assertEquals(getUnsentIdentifyCount(), 0);\n JSONArray events = getUnsentEvents(2);\n assertEquals(events.optJSONObject(0).optString(\"event_type\"), \"test1\");\n assertEquals(events.optJSONObject(1).optString(\"event_type\"), \"test2\");\n }\n\n @Test\n public void testLimitTrackingEnabled() {\n amplitude.logEvent(\"test\");\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n JSONObject apiProperties = getLastUnsentEvent().optJSONObject(\"api_properties\");\n assertTrue(apiProperties.has(\"limit_ad_tracking\"));\n assertFalse(apiProperties.optBoolean(\"limit_ad_tracking\"));\n assertFalse(apiProperties.has(\"androidADID\"));\n }\n\n @Test\n public void testTruncateString() {\n String longString = generateStringWithLength(Constants.MAX_STRING_LENGTH * 2, 'c');\n assertEquals(longString.length(), Constants.MAX_STRING_LENGTH * 2);\n String truncatedString = amplitude.truncate(longString);\n assertEquals(truncatedString.length(), Constants.MAX_STRING_LENGTH);\n assertEquals(truncatedString, generateStringWithLength(Constants.MAX_STRING_LENGTH, 'c'));\n }\n\n @Test\n public void testTruncateJSONObject() throws JSONException {\n String longString = generateStringWithLength(Constants.MAX_STRING_LENGTH * 2, 'c');\n String truncString = generateStringWithLength(Constants.MAX_STRING_LENGTH, 'c');\n JSONObject object = new JSONObject();\n object.put(\"int value\", 10);\n object.put(\"bool value\", false);\n object.put(\"long string\", longString);\n object.put(\"array\", new JSONArray().put(longString).put(10));\n object.put(\"jsonobject\", new JSONObject().put(\"long string\", longString));\n object.put(Constants.AMP_REVENUE_RECEIPT, longString);\n object.put(Constants.AMP_REVENUE_RECEIPT_SIG, longString);\n\n object = amplitude.truncate(object);\n assertEquals(object.optInt(\"int value\"), 10);\n assertEquals(object.optBoolean(\"bool value\"), false);\n assertEquals(object.optString(\"long string\"), truncString);\n assertEquals(object.optJSONArray(\"array\").length(), 2);\n assertEquals(object.optJSONArray(\"array\").getString(0), truncString);\n assertEquals(object.optJSONArray(\"array\").getInt(1), 10);\n assertEquals(object.optJSONObject(\"jsonobject\").length(), 1);\n assertEquals(object.optJSONObject(\"jsonobject\").optString(\"long string\"), truncString);\n\n // receipt and receipt sig should not be truncated\n assertEquals(object.optString(Constants.AMP_REVENUE_RECEIPT), longString);\n assertEquals(object.optString(Constants.AMP_REVENUE_RECEIPT_SIG), longString);\n }\n\n @Test\n public void testTruncateNullJSONObject() throws JSONException {\n assertTrue(Utils.compareJSONObjects(\n amplitude.truncate((JSONObject) null), new JSONObject()\n ));\n assertEquals(amplitude.truncate((JSONArray) null).length(), 0);\n }\n\n @Test\n public void testTruncateEventAndIdentify() throws JSONException {\n String longString = generateStringWithLength(Constants.MAX_STRING_LENGTH * 2, 'c');\n String truncString = generateStringWithLength(Constants.MAX_STRING_LENGTH, 'c');\n\n long [] timestamps = {1, 1, 2, 3};\n clock.setTimestamps(timestamps);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n amplitude.identify(new Identify().set(\"long_string\", longString));\n amplitude.logEvent(\"test\", new JSONObject().put(\"long_string\", longString));\n\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n assertEquals(events.optJSONObject(0).optString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertTrue(Utils.compareJSONObjects(\n events.optJSONObject(0).optJSONObject(\"user_properties\").optJSONObject(Constants.AMP_OP_SET),\n new JSONObject().put(\"long_string\", truncString)\n ));\n assertEquals(events.optJSONObject(1).optString(\"event_type\"), \"test\");\n assertTrue(Utils.compareJSONObjects(\n events.optJSONObject(1).optJSONObject(\"event_properties\"),\n new JSONObject().put(\"long_string\", truncString)\n ));\n }\n\n @Test\n public void testAutoIncrementSequenceNumber() {\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n int limit = 10;\n for (int i = 0; i < limit; i++) {\n assertEquals(amplitude.getNextSequenceNumber(), i+1);\n assertEquals(dbHelper.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY), Long.valueOf(i+1));\n }\n }\n\n @Test\n public void testSetOffline() throws JSONException {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n amplitude.setOffline(true);\n\n amplitude.logEvent(\"test1\");\n amplitude.logEvent(\"test2\");\n amplitude.identify(new Identify().unset(\"key1\"));\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 2);\n assertEquals(getUnsentIdentifyCount(), 1);\n\n amplitude.setOffline(false);\n looper.runToEndOfTasks();\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n looper.runToEndOfTasks();\n\n assertEquals(events.length(), 3);\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 0);\n }\n\n @Test\n public void testSetOfflineTruncate() throws JSONException {\n long [] timestamps = {1, 2, 3, 4, 5, 6, 7, 8, 9};\n clock.setTimestamps(timestamps);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n int eventMaxCount = 3;\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n amplitude.setEventMaxCount(eventMaxCount).setOffline(true);\n\n amplitude.logEvent(\"test1\");\n amplitude.logEvent(\"test2\");\n amplitude.logEvent(\"test3\");\n amplitude.identify(new Identify().unset(\"key1\"));\n amplitude.identify(new Identify().unset(\"key2\"));\n amplitude.identify(new Identify().unset(\"key3\"));\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), eventMaxCount);\n assertEquals(getUnsentIdentifyCount(), eventMaxCount);\n\n amplitude.logEvent(\"test4\");\n amplitude.identify(new Identify().unset(\"key4\"));\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), eventMaxCount);\n assertEquals(getUnsentIdentifyCount(), eventMaxCount);\n\n List events = dbHelper.getEvents(-1, -1);\n assertEquals(events.size(), eventMaxCount);\n assertEquals(events.get(0).optString(\"event_type\"), \"test2\");\n assertEquals(events.get(1).optString(\"event_type\"), \"test3\");\n assertEquals(events.get(2).optString(\"event_type\"), \"test4\");\n\n List identifys = dbHelper.getIdentifys(-1, -1);\n assertEquals(identifys.size(), eventMaxCount);\n assertEquals(identifys.get(0).optJSONObject(\"user_properties\").optJSONObject(\"$unset\").optString(\"key2\"), \"-\");\n assertEquals(identifys.get(1).optJSONObject(\"user_properties\").optJSONObject(\"$unset\").optString(\"key3\"), \"-\");\n assertEquals(identifys.get(2).optJSONObject(\"user_properties\").optJSONObject(\"$unset\").optString(\"key4\"), \"-\");\n }\n\n @Test\n public void testTruncateEventsQueues() {\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n int eventMaxCount = 50;\n assertTrue(eventMaxCount > Constants.EVENT_REMOVE_BATCH_SIZE);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n amplitude.setEventMaxCount(eventMaxCount).setOffline(true);\n\n for (int i = 0; i < eventMaxCount; i++) {\n amplitude.logEvent(\"test\");\n }\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), eventMaxCount);\n\n amplitude.logEvent(\"test\");\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), eventMaxCount - (eventMaxCount/10) + 1);\n }\n\n @Test\n public void testTruncateEventsQueuesWithOneEvent() {\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n int eventMaxCount = 1;\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n amplitude.setEventMaxCount(eventMaxCount).setOffline(true);\n\n amplitude.logEvent(\"test1\");\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), eventMaxCount);\n\n amplitude.logEvent(\"test2\");\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), eventMaxCount);\n\n JSONObject event = getLastUnsentEvent();\n assertEquals(event.optString(\"event_type\"), \"test2\");\n }\n\n @Test\n public void testClearUserProperties() throws JSONException {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n\n amplitude.clearUserProperties();\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 1);\n JSONObject event = getLastUnsentIdentify();\n assertEquals(Constants.IDENTIFY_EVENT, event.optString(\"event_type\"));\n assertTrue(Utils.compareJSONObjects(\n event.optJSONObject(\"event_properties\"), new JSONObject()\n ));\n\n JSONObject userPropertiesOperations = event.optJSONObject(\"user_properties\");\n assertEquals(userPropertiesOperations.length(), 1);\n assertTrue(userPropertiesOperations.has(Constants.AMP_OP_CLEAR_ALL));\n\n assertEquals(\n \"-\", userPropertiesOperations.optString(Constants.AMP_OP_CLEAR_ALL)\n );\n }\n\n @Test\n public void testSetGroup() throws JSONException {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n\n amplitude.setGroup(\"orgId\", new JSONArray().put(10).put(15));\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 1);\n JSONObject event = getLastUnsentIdentify();\n assertEquals(Constants.IDENTIFY_EVENT, event.optString(\"event_type\"));\n assertTrue(Utils.compareJSONObjects(\n event.optJSONObject(\"event_properties\"), new JSONObject()\n ));\n\n JSONObject userPropertiesOperations = event.optJSONObject(\"user_properties\");\n assertEquals(userPropertiesOperations.length(), 1);\n assertTrue(userPropertiesOperations.has(Constants.AMP_OP_SET));\n\n JSONObject groups = event.optJSONObject(\"groups\");\n assertEquals(groups.length(), 1);\n assertEquals(groups.optJSONArray(\"orgId\"), new JSONArray().put(10).put(15));\n\n JSONObject setOperations = userPropertiesOperations.optJSONObject(Constants.AMP_OP_SET);\n assertEquals(setOperations.length(), 1);\n assertEquals(setOperations.optJSONArray(\"orgId\"), new JSONArray().put(10).put(15));\n }\n\n @Test\n public void testLogEventWithGroups() throws JSONException {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n\n JSONObject groups = new JSONObject().put(\"orgId\", 10).put(\"sport\", \"tennis\");\n amplitude.logEvent(\"test\", null, groups);\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n assertEquals(getUnsentEventCount(), 1);\n assertEquals(getUnsentIdentifyCount(), 0);\n JSONObject event = getLastUnsentEvent();\n assertEquals(event.optString(\"event_type\"), \"test\");\n assertTrue(Utils.compareJSONObjects(\n event.optJSONObject(\"event_properties\"), new JSONObject()\n ));\n assertTrue(Utils.compareJSONObjects(\n event.optJSONObject(\"user_properties\"), new JSONObject()\n ));\n\n JSONObject eventGroups = event.optJSONObject(\"groups\");\n assertEquals(eventGroups.length(), 2);\n assertEquals(eventGroups.optInt(\"orgId\"), 10);\n assertEquals(eventGroups.optString(\"sport\"), \"tennis\");\n }\n\n @Test\n public void testMergeEventsArrayIndexOutOfBounds() throws JSONException {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n\n amplitude.setOffline(true);\n\n amplitude.logEvent(\"testEvent1\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n // force failure case\n amplitude.setLastEventId(0);\n\n amplitude.setOffline(false);\n looper.runToEndOfTasks();\n\n // make sure next upload succeeds\n amplitude.setLastEventId(1);\n amplitude.logEvent(\"testEvent2\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n assertEquals(events.length(), 2);\n\n assertEquals(events.getJSONObject(0).optString(\"event_type\"), \"testEvent1\");\n assertEquals(events.getJSONObject(0).optLong(\"event_id\"), 1);\n\n assertEquals(events.getJSONObject(1).optString(\"event_type\"), \"testEvent2\");\n assertEquals(events.getJSONObject(1).optLong(\"event_id\"), 2);\n }\n\n @Test\n public void testCursorWindowAllocationException() {\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n\n // log an event successfully\n amplitude.logEvent(\"testEvent1\");\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 1);\n assertEquals(getUnsentIdentifyCount(), 0);\n\n // mock out database helper to force CursorWindowAllocationExceptions\n DatabaseHelper.instances.put(Constants.DEFAULT_INSTANCE, new MockDatabaseHelper(context));\n\n // force an upload and verify no request sent\n // make sure we catch it during sending of events and defer sending\n RecordedRequest request = runRequest(amplitude);\n assertNull(request);\n assertEquals(getUnsentEventCount(), 1);\n assertEquals(getUnsentIdentifyCount(), 0);\n\n // make sure we catch it during initialization and treat as uninitialized\n amplitude.initialized = false;\n amplitude.initialize(context, apiKey);\n looper.runToEndOfTasks();\n assertNull(amplitude.apiKey);\n\n // since event meta data is loaded during initialize, in theory we should\n // be able to log an event even if we can't query from it\n amplitude.context = context;\n amplitude.apiKey = apiKey;\n Identify identify = new Identify().set(\"car\", \"blue\");\n amplitude.identify(identify);\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 1);\n assertEquals(getUnsentIdentifyCount(), 1);\n }\n\n @Test\n public void testBlockTooManyEventUserProperties() throws JSONException {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n\n JSONObject eventProperties = new JSONObject();\n JSONObject userProperties = new JSONObject();\n Identify identify = new Identify();\n\n for (int i = 0; i < Constants.MAX_PROPERTY_KEYS + 1; i++) {\n eventProperties.put(String.valueOf(i), i);\n userProperties.put(String.valueOf(i*2), i*2);\n identify.setOnce(String.valueOf(i), i);\n }\n\n // verify user properties is filtered out\n amplitude.setUserProperties(userProperties);\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(getUnsentIdentifyCount(), 0);\n\n // verify scrubbed from events\n amplitude.logEvent(\"test event\", eventProperties);\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 1);\n JSONObject event = getLastUnsentEvent();\n assertEquals(event.optString(\"event_type\"), \"test event\");\n assertTrue(Utils.compareJSONObjects(\n event.optJSONObject(\"event_properties\"), new JSONObject()\n ));\n\n // verify scrubbed from identifys - but leaves an empty JSONObject\n amplitude.identify(identify);\n looper.runToEndOfTasks();\n assertEquals(getUnsentIdentifyCount(), 1);\n JSONObject identifyEvent = getLastUnsentIdentify();\n assertEquals(identifyEvent.optString(\"event_type\"), \"$identify\");\n assertTrue(Utils.compareJSONObjects(\n identifyEvent.optJSONObject(\"user_properties\"),\n new JSONObject().put(\"$setOnce\", new JSONObject())\n ));\n }\n\n @Test\n public void testLogEventWithTimestamp() throws JSONException {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n\n amplitude.logEvent(\"test\", null, null, 1000, false);\n looper.runToEndOfTasks();\n JSONObject event = getLastUnsentEvent();\n assertEquals(event.optLong(\"timestamp\"), 1000);\n\n amplitude.logEventSync(\"test\", null, null, 2000, false);\n looper.runToEndOfTasks();\n event = getLastUnsentEvent();\n assertEquals(event.optLong(\"timestamp\"), 2000);\n }\n\n @Test\n public void testRegenerateDeviceId() {\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n String oldDeviceId = amplitude.getDeviceId();\n assertEquals(oldDeviceId, dbHelper.getValue(\"device_id\"));\n\n amplitude.regenerateDeviceId();\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n String newDeviceId = amplitude.getDeviceId();\n assertNotEquals(oldDeviceId, newDeviceId);\n assertEquals(newDeviceId, dbHelper.getValue(\"device_id\"));\n assertTrue(newDeviceId.endsWith(\"R\"));\n }\n\n @Test\n public void testSendNullEvents() throws JSONException {\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n\n dbHelper.addEvent(null);\n amplitude.setLastEventId(1);\n amplitude.getNextSequenceNumber();\n assertEquals(getUnsentEventCount(), 1);\n\n amplitude.logEvent(\"test event\");\n looper.runToEndOfTasks();\n\n amplitude.updateServer();\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n assertEquals(events.length(), 1);\n assertEquals(events.optJSONObject(0).optString(\"event_type\"), \"test event\");\n }\n\n @Test\n @PrepareForTest(OkHttpClient.class)\n public void testHandleUploadExceptions() throws Exception {\n ShadowLooper logLooper = Shadows.shadowOf(amplitude.logThread.getLooper());\n ShadowLooper httpLooper = Shadows.shadowOf(amplitude.httpThread.getLooper());\n IOException error = new IOException(\"test IO Exception\");\n\n // mock out client\n Call.Factory oldClient = amplitude.callFactory;\n OkHttpClient mockClient = PowerMockito.mock(OkHttpClient.class);\n\n // need to have mock client return mock call that throws exception\n Call mockCall = PowerMockito.mock(Call.class);\n PowerMockito.when(mockCall.execute()).thenThrow(error);\n PowerMockito.when(mockClient.newCall(Matchers.any(Request.class))).thenReturn(mockCall);\n\n // attach mock client to amplitude\n amplitude.callFactory = mockClient;\n amplitude.logEvent(\"test event\");\n logLooper.runToEndOfTasks();\n logLooper.runToEndOfTasks();\n httpLooper.runToEndOfTasks();\n\n assertEquals(amplitude.lastError, error);\n\n // restore old client\n amplitude.callFactory = oldClient;\n }\n\n @Test\n public void testDefaultPlatform() throws InterruptedException {\n long [] timestamps = {1, 2, 3, 4, 5, 6, 7};\n clock.setTimestamps(timestamps);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n\n assertEquals(amplitude.platform, Constants.PLATFORM);\n\n amplitude.logEvent(\"test_event1\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n assertEquals(getUnsentEventCount(), 1);\n assertEquals(getUnsentIdentifyCount(), 0);\n JSONArray events = getUnsentEvents(1);\n for (int i = 0; i < 1; i++) {\n assertEquals(events.optJSONObject(i).optString(\"event_type\"), \"test_event\" + (i+1));\n assertEquals(events.optJSONObject(i).optLong(\"timestamp\"), timestamps[i]);\n assertEquals(events.optJSONObject(i).optString(\"platform\"), Constants.PLATFORM);\n }\n runRequest(amplitude);\n }\n\n @Test\n public void testOverridePlatform() throws InterruptedException {\n long [] timestamps = {1, 2, 3, 4, 5, 6, 7};\n clock.setTimestamps(timestamps);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n\n String customPlatform = \"test_custom_platform\";\n\n // force re-initialize to override platform\n amplitude.initialized = false;\n amplitude.initialize(context, apiKey, null, customPlatform, false);\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n assertEquals(amplitude.platform, customPlatform);\n\n amplitude.logEvent(\"test_event1\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n assertEquals(getUnsentEventCount(), 1);\n assertEquals(getUnsentIdentifyCount(), 0);\n JSONArray events = getUnsentEvents(1);\n for (int i = 0; i < 1; i++) {\n assertEquals(events.optJSONObject(i).optString(\"event_type\"), \"test_event\" + (i+1));\n assertEquals(events.optJSONObject(i).optLong(\"timestamp\"), timestamps[i]);\n assertEquals(events.optJSONObject(i).optString(\"platform\"), customPlatform);\n }\n runRequest(amplitude);\n }\n\n @Test\n public void testSetTrackingConfig() throws JSONException {\n long [] timestamps = {1, 2, 3, 4, 5, 6, 7};\n clock.setTimestamps(timestamps);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n\n TrackingOptions options = new TrackingOptions().disableCity().disableCountry().disableIpAddress().disableLanguage().disableLatLng();\n amplitude.setTrackingOptions(options);\n\n assertEquals(amplitude.appliedTrackingOptions, options);\n assertTrue(Utils.compareJSONObjects(amplitude.apiPropertiesTrackingOptions, options.getApiPropertiesTrackingOptions()));\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackCity());\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackCountry());\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackIpAddress());\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackLanguage());\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackLatLng());\n\n amplitude.logEvent(\"test event\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n JSONArray events = getUnsentEvents(1);\n assertEquals(events.length(), 1);\n JSONObject event = events.getJSONObject(0);\n\n // verify we do have platform and carrier since those were not filtered out\n assertTrue(event.has(\"carrier\"));\n assertTrue(event.has(\"platform\"));\n\n // verify we do not have any of the filtered out fields\n assertFalse(event.has(\"city\"));\n assertFalse(event.has(\"country\"));\n assertFalse(event.has(\"language\"));\n\n // verify api properties contains tracking options for location filtering\n JSONObject apiProperties = event.getJSONObject(\"api_properties\");\n assertFalse(apiProperties.getBoolean(\"limit_ad_tracking\"));\n assertFalse(apiProperties.getBoolean(\"gps_enabled\"));\n assertTrue(apiProperties.has(\"tracking_options\"));\n\n JSONObject trackingOptions = apiProperties.getJSONObject(\"tracking_options\");\n assertEquals(trackingOptions.length(), 4);\n assertFalse(trackingOptions.getBoolean(\"city\"));\n assertFalse(trackingOptions.getBoolean(\"country\"));\n assertFalse(trackingOptions.getBoolean(\"ip_address\"));\n assertFalse(trackingOptions.getBoolean(\"lat_lng\"));\n }\n\n @Test\n public void testEnableCoppaControl() throws JSONException {\n long [] timestamps = {1, 2, 3, 4, 5, 6, 7};\n clock.setTimestamps(timestamps);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n\n amplitude.disableCoppaControl(); // this shouldn't do anything\n\n TrackingOptions options = new TrackingOptions();\n assertEquals(amplitude.inputTrackingOptions, options);\n assertEquals(amplitude.appliedTrackingOptions, options);\n assertTrue(Utils.compareJSONObjects(amplitude.apiPropertiesTrackingOptions, options.getApiPropertiesTrackingOptions()));\n\n // haven't merged in the privacy guard settings yet\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackLanguage());\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackCarrier());\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackIpAddress());\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackAdid());\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackCity());\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackLatLng());\n\n amplitude.logEvent(\"test event\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n JSONArray events = getUnsentEvents(1);\n assertEquals(events.length(), 1);\n JSONObject event = events.getJSONObject(0);\n\n // verify we do have platform and carrier since those were not filtered out\n assertTrue(event.has(\"country\"));\n assertTrue(event.has(\"platform\"));\n assertTrue(event.has(\"carrier\"));\n assertTrue(event.has(\"language\"));\n\n // verify api properties contains tracking options for location filtering\n JSONObject apiProperties = event.getJSONObject(\"api_properties\");\n assertFalse(apiProperties.getBoolean(\"limit_ad_tracking\"));\n assertFalse(apiProperties.getBoolean(\"gps_enabled\"));\n assertFalse(apiProperties.has(\"tracking_options\"));\n\n // test enabling privacy guard\n amplitude.enableCoppaControl();\n assertEquals(amplitude.inputTrackingOptions, options);\n assertNotEquals(amplitude.appliedTrackingOptions, options);\n\n // make sure we merge in the privacy guard options\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackIpAddress());\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackAdid());\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackCity());\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackLatLng());\n\n amplitude.logEvent(\"test event 1\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n events = getUnsentEvents(2);\n assertEquals(events.length(), 2);\n event = events.getJSONObject(1);\n\n // verify we do have platform and carrier since those were not filtered out\n assertTrue(event.has(\"country\"));\n assertTrue(event.has(\"platform\"));\n assertTrue(event.has(\"carrier\"));\n assertTrue(event.has(\"language\"));\n\n // verify api properties contains tracking options for location filtering\n apiProperties = event.getJSONObject(\"api_properties\");\n assertFalse(apiProperties.getBoolean(\"limit_ad_tracking\"));\n assertFalse(apiProperties.getBoolean(\"gps_enabled\"));\n assertTrue(apiProperties.has(\"tracking_options\"));\n\n JSONObject trackingOptions = apiProperties.getJSONObject(\"tracking_options\");\n assertEquals(trackingOptions.length(), 3);\n assertFalse(trackingOptions.getBoolean(\"ip_address\"));\n assertFalse(trackingOptions.getBoolean(\"city\"));\n assertFalse(trackingOptions.getBoolean(\"lat_lng\"));\n\n // test disabling privacy guard\n amplitude.disableCoppaControl();\n\n assertEquals(amplitude.inputTrackingOptions, options);\n assertEquals(amplitude.appliedTrackingOptions, options);\n assertTrue(Utils.compareJSONObjects(amplitude.apiPropertiesTrackingOptions, options.getApiPropertiesTrackingOptions()));\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackLanguage());\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackCarrier());\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackIpAddress());\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackAdid());\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackCity());\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackLatLng());\n\n amplitude.logEvent(\"test event 2\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n events = getUnsentEvents(3);\n assertEquals(events.length(), 3);\n event = events.getJSONObject(2);\n\n // verify we do have platform and carrier since those were not filtered out\n assertTrue(event.has(\"country\"));\n assertTrue(event.has(\"platform\"));\n assertTrue(event.has(\"carrier\"));\n assertTrue(event.has(\"language\"));\n\n // verify api properties contains tracking options for location filtering\n apiProperties = event.getJSONObject(\"api_properties\");\n assertFalse(apiProperties.getBoolean(\"limit_ad_tracking\"));\n assertFalse(apiProperties.getBoolean(\"gps_enabled\"));\n assertFalse(apiProperties.has(\"tracking_options\"));\n }\n\n @Test\n public void testEnableCoppaControlWithOptions() throws JSONException {\n long [] timestamps = {1, 2, 3, 4, 5, 6, 7};\n clock.setTimestamps(timestamps);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n\n TrackingOptions options = new TrackingOptions().disableLanguage().disableCarrier().disableIpAddress();\n amplitude.setTrackingOptions(options);\n\n assertEquals(amplitude.inputTrackingOptions, options);\n assertEquals(amplitude.appliedTrackingOptions, options);\n assertTrue(Utils.compareJSONObjects(amplitude.apiPropertiesTrackingOptions, options.getApiPropertiesTrackingOptions()));\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackLanguage());\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackCarrier());\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackIpAddress());\n\n // haven't merged in the privacy guard settings yet\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackAdid());\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackCity());\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackLatLng());\n\n amplitude.logEvent(\"test event\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n JSONArray events = getUnsentEvents(1);\n assertEquals(events.length(), 1);\n JSONObject event = events.getJSONObject(0);\n\n // verify we do have platform and carrier since those were not filtered out\n assertTrue(event.has(\"country\"));\n assertTrue(event.has(\"platform\"));\n\n // verify we do not have any of the filtered out fields\n assertFalse(event.has(\"carrier\"));\n assertFalse(event.has(\"language\"));\n\n // verify api properties contains tracking options for location filtering\n JSONObject apiProperties = event.getJSONObject(\"api_properties\");\n assertFalse(apiProperties.getBoolean(\"limit_ad_tracking\"));\n assertFalse(apiProperties.getBoolean(\"gps_enabled\"));\n assertTrue(apiProperties.has(\"tracking_options\"));\n\n JSONObject trackingOptions = apiProperties.getJSONObject(\"tracking_options\");\n assertEquals(trackingOptions.length(), 1);\n assertFalse(trackingOptions.getBoolean(\"ip_address\"));\n\n // when we enable privacy guard, make sure we maintain original tracking options\n amplitude.enableCoppaControl();\n assertEquals(amplitude.inputTrackingOptions, options);\n assertNotEquals(amplitude.appliedTrackingOptions, options);\n\n // also make sure we merge in the privacy guard options\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackLanguage());\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackCarrier());\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackIpAddress());\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackAdid());\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackCity());\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackLatLng());\n\n amplitude.logEvent(\"test event 1\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n events = getUnsentEvents(2);\n assertEquals(events.length(), 2);\n event = events.getJSONObject(1);\n\n // verify we do have platform and carrier since those were not filtered out\n assertTrue(event.has(\"country\"));\n assertTrue(event.has(\"platform\"));\n\n // verify we do not have any of the filtered out fields\n assertFalse(event.has(\"carrier\"));\n assertFalse(event.has(\"language\"));\n\n // verify api properties contains tracking options for location filtering\n apiProperties = event.getJSONObject(\"api_properties\");\n assertFalse(apiProperties.getBoolean(\"limit_ad_tracking\"));\n assertFalse(apiProperties.getBoolean(\"gps_enabled\"));\n assertTrue(apiProperties.has(\"tracking_options\"));\n\n trackingOptions = apiProperties.getJSONObject(\"tracking_options\");\n assertEquals(trackingOptions.length(), 3);\n assertFalse(trackingOptions.getBoolean(\"ip_address\"));\n assertFalse(trackingOptions.getBoolean(\"city\"));\n assertFalse(trackingOptions.getBoolean(\"lat_lng\"));\n\n // disable privacy guard and make sure original user input is maintained\n amplitude.disableCoppaControl();\n\n assertEquals(amplitude.inputTrackingOptions, options);\n assertEquals(amplitude.appliedTrackingOptions, options);\n assertTrue(Utils.compareJSONObjects(amplitude.apiPropertiesTrackingOptions, options.getApiPropertiesTrackingOptions()));\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackLanguage());\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackCarrier());\n assertFalse(amplitude.appliedTrackingOptions.shouldTrackIpAddress());\n\n // haven't merged in the privacy guard settings yet\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackAdid());\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackCity());\n assertTrue(amplitude.appliedTrackingOptions.shouldTrackLatLng());\n\n amplitude.logEvent(\"test event 2\");\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n events = getUnsentEvents(3);\n assertEquals(events.length(), 3);\n event = events.getJSONObject(2);\n\n // verify we do have platform and carrier since those were not filtered out\n assertTrue(event.has(\"country\"));\n assertTrue(event.has(\"platform\"));\n\n // verify we do not have any of the filtered out fields\n assertFalse(event.has(\"carrier\"));\n assertFalse(event.has(\"language\"));\n\n // verify api properties contains tracking options for location filtering\n apiProperties = event.getJSONObject(\"api_properties\");\n assertFalse(apiProperties.getBoolean(\"limit_ad_tracking\"));\n assertFalse(apiProperties.getBoolean(\"gps_enabled\"));\n assertTrue(apiProperties.has(\"tracking_options\"));\n\n trackingOptions = apiProperties.getJSONObject(\"tracking_options\");\n assertEquals(trackingOptions.length(), 1);\n assertFalse(trackingOptions.getBoolean(\"ip_address\"));\n }\n\n @Test\n public void testGroupIdentifyMultipleOperations() throws JSONException {\n String groupType = \"test group type\";\n String groupName = \"test group name\";\n\n String property1 = \"string value\";\n String value1 = \"testValue\";\n\n String property2 = \"double value\";\n double value2 = 0.123;\n\n String property3 = \"boolean value\";\n boolean value3 = true;\n\n String property4 = \"json value\";\n\n Identify identify = new Identify().setOnce(property1, value1).add(property2, value2);\n identify.set(property3, value3).unset(property4);\n\n // identify should ignore this since duplicate key\n identify.set(property4, value3);\n\n amplitude.groupIdentify(groupType, groupName, identify);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n assertEquals(getUnsentIdentifyCount(), 1);\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(0, getIdentifyInterceptorCount());\n JSONObject event = getLastUnsentIdentify();\n assertEquals(Constants.GROUP_IDENTIFY_EVENT, event.optString(\"event_type\"));\n\n assertTrue(Utils.compareJSONObjects(event.optJSONObject(\"event_properties\"), new JSONObject()));\n assertTrue(Utils.compareJSONObjects(event.optJSONObject(\"user_properties\"), new JSONObject()));\n\n JSONObject groups = event.optJSONObject(\"groups\");\n JSONObject expectedGroups = new JSONObject();\n expectedGroups.put(groupType, groupName);\n assertTrue(Utils.compareJSONObjects(groups, expectedGroups));\n\n JSONObject groupProperties = event.optJSONObject(\"group_properties\");\n JSONObject expected = new JSONObject();\n expected.put(Constants.AMP_OP_SET_ONCE, new JSONObject().put(property1, value1));\n expected.put(Constants.AMP_OP_ADD, new JSONObject().put(property2, value2));\n expected.put(Constants.AMP_OP_SET, new JSONObject().put(property3, value3));\n expected.put(Constants.AMP_OP_UNSET, new JSONObject().put(property4, \"-\"));\n assertTrue(Utils.compareJSONObjects(groupProperties, expected));\n }\n\n @Test\n public void testGroupIdentifyPropertiesObject() throws JSONException {\n String groupType = \"test group type\";\n String groupName = \"test group name\";\n\n String property1 = \"string value\";\n String value1 = \"testValue\";\n\n String property2 = \"double value\";\n double value2 = 0.123;\n\n String property3 = \"boolean value\";\n boolean value3 = true;\n\n String property4 = \"json value\";\n\n JSONObject properties = new JSONObject()\n .put(property1, value1)\n .put(property2, value2)\n .put(property3, value3)\n .put(property4, null);\n\n amplitude.groupIdentify(groupType, groupName, properties, false, null);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n assertEquals(getUnsentIdentifyCount(), 1);\n assertEquals(getUnsentEventCount(), 0);\n JSONObject event = getLastUnsentIdentify();\n assertEquals(Constants.GROUP_IDENTIFY_EVENT, event.optString(\"event_type\"));\n\n assertTrue(Utils.compareJSONObjects(event.optJSONObject(\"event_properties\"), new JSONObject()));\n assertTrue(Utils.compareJSONObjects(event.optJSONObject(\"user_properties\"), new JSONObject()));\n\n JSONObject groups = event.optJSONObject(\"groups\");\n JSONObject expectedGroups = new JSONObject();\n expectedGroups.put(groupType, groupName);\n assertTrue(Utils.compareJSONObjects(groups, expectedGroups));\n\n JSONObject groupProperties = event.optJSONObject(\"group_properties\");\n JSONObject expected = new JSONObject();\n expected.put(Constants.AMP_OP_SET, new JSONObject()\n .put(property1, value1)\n .put(property2, value2)\n .put(property3, value3));\n assertTrue(Utils.compareJSONObjects(groupProperties, expected));\n }\n\n @Test\n public void testSetLogCallback() {\n class TestLogCallback implements AmplitudeLogCallback {\n String errorMsg = null;\n\n @Override\n public void onError(String tag, String message) {\n this.errorMsg = message;\n }\n\n private String getErrorMsg() {\n return this.errorMsg;\n }\n }\n TestLogCallback callback = new TestLogCallback();\n amplitude.setLogCallback(callback);\n assertNull(callback.getErrorMsg());\n amplitude.validateLogEvent(\"\");\n assertEquals(\"Argument eventType cannot be null or blank in logEvent()\", callback.getErrorMsg());\n }\n\n @Test\n public void testSetPlan() {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n String branch = \"main\";\n String version = \"1.0.0\";\n String versionId = \"9ec23ba0-275f-468f-80d1-66b88bff9529\";\n Plan plan = new Plan().setBranch(branch).setVersion(version).setVersionId(versionId);\n amplitude.setPlan(plan);\n amplitude.logEvent(\"test\");\n looper.runToEndOfTasks();\n\n JSONObject event = getLastEvent();\n assertNotNull(event);\n try {\n JSONObject planJsonObject = event.getJSONObject(\"plan\");\n assertEquals(branch, planJsonObject.getString(\"branch\"));\n assertEquals(version, planJsonObject.getString(\"version\"));\n assertEquals(versionId, planJsonObject.getString(\"versionId\"));\n } catch (Exception e) {\n Assert.fail();\n }\n }\n\n @Test\n public void testSetIngestionMetadata() {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n String sourceName = \"ampli\";\n String sourceVersion = \"1.0.0\";\n IngestionMetadata ingestionMetadata = new IngestionMetadata()\n .setSourceName(sourceName)\n .setSourceVersion(sourceVersion);\n amplitude.setIngestionMetadata(ingestionMetadata);\n amplitude.logEvent(\"test\");\n looper.runToEndOfTasks();\n\n JSONObject event = getLastEvent();\n assertNotNull(event);\n try {\n JSONObject jsonObject = event.getJSONObject(\"ingestion_metadata\");\n assertEquals(sourceName, jsonObject.getString(\"source_name\"));\n assertEquals(sourceVersion, jsonObject.getString(\"source_version\"));\n } catch (Exception e) {\n Assert.fail();\n }\n }\n\n @Test\n public void testSetServerZoneWithoutUpdateServerUrl() {\n String urlBeforeChange = amplitude.url;\n AmplitudeServerZone euZone = AmplitudeServerZone.EU;\n amplitude.setServerZone(euZone, false);\n assertEquals(euZone, getPrivateFieldValueFromClient(amplitude, \"serverZone\"));\n assertEquals(urlBeforeChange, amplitude.url);\n }\n\n @Test\n public void testSetServerZoneAndUpdateServerUrl() {\n AmplitudeServerZone euZone = AmplitudeServerZone.EU;\n amplitude.setServerZone(euZone);\n assertEquals(euZone, getPrivateFieldValueFromClient(amplitude, \"serverZone\"));\n assertEquals(Constants.EVENT_LOG_EU_URL, amplitude.url);\n }\n\n @Test\n public void testMiddlewareSupport() throws JSONException {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n Map extraMap = new HashMap<>();\n extraMap.put(\"description\", \"extra description\");\n MiddlewareExtra extra = new MiddlewareExtra(extraMap);\n Middleware middleware = new Middleware() {\n @Override\n public void run(MiddlewarePayload payload, MiddlewareNext next) {\n try {\n payload.event.optJSONObject(\"event_properties\").put(\"description\", \"extra description\");\n } catch (JSONException e) {\n e.printStackTrace();\n }\n\n next.run(payload);\n }\n };\n amplitude.addEventMiddleware(middleware);\n amplitude.logEvent(\"middleware_event_type\", new JSONObject().put(\"user_id\", \"middleware_user\"), extra);\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n assertEquals(getUnsentEventCount(), 1);\n JSONArray eventObject = getUnsentEvents(1);;\n assertEquals(eventObject.optJSONObject(0).optString(\"event_type\"), \"middleware_event_type\");\n assertEquals(eventObject.optJSONObject(0).optJSONObject(\"event_properties\").getString(\"description\"), \"extra description\");\n assertEquals(eventObject.optJSONObject(0).optJSONObject(\"event_properties\").optString(\"user_id\"), \"middleware_user\");\n }\n\n @Test\n public void testWithSwallowMiddleware() throws JSONException {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n Middleware middleware = new Middleware() {\n @Override\n public void run(MiddlewarePayload payload, MiddlewareNext next) {\n }\n };\n amplitude.addEventMiddleware(middleware);\n amplitude.logEvent(\"middleware_event_type\", new JSONObject().put(\"user_id\", \"middleware_user\"));\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n assertEquals(getUnsentEventCount(), 0);\n }\n\n @Test\n public void setIdentifyBatchIntervalMillis() {\n amplitude.setIdentifyBatchIntervalMillis(10000);\n assertEquals(30000L, getPrivateFieldValueFromClient(amplitude, \"identifyBatchIntervalMillis\"));\n amplitude.setIdentifyBatchIntervalMillis(40000);\n assertEquals(40000L, getPrivateFieldValueFromClient(amplitude, \"identifyBatchIntervalMillis\"));\n }\n\n @Test\n public void testMultipleIdentifyWithSetActionOnlySendOneIdentifyEvent() throws JSONException {\n long [] timestamps = {1000, 1001, 1002, 1003};\n clock.setTimestamps(timestamps);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n amplitude.identify(new Identify().set(\"key1\", \"key1-value1\").set(\"key2\", \"key2-value1\").set(\"key3\", \"key3-value1\"));\n amplitude.identify(new Identify().set(\"key1\", \"key1-value2\").set(\"key4\", \"key4-value1\"));\n amplitude.identify(new Identify().set(\"key2\", \"key2-value2\"));\n amplitude.identify(new Identify().set(\"key3\", \"key3-value2\").set(\"key4\", \"key4-value2\"));\n looper.runToEndOfTasks();\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n assertEquals(4L, getIdentifyInterceptorCount());\n assertEquals(0, getUnsentIdentifyCount());\n assertNull(dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY));\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY), 4L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY), timestamps[3]);\n\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n assertEquals(events.length(), 1);\n JSONObject identify = events.getJSONObject(0);\n assertEquals(identify.getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(identify.getLong(\"event_id\"), 1);\n assertEquals(identify.getLong(\"timestamp\"), timestamps[0]);\n assertEquals(identify.getLong(\"sequence_number\"), 1);\n JSONObject userProperties = identify.getJSONObject(\"user_properties\");\n assertEquals(userProperties.length(), 1);\n assertTrue(userProperties.has(Constants.AMP_OP_SET));\n JSONObject expected = new JSONObject();\n expected.put(\"key1\", \"key1-value2\");\n expected.put(\"key2\", \"key2-value2\");\n expected.put(\"key3\", \"key3-value2\");\n expected.put(\"key4\", \"key4-value2\");\n assertTrue(Utils.compareJSONObjects(userProperties.getJSONObject(Constants.AMP_OP_SET), expected));\n assertEquals(0, getIdentifyInterceptorCount());\n assertEquals((long)dbHelper.getLastIdentifyInterceptorId(), -1L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY), 1L);\n }\n\n @Test\n public void testMultipleIdentifyWithSetActionOnlyAndOneEvent() throws JSONException {\n long [] timestamps = {1000, 1001, 1002, 1003, 1004};\n clock.setTimestamps(timestamps);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n Middleware middleware = new Middleware() {\n @Override\n public void run(MiddlewarePayload payload, MiddlewareNext next) {\n try {\n if (!payload.event.getString(\"event_type\").equals(Constants.IDENTIFY_EVENT)) {\n payload.event.optJSONObject(\"user_properties\").put(\"key1\", \"key1-value3\")\n .put(\"key2\", \"key2-value3\");\n }\n } catch (JSONException e) {\n e.printStackTrace();\n }\n\n next.run(payload);\n }\n };\n amplitude.addEventMiddleware(middleware);\n amplitude.identify(new Identify().set(\"key1\", \"key1-value1\").set(\"key2\", \"key2-value1\").set(\"key3\", \"key3-value1\"));\n amplitude.identify(new Identify().set(\"key1\", \"key1-value2\").set(\"key4\", \"key4-value1\"));\n amplitude.identify(new Identify().set(\"key2\", \"key2-value2\"));\n amplitude.identify(new Identify().set(\"key3\", \"key3-value2\").set(\"key4\", \"key4-value2\"));\n amplitude.logEvent(\"test_event\", new JSONObject().put(\"test_event_prop_key\", \"test_event_prop_value\"));\n looper.runToEndOfTasks();\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n assertEquals(1L, getUnsentIdentifyCount());\n assertEquals(1L, getUnsentEventCount());\n assertEquals(1L, (long)dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY));\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY), 5L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY), timestamps[4]);\n\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n assertEquals(events.length(), 2);\n JSONObject event = events.getJSONObject(0);\n assertEquals(event.getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(event.getLong(\"event_id\"), 1);\n assertEquals(event.getLong(\"timestamp\"), timestamps[0]);\n assertEquals(event.getLong(\"sequence_number\"), 1);\n JSONObject userProperties = event.getJSONObject(\"user_properties\").getJSONObject(Constants.AMP_OP_SET);\n assertEquals(userProperties.length(), 4);\n JSONObject expected = new JSONObject();\n expected.put(\"key1\", \"key1-value2\");\n expected.put(\"key2\", \"key2-value2\");\n expected.put(\"key3\", \"key3-value2\");\n expected.put(\"key4\", \"key4-value2\");\n assertTrue(Utils.compareJSONObjects(userProperties, expected));\n\n JSONObject event2 = events.getJSONObject(1);\n assertEquals(event2.getString(\"event_type\"), \"test_event\");\n assertEquals(event2.getLong(\"event_id\"), 1);\n assertEquals(event2.getLong(\"timestamp\"), timestamps[4]);\n assertEquals(event2.getLong(\"sequence_number\"), 5);\n JSONObject userProperties2 = event2.getJSONObject(\"user_properties\");\n assertEquals(userProperties2.length(), 2);\n JSONObject expected2 = new JSONObject();\n expected2.put(\"key1\", \"key1-value3\");\n expected2.put(\"key2\", \"key2-value3\");\n assertTrue(Utils.compareJSONObjects(userProperties2, expected2));\n assertEquals(0, getIdentifyInterceptorCount());\n assertEquals((long)dbHelper.getLastIdentifyInterceptorId(), -1L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY), 1L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_ID_KEY), 1L);\n }\n\n @Test\n public void testMultipleIdentifyWithSetActionAndOneEventAndIdentify() throws JSONException {\n long [] timestamps = {1000, 1001, 1002, 1003, 1004, 1005};\n clock.setTimestamps(timestamps);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n Middleware middleware = new Middleware() {\n @Override\n public void run(MiddlewarePayload payload, MiddlewareNext next) {\n try {\n if (!payload.event.getString(\"event_type\").equals(Constants.IDENTIFY_EVENT)) {\n payload.event.optJSONObject(\"user_properties\").put(\"key1\", \"key1-value3\")\n .put(\"key2\", \"key2-value3\");\n }\n } catch (JSONException e) {\n e.printStackTrace();\n }\n\n next.run(payload);\n }\n };\n amplitude.addEventMiddleware(middleware);\n amplitude.identify(new Identify().set(\"key1\", \"key1-value1\").set(\"key2\", \"key2-value1\").set(\"key3\", \"key3-value1\"));\n amplitude.identify(new Identify().set(\"key1\", \"key1-value2\").set(\"key4\", \"key4-value1\"));\n amplitude.identify(new Identify().set(\"key2\", \"key2-value2\"));\n amplitude.identify(new Identify().set(\"key3\", \"key3-value2\").set(\"key4\", \"key4-value2\"));\n amplitude.logEvent(\"test_event\", new JSONObject().put(\"test_event_prop_key\", \"test_event_prop_value\"));\n amplitude.setUserProperties(new JSONObject().put(\"key1\", \"key1-value4\"));\n looper.runToEndOfTasks();\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n assertEquals(1L, getUnsentIdentifyCount());\n assertEquals(1L, getUnsentEventCount());\n assertEquals(1L, (long)dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY));\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY), 6L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY), timestamps[5]);\n\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n assertEquals(events.length(), 3);\n JSONObject event = events.getJSONObject(0);\n assertEquals(event.getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(event.getLong(\"event_id\"), 1);\n assertEquals(event.getLong(\"timestamp\"), timestamps[0]);\n assertEquals(event.getLong(\"sequence_number\"), 1);\n JSONObject userProperties = event.getJSONObject(\"user_properties\").getJSONObject(Constants.AMP_OP_SET);\n assertEquals(userProperties.length(), 4);\n JSONObject expected = new JSONObject();\n expected.put(\"key1\", \"key1-value2\");\n expected.put(\"key2\", \"key2-value2\");\n expected.put(\"key3\", \"key3-value2\");\n expected.put(\"key4\", \"key4-value2\");\n\n JSONObject event2 = events.getJSONObject(1);\n assertEquals(event2.getString(\"event_type\"), \"test_event\");\n assertEquals(event2.getLong(\"event_id\"), 1);\n assertEquals(event2.getLong(\"timestamp\"), timestamps[4]);\n assertEquals(event2.getLong(\"sequence_number\"), 5);\n JSONObject userProperties2 = event2.getJSONObject(\"user_properties\");\n assertEquals(userProperties.length(), 4);\n JSONObject expected2 = new JSONObject();\n expected2.put(\"key1\", \"key1-value3\");\n expected2.put(\"key2\", \"key2-value3\");\n assertTrue(Utils.compareJSONObjects(userProperties2, expected2));\n\n JSONObject event3 = events.getJSONObject(2);\n assertEquals(event3.getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(event3.getLong(\"event_id\"), 2);\n assertEquals(event3.getLong(\"timestamp\"), timestamps[5]);\n assertEquals(event3.getLong(\"sequence_number\"), 6);\n JSONObject userProperties3 = event3.getJSONObject(\"user_properties\").getJSONObject(Constants.AMP_OP_SET);\n assertEquals(userProperties2.length(), 2);\n JSONObject expected3 = new JSONObject();\n expected3.put(\"key1\", \"key1-value4\");\n assertTrue(Utils.compareJSONObjects(userProperties3, expected3));\n assertEquals(0, getIdentifyInterceptorCount());\n assertEquals((long)dbHelper.getLastIdentifyInterceptorId(), -1L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY), 2L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_ID_KEY), 1L);\n }\n\n @Test\n public void testIdentifyInterceptWithSetAndClearAll() throws JSONException {\n long [] timestamps = {1000, 1001, 1002, 1003, 1004};\n clock.setTimestamps(timestamps);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n amplitude.identify(new Identify().set(\"key1\", \"key1-value1\").set(\"key2\", \"key2-value1\").set(\"key3\", \"key3-value1\"));\n amplitude.identify(new Identify().set(\"key1\", \"key1-value2\").set(\"key4\", \"key4-value1\"));\n amplitude.identify(new Identify().set(\"key2\", \"key2-value2\"));\n amplitude.identify(new Identify().set(\"key3\", \"key3-value2\").set(\"key4\", \"key4-value2\"));\n amplitude.identify(new Identify().clearAll());\n looper.runToEndOfTasks();\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n assertEquals(0, getIdentifyInterceptorCount());\n assertEquals(1, getUnsentIdentifyCount());\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY), 1L);\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY), 5L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY), timestamps[4]);\n\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n assertEquals(events.length(), 1);\n JSONObject identify = events.getJSONObject(0);\n assertEquals(identify.getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(identify.getLong(\"event_id\"), 1);\n assertEquals(identify.getLong(\"timestamp\"), timestamps[4]);\n assertEquals(identify.getLong(\"sequence_number\"), 5);\n JSONObject userProperties = identify.getJSONObject(\"user_properties\");\n assertEquals(userProperties.length(), 1);\n assertTrue(userProperties.has(Constants.AMP_OP_CLEAR_ALL));\n assertFalse(userProperties.has(Constants.AMP_OP_SET));\n assertEquals(0, getIdentifyInterceptorCount());\n assertEquals((long)dbHelper.getLastIdentifyInterceptorId(), -1L);\n }\n\n @Test\n public void testMultipleIdentifyWithSetActionAndAnotherIdentify() throws JSONException {\n long [] timestamps = {1000, 1001, 1002, 1003, 1004};\n clock.setTimestamps(timestamps);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n amplitude.identify(new Identify().set(\"key1\", \"key1-value1\").set(\"key2\", \"key2-value1\").set(\"key3\", \"key3-value1\"));\n amplitude.identify(new Identify().set(\"key1\", \"key1-value2\").set(\"key4\", \"key4-value1\"));\n amplitude.identify(new Identify().set(\"key2\", \"key2-value2\"));\n amplitude.identify(new Identify().set(\"key3\", \"key3-value2\").set(\"key4\", \"key4-value2\"));\n amplitude.identify(new Identify().add(\"key5\", 2));\n looper.runToEndOfTasks();\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n assertEquals(2L, getUnsentIdentifyCount());\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY), 2L);\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY), 5L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY), timestamps[4]);\n\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n assertEquals(events.length(), 2);\n JSONObject event = events.getJSONObject(0);\n assertEquals(event.getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(event.getLong(\"event_id\"), 1);\n assertEquals(event.getLong(\"timestamp\"), timestamps[0]);\n assertEquals(event.getLong(\"sequence_number\"), 1);\n JSONObject userProperties = event.getJSONObject(\"user_properties\");\n assertEquals(userProperties.length(), 1);\n JSONObject expected = new JSONObject();\n expected.put(\"key1\", \"key1-value2\");\n expected.put(\"key2\", \"key2-value2\");\n expected.put(\"key3\", \"key3-value2\");\n expected.put(\"key4\", \"key4-value2\");\n assertTrue(Utils.compareJSONObjects(userProperties.getJSONObject(Constants.AMP_OP_SET), expected));\n\n JSONObject event2 = events.getJSONObject(1);\n assertEquals(event2.getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(event2.getLong(\"event_id\"), 2);\n assertEquals(event2.getLong(\"timestamp\"), timestamps[4]);\n assertEquals(event2.getLong(\"sequence_number\"), 5);\n JSONObject userProperties2 = event2.getJSONObject(\"user_properties\");\n assertEquals(userProperties2.length(), 1);\n JSONObject expectedAdd = new JSONObject();\n expectedAdd.put(\"key5\", 2);\n assertTrue(Utils.compareJSONObjects(userProperties2.getJSONObject(Constants.AMP_OP_ADD), expectedAdd));\n\n assertEquals(0, getIdentifyInterceptorCount());\n assertEquals((long)dbHelper.getLastIdentifyInterceptorId(), -1L);\n }\n\n @Test\n public void testUploadEventsSendInterceptedIdentify() throws JSONException {\n long [] timestamps = {1000, 1001, 1002, 1003};\n clock.setTimestamps(timestamps);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n amplitude.identify(new Identify().set(\"key1\", \"key1-value1\").set(\"key2\", \"key2-value1\").set(\"key3\", \"key3-value1\"));\n amplitude.identify(new Identify().set(\"key1\", \"key1-value2\").set(\"key4\", \"key4-value1\"));\n amplitude.setUserProperties(new JSONObject().put(\"key2\", \"key2-value2\"));\n amplitude.identify(new Identify().set(\"key3\", \"key3-value2\").set(\"key4\", \"key4-value2\"));\n looper.runToEndOfTasks();\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n assertEquals(4L, getIdentifyInterceptorCount());\n assertEquals(0, getUnsentIdentifyCount());\n assertNull(dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY));\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY), 4L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY), timestamps[3]);\n\n amplitude.uploadEvents();\n looper.runToEndOfTasks();\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n assertEquals(events.length(), 1);\n JSONObject identify = events.getJSONObject(0);\n assertEquals(identify.getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(identify.getLong(\"event_id\"), 1);\n assertEquals(identify.getLong(\"timestamp\"), timestamps[0]);\n assertEquals(identify.getLong(\"sequence_number\"), 1);\n JSONObject userProperties = identify.getJSONObject(\"user_properties\");\n assertEquals(userProperties.length(), 1);\n assertTrue(userProperties.has(Constants.AMP_OP_SET));\n JSONObject expected = new JSONObject();\n expected.put(\"key1\", \"key1-value2\");\n expected.put(\"key2\", \"key2-value2\");\n expected.put(\"key3\", \"key3-value2\");\n expected.put(\"key4\", \"key4-value2\");\n assertTrue(Utils.compareJSONObjects(userProperties.getJSONObject(Constants.AMP_OP_SET), expected));\n assertEquals(0, getIdentifyInterceptorCount());\n assertEquals((long)dbHelper.getLastIdentifyInterceptorId(), -1L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY), 1L);\n }\n\n @Test\n public void testMultipleIdentifyWithSetActionAndSetGroup() throws JSONException {\n long [] timestamps = {1000, 1001, 1002, 1003, 1004, 1005};\n clock.setTimestamps(timestamps);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n amplitude.identify(new Identify().set(\"key1\", \"key1-value1\").set(\"key2\", \"key2-value1\").set(\"key3\", \"key3-value1\"));\n amplitude.identify(new Identify().set(\"key1\", \"key1-value2\").set(\"key4\", \"key4-value1\"));\n amplitude.identify(new Identify().set(\"key2\", \"key2-value2\"));\n amplitude.identify(new Identify().set(\"key3\", \"key3-value2\").set(\"key4\", \"key4-value2\"));\n amplitude.setGroup(\"test-group-type\", \"test-group-value\");\n amplitude.identify(new Identify().set(\"key3\", \"key3-value3\").set(\"key4\", \"key4-value3\"));\n looper.runToEndOfTasks();\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n assertEquals(2L, getUnsentIdentifyCount());\n assertEquals(1L, getIdentifyInterceptorCount());\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY), 2L);\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY), 6L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY), timestamps[5]);\n\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n assertEquals(events.length(), 3);\n JSONObject event = events.getJSONObject(0);\n assertEquals(event.getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(event.getLong(\"event_id\"), 1);\n assertEquals(event.getLong(\"timestamp\"), timestamps[0]);\n assertEquals(event.getLong(\"sequence_number\"), 1);\n JSONObject userProperties = event.getJSONObject(\"user_properties\");\n assertEquals(userProperties.length(), 1);\n JSONObject expected = new JSONObject();\n expected.put(\"key1\", \"key1-value2\");\n expected.put(\"key2\", \"key2-value2\");\n expected.put(\"key3\", \"key3-value2\");\n expected.put(\"key4\", \"key4-value2\");\n assertTrue(Utils.compareJSONObjects(userProperties.getJSONObject(Constants.AMP_OP_SET), expected));\n\n JSONObject event2 = events.getJSONObject(1);\n assertEquals(event2.getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(event2.getLong(\"event_id\"), 2);\n assertEquals(event2.getLong(\"timestamp\"), timestamps[4]);\n assertEquals(event2.getLong(\"sequence_number\"), 5);\n JSONObject userProperties2 = event2.getJSONObject(\"user_properties\");\n assertEquals(userProperties2.length(), 1);\n JSONObject expected2 = new JSONObject();\n expected2.put(\"test-group-type\", \"test-group-value\");\n JSONObject expectedGroups = new JSONObject();\n expectedGroups.put(\"test-group-type\", \"test-group-value\");\n assertTrue(Utils.compareJSONObjects(userProperties2.getJSONObject(Constants.AMP_OP_SET), expected2));\n assertTrue(Utils.compareJSONObjects(event2.getJSONObject(\"groups\"), expectedGroups));\n\n JSONObject event3 = events.getJSONObject(2);\n assertEquals(event3.getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(event3.getLong(\"event_id\"), 3);\n assertEquals(event3.getLong(\"timestamp\"), timestamps[5]);\n assertEquals(event3.getLong(\"sequence_number\"), 6);\n JSONObject userProperties3 = event3.getJSONObject(\"user_properties\");\n assertEquals(userProperties3.length(), 1);\n JSONObject expected3 = new JSONObject();\n expected3.put(\"key3\", \"key3-value3\");\n expected3.put(\"key4\", \"key4-value3\");\n assertTrue(Utils.compareJSONObjects(userProperties3.getJSONObject(Constants.AMP_OP_SET), expected3));\n\n assertEquals(0, getIdentifyInterceptorCount());\n assertEquals((long)dbHelper.getLastIdentifyInterceptorId(), -1L);\n }\n\n @Test\n public void testMultipleIdentifyWithSetActionAndUserIdUpdated() throws JSONException {\n long [] timestamps = {1000, 1001, 1002, 1003, 1004};\n clock.setTimestamps(timestamps);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n amplitude.identify(new Identify().set(\"key1\", \"key1-value1\").set(\"key2\", \"key2-value1\").set(\"key3\", \"key3-value1\"));\n amplitude.identify(new Identify().set(\"key1\", \"key1-value2\").set(\"key4\", \"key4-value1\"));\n amplitude.identify(new Identify().set(\"key2\", \"key2-value2\"));\n amplitude.identify(new Identify().set(\"key3\", \"key3-value2\").set(\"key4\", \"key4-value2\"));\n amplitude.setUserId(\"identify-user-id\");\n amplitude.identify(new Identify().set(\"key3\", \"key3-value3\").set(\"key4\", \"key4-value3\"));\n looper.runToEndOfTasks();\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n assertEquals(1L, getUnsentIdentifyCount());\n assertEquals(1L, getIdentifyInterceptorCount());\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY), 1L);\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY), 5L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY), timestamps[4]);\n\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n assertEquals(events.length(), 2);\n JSONObject event = events.getJSONObject(0);\n assertEquals(event.getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(event.getLong(\"event_id\"), 1);\n assertEquals(event.getLong(\"timestamp\"), timestamps[0]);\n assertEquals(event.getLong(\"sequence_number\"), 1);\n assertEquals(event.getString(\"user_id\"), \"null\");\n JSONObject userProperties = event.getJSONObject(\"user_properties\");\n assertEquals(userProperties.length(), 1);\n JSONObject expected = new JSONObject();\n expected.put(\"key1\", \"key1-value2\");\n expected.put(\"key2\", \"key2-value2\");\n expected.put(\"key3\", \"key3-value2\");\n expected.put(\"key4\", \"key4-value2\");\n assertTrue(Utils.compareJSONObjects(userProperties.getJSONObject(Constants.AMP_OP_SET), expected));\n\n JSONObject event2 = events.getJSONObject(1);\n assertEquals(event2.getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(event2.getLong(\"event_id\"), 2);\n assertEquals(event2.getLong(\"timestamp\"), timestamps[4]);\n assertEquals(event2.getLong(\"sequence_number\"), 5);\n assertEquals(event2.getString(\"user_id\"), \"identify-user-id\");\n JSONObject userProperties2 = event2.getJSONObject(\"user_properties\");\n assertEquals(userProperties2.length(), 1);\n JSONObject expected2 = new JSONObject();\n expected2.put(\"key3\", \"key3-value3\");\n expected2.put(\"key4\", \"key4-value3\");\n assertTrue(Utils.compareJSONObjects(userProperties2.getJSONObject(Constants.AMP_OP_SET), expected2));\n\n assertEquals(0, getIdentifyInterceptorCount());\n assertEquals((long)dbHelper.getLastIdentifyInterceptorId(), -1L);\n }\n\n @Test\n public void testNullUserPropertyFilteredOut() throws JSONException {\n long [] timestamps = {1000, 1001, 1002};\n clock.setTimestamps(timestamps);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n amplitude.identify(new Identify().set(\"key1\", \"key1-value1\").set(\"key2\", \"key2-value1\").set(\"key3\", \"key3-value1\"));\n amplitude.identify(new Identify().set(\"key1\", \"key1-value2\").set(\"key4\", \"key4-value1\"));\n amplitude.setUserProperties(new JSONObject().put(\"key2\", null).put(\"key4\", JSONObject.NULL));\n looper.runToEndOfTasks();\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n assertEquals(3L, getIdentifyInterceptorCount());\n assertEquals(0, getUnsentIdentifyCount());\n assertNull(dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY));\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY), 3L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY), timestamps[2]);\n\n amplitude.uploadEvents();\n looper.runToEndOfTasks();\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n assertEquals(events.length(), 1);\n JSONObject identify = events.getJSONObject(0);\n assertEquals(identify.getString(\"event_type\"), Constants.IDENTIFY_EVENT);\n assertEquals(identify.getLong(\"event_id\"), 1);\n assertEquals(identify.getLong(\"timestamp\"), timestamps[0]);\n assertEquals(identify.getLong(\"sequence_number\"), 1);\n JSONObject userProperties = identify.getJSONObject(\"user_properties\");\n assertEquals(userProperties.length(), 1);\n assertTrue(userProperties.has(Constants.AMP_OP_SET));\n JSONObject expected = new JSONObject();\n expected.put(\"key1\", \"key1-value2\");\n expected.put(\"key2\", \"key2-value1\");\n expected.put(\"key3\", \"key3-value1\");\n expected.put(\"key4\", \"key4-value1\");\n System.out.println(userProperties.getJSONObject(Constants.AMP_OP_SET));\n assertTrue(Utils.compareJSONObjects(userProperties.getJSONObject(Constants.AMP_OP_SET), expected));\n assertEquals(0, getIdentifyInterceptorCount());\n assertEquals((long)dbHelper.getLastIdentifyInterceptorId(), -1L);\n assertEquals((long)dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY), 1L);\n }\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/AmplitudeServerZoneTest.java",
"content": "package com.amplitude.api;\n\nimport org.junit.Test;\nimport org.junit.runner.RunWith;\nimport org.junit.runners.Parameterized;\nimport org.robolectric.annotation.Config;\n\nimport java.util.Arrays;\nimport java.util.Collection;\n\nimport static org.junit.Assert.assertEquals;\n\n@RunWith(Parameterized.class)\n@Config(manifest=Config.NONE)\npublic class AmplitudeServerZoneTest {\n\n @Parameterized.Parameters\n public static Collection data() {\n return Arrays.asList(new Object[][] {\n { AmplitudeServerZone.US, Constants.EVENT_LOG_URL, Constants.DYNAMIC_CONFIG_URL, \"US\"},\n { null, Constants.EVENT_LOG_URL, Constants.DYNAMIC_CONFIG_URL, \"\"},\n { AmplitudeServerZone.EU, Constants.EVENT_LOG_EU_URL, Constants.DYNAMIC_CONFIG_EU_URL, \"EU\"}\n });\n }\n\n private AmplitudeServerZone serverZone;\n private String expectedEventLogUrl;\n private String expectedDynamicConfigUrl;\n private String serverZoneString;\n\n public AmplitudeServerZoneTest(\n AmplitudeServerZone serverZone,\n String expectedEventLogUrl,\n String expectedDynamicConfigUrl,\n String serverZoneString\n ) {\n this.serverZone = serverZone;\n this.expectedEventLogUrl = expectedEventLogUrl;\n this.expectedDynamicConfigUrl = expectedDynamicConfigUrl;\n this.serverZoneString = serverZoneString;\n }\n\n @Test\n public void testGetCorrectUrlForAmplitudeServerZone() {\n assertEquals(expectedEventLogUrl, AmplitudeServerZone.getEventLogApiForZone(serverZone));\n assertEquals(expectedDynamicConfigUrl, AmplitudeServerZone.getDynamicConfigApi(serverZone));\n AmplitudeServerZone expectedServerZone = serverZone != null ? serverZone : AmplitudeServerZone.US;\n assertEquals(expectedServerZone, AmplitudeServerZone.getServerZone(serverZoneString));\n }\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/AmplitudeTest.java",
"content": "package com.amplitude.api;\n\nimport android.content.Context;\nimport android.content.SharedPreferences;\n\nimport androidx.test.ext.junit.runners.AndroidJUnit4;\n\nimport org.junit.After;\nimport org.junit.Before;\nimport org.junit.Test;\nimport org.junit.runner.RunWith;\nimport org.robolectric.Shadows;\nimport org.robolectric.annotation.Config;\n\nimport static org.junit.Assert.assertEquals;\nimport static org.junit.Assert.assertFalse;\nimport static org.junit.Assert.assertNotSame;\nimport static org.junit.Assert.assertSame;\nimport static org.junit.Assert.assertTrue;\n\n@RunWith(AndroidJUnit4.class)\n@Config(manifest = Config.NONE)\npublic class AmplitudeTest extends BaseTest {\n\n @Before\n public void setUp() throws Exception {\n super.setUp();\n }\n\n @After\n public void tearDown() throws Exception {\n super.tearDown();\n }\n\n @Test\n public void testGetInstance() {\n AmplitudeClient a = Amplitude.getInstance();\n AmplitudeClient b = Amplitude.getInstance(\"\");\n AmplitudeClient c = Amplitude.getInstance(null);\n AmplitudeClient d = Amplitude.getInstance(Constants.DEFAULT_INSTANCE);\n AmplitudeClient e = Amplitude.getInstance(\"app1\");\n AmplitudeClient f = Amplitude.getInstance(\"app2\");\n\n assertSame(a, b);\n assertSame(b, c);\n assertSame(c, d);\n assertSame(d, Amplitude.getInstance());\n assertNotSame(d, e);\n assertSame(e, Amplitude.getInstance(\"app1\"));\n assertNotSame(e, f);\n assertSame(f, Amplitude.getInstance(\"app2\"));\n\n // test for instance name case insensitivity\n assertSame(e, Amplitude.getInstance(\"APP1\"));\n assertSame(e, Amplitude.getInstance(\"App1\"));\n assertSame(e, Amplitude.getInstance(\"aPP1\"));\n assertSame(e, Amplitude.getInstance(\"apP1\"));\n\n assertTrue(Amplitude.instances.size() == 3);\n assertTrue(Amplitude.instances.containsKey(Constants.DEFAULT_INSTANCE));\n assertTrue(Amplitude.instances.containsKey(\"app1\"));\n assertTrue(Amplitude.instances.containsKey(\"app2\"));\n }\n\n @Test\n public void testSeparateInstancesLogEventsSeparately() {\n Amplitude.instances.clear();\n DatabaseHelper.instances.clear();\n\n String newInstance1 = \"newApp1\";\n String newApiKey1 = \"1234567890\";\n String newInstance2 = \"newApp2\";\n String newApiKey2 = \"0987654321\";\n\n DatabaseHelper oldDbHelper = DatabaseHelper.getDatabaseHelper(context);\n DatabaseHelper newDbHelper1 = DatabaseHelper.getDatabaseHelper(context, newInstance1);\n DatabaseHelper newDbHelper2 = DatabaseHelper.getDatabaseHelper(context, newInstance2);\n\n // Setup existing Databasefile\n oldDbHelper.insertOrReplaceKeyValue(\"device_id\", \"oldDeviceId\");\n oldDbHelper.insertOrReplaceKeyLongValue(\"sequence_number\", 1000L);\n oldDbHelper.addEvent(\"oldEvent1\");\n oldDbHelper.addIdentify(\"oldIdentify1\");\n oldDbHelper.addIdentify(\"oldIdentify2\");\n\n // Verify persistence of old database file in default instance\n Amplitude.getInstance().initialize(context, apiKey);\n Shadows.shadowOf(Amplitude.getInstance().logThread.getLooper()).runToEndOfTasks();\n assertEquals(Amplitude.getInstance().getDeviceId(), \"oldDeviceId\");\n assertEquals(Amplitude.getInstance().getNextSequenceNumber(), 1001L);\n assertTrue(oldDbHelper.dbFileExists());\n assertFalse(newDbHelper1.dbFileExists());\n assertFalse(newDbHelper2.dbFileExists());\n\n // init first new app and verify separate database file\n Amplitude.getInstance(newInstance1).initialize(context, newApiKey1);\n Shadows.shadowOf(\n Amplitude.getInstance(newInstance1).logThread.getLooper()\n ).runToEndOfTasks();\n assertTrue(newDbHelper1.dbFileExists()); // db file is created after deviceId initialization\n\n assertFalse(newDbHelper1.getValue(\"device_id\").equals(\"oldDeviceId\"));\n assertEquals(\n newDbHelper1.getValue(\"device_id\"), Amplitude.getInstance(newInstance1).getDeviceId()\n );\n assertEquals(Amplitude.getInstance(newInstance1).getNextSequenceNumber(), 1L);\n assertEquals(newDbHelper1.getEventCount(), 0);\n assertEquals(newDbHelper1.getIdentifyCount(), 0);\n\n // init second new app and verify separate database file\n Amplitude.getInstance(newInstance2).initialize(context, newApiKey2);\n Shadows.shadowOf(\n Amplitude.getInstance(newInstance2).logThread.getLooper()\n ).runToEndOfTasks();\n assertTrue(newDbHelper2.dbFileExists()); // db file is created after deviceId initialization\n\n assertFalse(newDbHelper2.getValue(\"device_id\").equals(\"oldDeviceId\"));\n assertEquals(\n newDbHelper2.getValue(\"device_id\"), Amplitude.getInstance(newInstance2).getDeviceId()\n );\n assertEquals(Amplitude.getInstance(newInstance2).getNextSequenceNumber(), 1L);\n assertEquals(newDbHelper2.getEventCount(), 0);\n assertEquals(newDbHelper2.getIdentifyCount(), 0);\n\n // verify existing database still intact\n assertTrue(oldDbHelper.dbFileExists());\n assertEquals(oldDbHelper.getValue(\"device_id\"), \"oldDeviceId\");\n assertEquals(oldDbHelper.getLongValue(\"sequence_number\").longValue(), 1001L);\n assertEquals(oldDbHelper.getEventCount(), 1);\n assertEquals(oldDbHelper.getIdentifyCount(), 2);\n\n // verify both apps can modify their database independently and not affect old database\n newDbHelper1.insertOrReplaceKeyValue(\"device_id\", \"fakeDeviceId\");\n assertEquals(newDbHelper1.getValue(\"device_id\"), \"fakeDeviceId\");\n assertFalse(newDbHelper2.getValue(\"device_id\").equals(\"fakeDeviceId\"));\n assertEquals(oldDbHelper.getValue(\"device_id\"), \"oldDeviceId\");\n newDbHelper1.addIdentify(\"testIdentify3\");\n assertEquals(newDbHelper1.getIdentifyCount(), 1);\n assertEquals(newDbHelper2.getIdentifyCount(), 0);\n assertEquals(oldDbHelper.getIdentifyCount(), 2);\n\n newDbHelper2.insertOrReplaceKeyValue(\"device_id\", \"brandNewDeviceId\");\n assertEquals(newDbHelper1.getValue(\"device_id\"), \"fakeDeviceId\");\n assertEquals(newDbHelper2.getValue(\"device_id\"), \"brandNewDeviceId\");\n assertEquals(oldDbHelper.getValue(\"device_id\"), \"oldDeviceId\");\n newDbHelper2.addEvent(\"testEvent2\");\n newDbHelper2.addEvent(\"testEvent3\");\n assertEquals(newDbHelper1.getEventCount(), 0);\n assertEquals(newDbHelper2.getEventCount(), 2);\n assertEquals(oldDbHelper.getEventCount(), 1);\n }\n\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/BaseTest.java",
"content": "package com.amplitude.api;\n\nimport android.content.Context;\nimport android.content.pm.ApplicationInfo;\nimport android.content.pm.PackageInfo;\nimport android.content.pm.PackageManager;\nimport android.database.Cursor;\nimport android.database.sqlite.SQLiteDatabase;\n\nimport androidx.test.core.app.ApplicationProvider;\n\nimport org.json.JSONArray;\nimport org.json.JSONException;\nimport org.json.JSONObject;\nimport org.robolectric.RuntimeEnvironment;\nimport org.robolectric.shadows.ShadowLooper;\nimport org.robolectric.shadows.ShadowPackageManager;\n\nimport java.io.UnsupportedEncodingException;\nimport java.lang.reflect.Field;\nimport java.net.URLDecoder;\nimport java.util.LinkedHashMap;\nimport java.util.List;\nimport java.util.Map;\n\nimport okhttp3.mockwebserver.MockResponse;\nimport okhttp3.mockwebserver.MockWebServer;\nimport okhttp3.mockwebserver.RecordedRequest;\n\nimport static java.util.concurrent.TimeUnit.SECONDS;\nimport static org.junit.Assert.fail;\nimport static org.robolectric.Shadows.shadowOf;\n\npublic class BaseTest {\n\n protected class MockClock {\n int index = 0;\n long timestamps [];\n\n public void setTimestamps(long [] timestamps) {\n this.timestamps = timestamps;\n }\n\n public long currentTimeMillis() {\n if (timestamps == null || index >= timestamps.length) {\n return System.currentTimeMillis();\n }\n return timestamps[index++];\n }\n }\n\n // override getCurrentTimeMillis to enforce time progression in tests\n protected class AmplitudeClientWithTime extends AmplitudeClient {\n MockClock mockClock;\n\n public AmplitudeClientWithTime(MockClock mockClock) { this.mockClock = mockClock; }\n\n @Override\n protected long getCurrentTimeMillis() { return mockClock.currentTimeMillis(); }\n }\n\n // override AmplitudeDatabaseHelper to throw Cursor Allocation Exception\n protected class MockDatabaseHelper extends DatabaseHelper {\n\n protected MockDatabaseHelper(Context context) {\n super(context);\n }\n\n @Override\n Cursor queryDb(\n SQLiteDatabase db, String table, String[] columns, String selection,\n String[] selectionArgs, String groupBy, String having, String orderBy, String limit\n ) {\n // cannot import CursorWindowAllocationException, so we throw the base class instead\n throw new RuntimeException(\"Cursor window allocation of 2048 kb failed.\");\n }\n }\n\n private static final String TEST_PACKAGE_NAME = \"com.amplitude.test\";\n private static final String TEST_VERSION_NAME = \"test_version\";\n\n protected AmplitudeClient amplitude;\n protected Context context;\n protected MockWebServer server;\n protected MockClock clock;\n protected String apiKey = \"1cc2c1978ebab0f6451112a8f5df4f4e\";\n protected String[] instanceNames = {Constants.DEFAULT_INSTANCE, \"app1\", \"app2\", \"newApp1\", \"newApp2\", \"new_app\"};\n\n protected PackageManager packageManager;\n protected ShadowPackageManager shadowPackageManager;\n\n public void setUp() throws Exception {\n setUp(true);\n }\n\n /**\n * Handle common test setup for default cases. Specific cases can\n * override the defaults by providing an amplitude object before\n * calling this method or passing false for withServer.\n */\n public void setUp(boolean withServer) throws Exception {\n context = ApplicationProvider.getApplicationContext();\n packageManager = RuntimeEnvironment.application.getPackageManager();\n shadowPackageManager = shadowOf(packageManager);\n\n ApplicationInfo applicationInfo = new ApplicationInfo();\n applicationInfo.packageName = TEST_PACKAGE_NAME;\n\n PackageInfo packageInfo = new PackageInfo();\n packageInfo.packageName = TEST_PACKAGE_NAME;\n packageInfo.versionName = TEST_VERSION_NAME;\n packageInfo.applicationInfo = applicationInfo;\n shadowPackageManager.addPackage(packageInfo);\n\n // Clear the database helper for each test. Better to have isolation.\n // See https://github.com/robolectric/robolectric/issues/569\n // and https://github.com/robolectric/robolectric/issues/1622\n Amplitude.instances.clear();\n DatabaseHelper.instances.clear();\n\n if (withServer) {\n server = new MockWebServer();\n server.start();\n }\n\n if (clock == null) {\n clock = new MockClock();\n }\n\n if (amplitude == null) {\n amplitude = new AmplitudeClientWithTime(clock);\n // this sometimes deadlocks with lock contention by logThread and httpThread for\n // a ShadowWrangler instance and the ShadowLooper class\n // Might be a sign of a bug, or just Robolectric's bug.\n }\n\n if (server != null) {\n amplitude.url = server.url(\"/\").toString();\n }\n }\n\n public void tearDown() throws Exception {\n if (amplitude != null) {\n amplitude.logThread.getLooper().quit();\n amplitude.httpThread.getLooper().quit();\n amplitude = null;\n }\n\n if (server != null) {\n server.shutdown();\n }\n\n Amplitude.instances.clear();\n DatabaseHelper.instances.clear();\n }\n\n public RecordedRequest runRequest(AmplitudeClient amplitude) {\n server.enqueue(new MockResponse().setBody(\"success\"));\n ShadowLooper httpLooper = shadowOf(amplitude.httpThread.getLooper());\n httpLooper.runToEndOfTasks();\n\n try {\n return server.takeRequest(1, SECONDS);\n } catch (InterruptedException e) {\n return null;\n }\n }\n\n public RecordedRequest sendEvent(AmplitudeClient amplitude, String name, JSONObject props) {\n shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n amplitude.logEvent(name, props);\n shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n\n return runRequest(amplitude);\n }\n\n public RecordedRequest sendIdentify(AmplitudeClient amplitude, Identify identify) {\n shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n amplitude.identify(identify);\n shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n\n return runRequest(amplitude);\n }\n\n public long getUnsentEventCount() {\n return DatabaseHelper.getDatabaseHelper(context).getEventCount();\n }\n\n public long getUnsentIdentifyCount() {\n return DatabaseHelper.getDatabaseHelper(context).getIdentifyCount();\n }\n\n public long getIdentifyInterceptorCount() {\n return DatabaseHelper.getDatabaseHelper(context).getIdentifyInterceptorCount();\n }\n\n public JSONObject getLastUnsentEvent() {\n JSONArray events = getUnsentEventsFromTable(DatabaseHelper.EVENT_TABLE_NAME, 1);\n return (JSONObject)events.opt(events.length() - 1);\n }\n\n public JSONObject getLastUnsentIdentify() {\n JSONArray events = getUnsentEventsFromTable(DatabaseHelper.IDENTIFY_TABLE_NAME, 1);\n return (JSONObject)events.opt(events.length() - 1);\n }\n\n public JSONArray getUnsentEvents(int limit) {\n return getUnsentEventsFromTable(DatabaseHelper.EVENT_TABLE_NAME, limit);\n }\n\n public JSONArray getUnsentIdentifys(int limit) {\n return getUnsentEventsFromTable(DatabaseHelper.IDENTIFY_TABLE_NAME, limit);\n }\n\n public JSONArray getUnsentEventsFromTable(String table, int limit) {\n try {\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n List events = table.equals(DatabaseHelper.IDENTIFY_TABLE_NAME) ?\n dbHelper.getIdentifys(-1, -1) : dbHelper.getEvents(-1, -1);\n\n JSONArray out = new JSONArray();\n int start = Math.max(limit - events.size(), 0);\n for (int i = start; i < limit; i++) {\n out.put(i, events.get(events.size() - limit + i));\n }\n return out;\n } catch (JSONException e) {\n fail(e.toString());\n }\n\n return null;\n }\n\n public JSONObject getLastEvent() {\n return getLastEventFromTable(DatabaseHelper.EVENT_TABLE_NAME);\n }\n\n public JSONObject getLastIdentify() {\n return getLastEventFromTable(DatabaseHelper.IDENTIFY_TABLE_NAME);\n }\n\n public JSONObject getLastEventFromTable(String table) {\n try {\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n List events = table.equals(DatabaseHelper.IDENTIFY_TABLE_NAME) ?\n dbHelper.getIdentifys(-1, -1) : dbHelper.getEvents(-1, -1);\n return events.get(events.size() - 1);\n } catch (JSONException e) {\n fail(e.toString());\n }\n return null;\n }\n\n public JSONObject getLastIdentifyInterceptor() {\n try {\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n List events = dbHelper.getIdentifyInterceptors(-1, -1);\n return events.get(events.size() - 1);\n } catch (JSONException e) {\n fail(e.toString());\n }\n return null;\n }\n\n public JSONArray getEventsFromRequest(RecordedRequest request) throws JSONException {\n Map parsedBody = parseRequest(request.getUtf8Body());\n if (parsedBody == null && !parsedBody.containsKey(\"e\")) {\n return null;\n }\n return new JSONArray(parsedBody.get(\"e\"));\n }\n\n // parse request string into a key:value map\n public static Map parseRequest(String request) {\n try {\n Map query_pairs = new LinkedHashMap();\n String[] pairs = request.split(\"&\");\n for (String pair : pairs) {\n int idx = pair.indexOf(\"=\");\n query_pairs.put(URLDecoder.decode(pair.substring(0, idx), \"UTF-8\"), URLDecoder.decode(pair.substring(idx + 1), \"UTF-8\"));\n }\n return query_pairs;\n } catch (UnsupportedEncodingException e) {\n fail(e.toString());\n }\n return null;\n }\n\n protected Object getPrivateFieldValueFromClient(AmplitudeClient client, String field) {\n try {\n Field privateField = AmplitudeClient.class.getDeclaredField(field);\n privateField.setAccessible(true);\n return privateField.get(client);\n } catch (IllegalAccessException e) {\n fail(e.toString());\n } catch (NoSuchFieldException e) {\n fail(e.toString());\n }\n return null;\n }\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/ConfigManagerTest.java",
"content": "package com.amplitude.api;\n\n\nimport androidx.test.ext.junit.runners.AndroidJUnit4;\n\nimport com.amplitude.api.util.MockHttpURLConnectionHelper;\nimport com.amplitude.api.util.MockURLStreamHandler;\n\nimport org.json.JSONObject;\nimport org.junit.Test;\nimport org.junit.runner.RunWith;\nimport org.robolectric.annotation.Config;\n\nimport java.net.HttpURLConnection;\nimport java.net.URL;\n\nimport static org.junit.Assert.assertEquals;\n\n@RunWith(AndroidJUnit4.class)\n@Config(manifest=Config.NONE)\npublic class ConfigManagerTest {\n private final MockURLStreamHandler mockURLStreamHandler = MockURLStreamHandler.getInstance();\n\n @Test\n public void testRefreshForEU() throws Exception {\n assertEquals(Constants.EVENT_LOG_URL, ConfigManager.getInstance().getIngestionEndpoint());\n JSONObject responseObject = new JSONObject();\n responseObject.put(\"code\", 200);\n responseObject.put(\"ingestionEndpoint\", \"api.eu.amplitude.com\");\n\n URL url = new URL(Constants.DYNAMIC_CONFIG_EU_URL);\n AmplitudeServerZone euZone = AmplitudeServerZone.EU;\n HttpURLConnection connection =\n MockHttpURLConnectionHelper.getMockHttpURLConnection(200, responseObject.toString());\n mockURLStreamHandler.setConnection(url, connection);\n\n ConfigManager.getInstance().refresh(new ConfigManager.RefreshListener() {\n @Override\n public void onFinished() {\n assertEquals(Constants.EVENT_LOG_EU_URL, ConfigManager.getInstance().getIngestionEndpoint() + \"/\");\n }\n }, euZone);\n }\n}"
},
{
"path": "src/test/java/com/amplitude/api/DatabaseHelperTest.java",
"content": "package com.amplitude.api;\n\nimport androidx.test.ext.junit.runners.AndroidJUnit4;\n\nimport org.json.JSONException;\nimport org.json.JSONObject;\nimport org.junit.After;\nimport org.junit.Before;\nimport org.junit.Test;\nimport org.junit.runner.RunWith;\nimport org.robolectric.annotation.Config;\n\nimport java.util.List;\n\nimport static org.junit.Assert.assertEquals;\nimport static org.junit.Assert.assertNotSame;\nimport static org.junit.Assert.assertNull;\nimport static org.junit.Assert.assertSame;\nimport static org.junit.Assert.assertTrue;\nimport static org.junit.Assert.fail;\n\n@RunWith(AndroidJUnit4.class)\n@Config(manifest = Config.NONE)\npublic class DatabaseHelperTest extends BaseTest {\n\n protected DatabaseHelper dbInstance;\n\n @Before\n public void setUp() throws Exception {\n super.setUp(false);\n dbInstance = DatabaseHelper.getDatabaseHelper(context);\n }\n\n @After\n public void tearDown() throws Exception {\n super.tearDown();\n dbInstance = null;\n }\n\n protected long addEvent(String type) {\n return addEventToTable(DatabaseHelper.EVENT_TABLE_NAME, type, new JSONObject());\n }\n\n protected long addEventToTable(String table, String type, JSONObject props) {\n try {\n props.put(\"event_type\", type);\n return table.equals(DatabaseHelper.IDENTIFY_TABLE_NAME) ?\n dbInstance.addIdentify(props.toString()) :\n dbInstance.addEvent(props.toString());\n } catch (JSONException e) {\n fail(e.toString());\n }\n return -1;\n }\n\n protected long addIdentify(String identifyEvent) {\n return addEventToTable(DatabaseHelper.IDENTIFY_TABLE_NAME, identifyEvent, new JSONObject());\n }\n\n protected long addIdentifyInterceptor(String property) {\n try {\n JSONObject props = new JSONObject();\n props.put(\"user_properties\", new JSONObject());\n props.getJSONObject(\"user_properties\").put(\"test_prop\", property);\n props.put(\"event_type\", Constants.IDENTIFY_EVENT);\n return dbInstance.addIdentifyInterceptor(props.toString());\n } catch (JSONException e) {\n fail(e.toString());\n }\n return -1;\n }\n\n protected long insertOrReplaceKeyValue(String key, String value) {\n return dbInstance.insertOrReplaceKeyValue(key, value);\n }\n\n protected long insertOrReplaceKeyLongValue(String key, Long value) {\n return dbInstance.insertOrReplaceKeyLongValue(key, value);\n }\n\n protected String getValue(String key) {\n return dbInstance.getValue(key);\n }\n\n protected Long getLongValue(String key) { return dbInstance.getLongValue(key); }\n\n @Test\n public void testCreate() {\n dbInstance.onCreate(dbInstance.getWritableDatabase());\n assertEquals(1, insertOrReplaceKeyValue(\"test_key\", \"test_value\"));\n assertEquals(1, insertOrReplaceKeyLongValue(\"test_key\", 1L));\n assertEquals(1, addEvent(\"test_create\"));\n assertEquals(1, addIdentify(\"test_create\"));\n assertEquals(1, addIdentifyInterceptor(\"test_create\"));\n }\n\n // need separate tests for different version to version upgrades since insertion failure\n // triggers database deletion and recreation - need to refetch writable database as well\n @Test\n public void testUpgradeVersion1ToVersion2() {\n // store table doesn't exist in v1, insert will fail\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.STORE_TABLE_NAME);\n String key = \"test_key\";\n String value = \"test_value\";\n assertEquals(-1, insertOrReplaceKeyValue(key, value));\n\n // long store table doesn't exist in v1, insert will fail\n Long longValue = 1L;\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.LONG_STORE_TABLE_NAME);\n assertEquals(-1, insertOrReplaceKeyLongValue(key, longValue));\n\n // identify table doesn't exist in v1, insert will fail\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_TABLE_NAME);\n assertEquals(-1, addIdentify(\"test_upgrade\"));\n\n // identify interceptor table doesn't exist in v1, insert will fail\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_INTERCEPTOR_TABLE_NAME);\n assertEquals(-1, addIdentifyInterceptor(\"test_upgrade\"));\n\n // only event inserts will work\n assertEquals(1, addEvent(\"test_upgrade\"));\n\n // after v2 upgrade, can insert into store table\n // still can't insert into identify table, long store table or identify interceptor table\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.STORE_TABLE_NAME);\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_TABLE_NAME);\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.LONG_STORE_TABLE_NAME);\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_INTERCEPTOR_TABLE_NAME);\n dbInstance.onUpgrade(dbInstance.getWritableDatabase(), 1, 2);\n assertEquals(2, addEvent(\"test_upgrade\"));\n assertEquals(1, insertOrReplaceKeyValue(key, value));\n assertEquals(-1, addIdentify(\"test_upgrade\"));\n assertEquals(-1, addIdentifyInterceptor(\"test_upgrade\"));\n assertEquals(-1, insertOrReplaceKeyLongValue(key, longValue));\n }\n\n @Test\n public void testUpgradeVersion2ToVersion3() {\n // identify table doesn't exist in v2, insert will fail\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_TABLE_NAME);\n assertEquals(-1, addIdentify(\"test_upgrade\"));\n\n // long store table doesn't exist in v2, insert will fail\n String key = \"test_key\";\n Long longValue = 1L;\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.LONG_STORE_TABLE_NAME);\n assertEquals(-1, insertOrReplaceKeyLongValue(key, longValue));\n\n // identify interceptor table doesn't exist in v2, insert will fail\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_INTERCEPTOR_TABLE_NAME);\n assertEquals(-1, addIdentifyInterceptor(\"test_upgrade\"));\n\n // events and store inserts will work\n String value = \"test_value\";\n assertEquals(1, insertOrReplaceKeyValue(key, value));\n assertEquals(1, addEvent(\"test_upgrade\"));\n\n // after v3 upgrade, can insert into identify table and long store\n // cannot insert into identify interceptor table\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_TABLE_NAME);\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_INTERCEPTOR_TABLE_NAME);\n dbInstance.onUpgrade(dbInstance.getWritableDatabase(), 2, 3);\n assertEquals(2, addEvent(\"test_upgrade\"));\n assertEquals(2, insertOrReplaceKeyValue(key, value));\n assertEquals(1, addIdentify(\"test_upgrade\"));\n assertEquals(1, insertOrReplaceKeyLongValue(key, longValue));\n assertEquals(-1, addIdentifyInterceptor(\"test_upgrade\"));\n }\n\n @Test\n public void testUpgradeVersion1ToVersion3() {\n // store table doesn't exist in v1, insert will fail\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.STORE_TABLE_NAME);\n String key = \"test_key\";\n String value = \"test_value\";\n assertEquals(-1, insertOrReplaceKeyValue(key, value));\n\n // identify table doesn't exist in v1, insert will fail\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_TABLE_NAME);\n assertEquals(-1, addIdentify(\"test_upgrade\"));\n\n // long store table doesn't exist in v1, insert will fail\n Long longValue = 1L;\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.LONG_STORE_TABLE_NAME);\n assertEquals(-1, insertOrReplaceKeyLongValue(key, longValue));\n\n // identify interceptor table doesn't exist in v1, insert will fail\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_INTERCEPTOR_TABLE_NAME);\n assertEquals(-1, addIdentifyInterceptor(\"test_upgrade\"));\n\n // only event inserts will work\n assertEquals(1, addEvent(\"test_upgrade\"));\n\n // after v3 upgrade, can insert into store and identify tables\n // cannot insert into identify interceptor table\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.STORE_TABLE_NAME);\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_TABLE_NAME);\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.LONG_STORE_TABLE_NAME);\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_INTERCEPTOR_TABLE_NAME);\n dbInstance.onUpgrade(dbInstance.getWritableDatabase(), 1, 3);\n assertEquals(2, addEvent(\"test_upgrade\"));\n assertEquals(1, insertOrReplaceKeyValue(key, value));\n assertEquals(1, addIdentify(\"test_upgrade\"));\n assertEquals(1, insertOrReplaceKeyLongValue(key, longValue));\n assertEquals(-1, addIdentifyInterceptor(\"test_upgrade\"));\n }\n\n @Test\n public void testUpgradeVersion1ToVersion4() {\n // store table doesn't exist in v1, insert will fail\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.STORE_TABLE_NAME);\n String key = \"test_key\";\n String value = \"test_value\";\n assertEquals(-1, insertOrReplaceKeyValue(key, value));\n\n // identify table doesn't exist in v1, insert will fail\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_TABLE_NAME);\n assertEquals(-1, addIdentify(\"test_upgrade\"));\n\n // long store table doesn't exist in v1, insert will fail\n Long longValue = 1L;\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.LONG_STORE_TABLE_NAME);\n assertEquals(-1, insertOrReplaceKeyLongValue(key, longValue));\n\n // identify interceptor table doesn't exist in v1, insert will fail\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_INTERCEPTOR_TABLE_NAME);\n assertEquals(-1, addIdentifyInterceptor(\"test_upgrade\"));\n\n // only event inserts will work\n assertEquals(1, addEvent(\"test_upgrade\"));\n\n // after v4 upgrade, can insert into store, identify, and identify interceptor tables\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.STORE_TABLE_NAME);\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_TABLE_NAME);\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.LONG_STORE_TABLE_NAME);\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_INTERCEPTOR_TABLE_NAME);\n dbInstance.onUpgrade(dbInstance.getWritableDatabase(), 1, 4);\n assertEquals(2, addEvent(\"test_upgrade\"));\n assertEquals(1, insertOrReplaceKeyValue(key, value));\n assertEquals(1, addIdentify(\"test_upgrade\"));\n assertEquals(1, insertOrReplaceKeyLongValue(key, longValue));\n assertEquals(1, addIdentifyInterceptor(\"test_upgrade\"));\n }\n\n @Test\n public void testUpgradeVersion2ToVersion4() {\n // identify table doesn't exist in v2, insert will fail\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_TABLE_NAME);\n assertEquals(-1, addIdentify(\"test_upgrade\"));\n\n // long store table doesn't exist in v2, insert will fail\n String key = \"test_key\";\n Long longValue = 1L;\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.LONG_STORE_TABLE_NAME);\n assertEquals(-1, insertOrReplaceKeyLongValue(key, longValue));\n\n // identify interceptor table doesn't exist in v2, insert will fail\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_INTERCEPTOR_TABLE_NAME);\n assertEquals(-1, addIdentifyInterceptor(\"test_upgrade\"));\n\n // events and store inserts will work\n String value = \"test_value\";\n assertEquals(1, insertOrReplaceKeyValue(key, value));\n assertEquals(1, addEvent(\"test_upgrade\"));\n\n // after v4 upgrade, can insert into identify table, long store and identify interceptor table\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_TABLE_NAME);\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_INTERCEPTOR_TABLE_NAME);\n dbInstance.onUpgrade(dbInstance.getWritableDatabase(), 2, 4);\n assertEquals(2, addEvent(\"test_upgrade\"));\n assertEquals(2, insertOrReplaceKeyValue(key, value));\n assertEquals(1, addIdentify(\"test_upgrade\"));\n assertEquals(1, insertOrReplaceKeyLongValue(key, longValue));\n assertEquals(1, addIdentifyInterceptor(\"test_upgrade\"));\n }\n\n @Test\n public void testUpgradeVersion3ToVersion4() {\n // identify interceptor table doesn't exist in v3, insert will fail\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_INTERCEPTOR_TABLE_NAME);\n assertEquals(-1, addIdentifyInterceptor(\"test_upgrade\"));\n\n String key = \"test_key\";\n Long longValue = 1L;\n // events, store and identify inserts will work\n String value = \"test_value\";\n assertEquals(1, insertOrReplaceKeyValue(key, value));\n assertEquals(1, addEvent(\"test_upgrade\"));\n assertEquals(1, addIdentify(\"test_upgrade\"));\n\n // after v4 upgrade, can insert into identify interceptor table\n dbInstance.getWritableDatabase().execSQL(\n \"DROP TABLE IF EXISTS \" + DatabaseHelper.IDENTIFY_INTERCEPTOR_TABLE_NAME);\n dbInstance.onUpgrade(dbInstance.getWritableDatabase(), 3, 4);\n assertEquals(2, addEvent(\"test_upgrade\"));\n assertEquals(2, insertOrReplaceKeyValue(key, value));\n assertEquals(2, addIdentify(\"test_upgrade\"));\n assertEquals(1, insertOrReplaceKeyLongValue(key, longValue));\n }\n\n @Test\n public void testInsertOrReplaceKeyValue() {\n String key = \"test_key\";\n String value1 = \"test_value1\";\n String value2 = \"test_value2\";\n assertEquals(null, getValue(key));\n\n insertOrReplaceKeyValue(key, value1);\n assertEquals(value1, getValue(key));\n\n insertOrReplaceKeyValue(key, value2);\n assertEquals(value2, getValue(key));\n }\n\n @Test\n public void testInsertOrReplaceKeyLongValue() {\n String key = \"test_key\";\n Long value1 = 1L;\n Long value2 = 2L;\n assertEquals(null, getLongValue(key));\n\n insertOrReplaceKeyLongValue(key, value1);\n assertEquals(value1, getLongValue(key));\n\n insertOrReplaceKeyLongValue(key, value2);\n assertEquals(value2, getLongValue(key));\n }\n\n @Test\n public void testInsertNullValues() {\n String key = \"test_key\";\n\n assertNull(getValue(key));\n insertOrReplaceKeyValue(key, \"test\");\n assertEquals(getValue(key), \"test\");\n insertOrReplaceKeyValue(key, null);\n assertNull(getValue(key));\n\n assertNull(getLongValue(key));\n insertOrReplaceKeyLongValue(key, 15L);\n assertEquals((long)getLongValue(key), 15L);\n insertOrReplaceKeyLongValue(key, null);\n assertNull(getValue(key));\n }\n\n @Test\n public void testAddEvent() {\n assertEquals(1, addEvent(\"test_add_event\"));\n assertEquals(1, getLastUnsentEvent().optLong(\"event_id\"));\n assertEquals(2, addEvent(\"test_add_event\"));\n assertEquals(2, getLastUnsentEvent().optLong(\"event_id\"));\n assertEquals(3, addEvent(\"test_add_event\"));\n assertEquals(3, getLastUnsentEvent().optLong(\"event_id\"));\n }\n\n @Test\n public void testAddIdentify() {\n assertEquals(1, addIdentify(\"test_add_identify\"));\n assertEquals(1, getLastUnsentIdentify().optLong(\"event_id\"));\n assertEquals(2, addIdentify(\"test_add_identify\"));\n assertEquals(2, getLastUnsentIdentify().optLong(\"event_id\"));\n assertEquals(3, addIdentify(\"test_add_identify\"));\n assertEquals(3, getLastUnsentIdentify().optLong(\"event_id\"));\n }\n\n @Test\n public void testGetEvents() {\n assertEquals(1, addEvent(\"test_get_events_1\"));\n assertEquals(2, addEvent(\"test_get_events_2\"));\n assertEquals(3, addEvent(\"test_get_events_3\"));\n assertEquals(4, addEvent(\"test_get_events_4\"));\n assertEquals(5, addEvent(\"test_get_events_5\"));\n\n try {\n List events;\n assertEquals(5, dbInstance.getEventCount());\n\n events = dbInstance.getEvents(-1, -1);\n assertEquals(5, events.size());\n assertEquals(1, (events.get(0)).getLong(\"event_id\"));\n assertEquals(\"test_get_events_1\", (events.get(0)).getString(\"event_type\"));\n\n events = dbInstance.getEvents(1, -1);\n assertEquals(1, events.size());\n\n events = dbInstance.getEvents(5, -1);\n assertEquals(5, events.size());\n assertEquals(5, (events.get(4)).getLong(\"event_id\"));\n assertEquals(\"test_get_events_5\", (events.get(4)).getString(\"event_type\"));\n\n events = dbInstance.getEvents(-1, 0);\n assertEquals(0, events.size());\n\n events = dbInstance.getEvents(-1, 1);\n assertEquals(1, events.size());\n assertEquals(1, (events.get(0)).getLong(\"event_id\"));\n assertEquals(\"test_get_events_1\", (events.get(0)).getString(\"event_type\"));\n\n events = dbInstance.getEvents(5, 1);\n assertEquals(1, events.size());\n assertEquals(1, (events.get(0)).getLong(\"event_id\"));\n assertEquals(\"test_get_events_1\", (events.get(0)).getString(\"event_type\"));\n\n dbInstance.removeEvent(1);\n events = dbInstance.getEvents(5, 1);\n assertEquals(1, events.size());\n assertEquals(2, (events.get(0)).getLong(\"event_id\"));\n assertEquals(\"test_get_events_2\", (events.get(0)).getString(\"event_type\"));\n\n dbInstance.removeEvents(3);\n events = dbInstance.getEvents(5, 1);\n assertEquals(1, events.size());\n assertEquals(4, (events.get(0)).getLong(\"event_id\"));\n assertEquals(\"test_get_events_4\", (events.get(0)).getString(\"event_type\"));\n\n } catch (JSONException e) {\n fail(e.toString());\n }\n }\n\n @Test\n public void testGetIdentifys() {\n assertEquals(1, addIdentify(\"test_get_identifys_1\"));\n assertEquals(2, addIdentify(\"test_get_identifys_2\"));\n assertEquals(3, addIdentify(\"test_get_identifys_3\"));\n assertEquals(4, addIdentify(\"test_get_identifys_4\"));\n assertEquals(5, addIdentify(\"test_get_identifys_5\"));\n\n try {\n List events;\n assertEquals(5, dbInstance.getIdentifyCount());\n\n events = dbInstance.getIdentifys(-1, -1);\n assertEquals(5, events.size());\n assertEquals(1, (events.get(0)).getLong(\"event_id\"));\n assertEquals(\"test_get_identifys_1\", (events.get(0)).getString(\"event_type\"));\n\n events = dbInstance.getIdentifys(1, -1);\n assertEquals(1, events.size());\n\n events = dbInstance.getIdentifys(5, -1);\n assertEquals(5, events.size());\n assertEquals(5, (events.get(4)).getLong(\"event_id\"));\n assertEquals(\"test_get_identifys_5\", (events.get(4)).getString(\"event_type\")\n );\n\n events = dbInstance.getIdentifys(-1, 0);\n assertEquals(0, events.size());\n\n events = dbInstance.getIdentifys(-1, 1);\n assertEquals(1, events.size());\n assertEquals(1, (events.get(0)).getLong(\"event_id\"));\n assertEquals(\"test_get_identifys_1\", (events.get(0)).getString(\"event_type\"));\n\n events = dbInstance.getIdentifys(5, 1);\n assertEquals(1, events.size());\n assertEquals(1, (events.get(0)).getLong(\"event_id\"));\n assertEquals(\"test_get_identifys_1\", (events.get(0)).getString(\"event_type\"));\n\n dbInstance.removeIdentify(1);\n events = dbInstance.getIdentifys(5, 1);\n assertEquals(1, events.size());\n assertEquals(2, (events.get(0)).getLong(\"event_id\"));\n assertEquals(\"test_get_identifys_2\", (events.get(0)).getString(\"event_type\")\n );\n\n dbInstance.removeIdentifys(3);\n events = dbInstance.getIdentifys(5, 1);\n assertEquals(1, events.size());\n assertEquals(4, (events.get(0)).getLong(\"event_id\"));\n assertEquals(\"test_get_identifys_4\", (events.get(0)).getString(\"event_type\")\n );\n } catch (JSONException e) {\n fail(e.toString());\n }\n }\n\n @Test\n public void testGetEventCount() {\n assertEquals(1, addEvent(\"test_get_event_count_1\"));\n assertEquals(2, addEvent(\"test_get_event_count_2\"));\n assertEquals(3, addEvent(\"test_get_event_count_3\"));\n assertEquals(4, addEvent(\"test_get_event_count_4\"));\n assertEquals(5, addEvent(\"test_get_event_count_5\"));\n\n assertEquals(5, dbInstance.getEventCount());\n\n dbInstance.removeEvent(1);\n assertEquals(4, dbInstance.getEventCount());\n\n dbInstance.removeEvents(3);\n assertEquals(2, dbInstance.getEventCount());\n\n dbInstance.removeEvents(10);\n assertEquals(0, dbInstance.getEventCount());\n }\n\n @Test\n public void testGetIdentifyCount() {\n assertEquals(1, addIdentify(\"test_get_identify_count_1\"));\n assertEquals(2, addIdentify(\"test_get_identify_count_2\"));\n assertEquals(3, addIdentify(\"test_get_identify_count_3\"));\n assertEquals(4, addIdentify(\"test_get_identify_count_4\"));\n assertEquals(5, addIdentify(\"test_get_identify_count_5\"));\n\n assertEquals(5, dbInstance.getIdentifyCount());\n\n dbInstance.removeIdentify(1);\n assertEquals(4, dbInstance.getIdentifyCount());\n\n dbInstance.removeIdentifys(3);\n assertEquals(2, dbInstance.getIdentifyCount());\n\n dbInstance.removeIdentifys(10);\n assertEquals(0, dbInstance.getIdentifyCount());\n }\n\n @Test\n public void testGetNthEventId() {\n assertEquals(1, addEvent(\"test_get_nth_event_id_1\"));\n assertEquals(2, addEvent(\"test_get_nth_event_id_2\"));\n assertEquals(3, addEvent(\"test_get_nth_event_id_3\"));\n assertEquals(4, addEvent(\"test_get_nth_event_id_4\"));\n assertEquals(5, addEvent(\"test_get_nth_event_id_5\"));\n\n assertEquals(1, dbInstance.getNthEventId(0));\n assertEquals(1, dbInstance.getNthEventId(1));\n assertEquals(2, dbInstance.getNthEventId(2));\n assertEquals(3, dbInstance.getNthEventId(3));\n assertEquals(4, dbInstance.getNthEventId(4));\n assertEquals(5, dbInstance.getNthEventId(5));\n\n dbInstance.removeEvent(1);\n assertEquals(2, dbInstance.getNthEventId(1));\n\n dbInstance.removeEvents(3);\n assertEquals(4, dbInstance.getNthEventId(1));\n\n dbInstance.removeEvents(10);\n assertEquals(-1, dbInstance.getNthEventId(1));\n }\n\n @Test\n public void testGetNthIdentifyId() {\n assertEquals(1, addIdentify(\"test_get_nth_identify_id_1\"));\n assertEquals(2, addIdentify(\"test_get_nth_identify_id_2\"));\n assertEquals(3, addIdentify(\"test_get_nth_identify_id_3\"));\n assertEquals(4, addIdentify(\"test_get_nth_identify_id_4\"));\n assertEquals(5, addIdentify(\"test_get_nth_identify_id_5\"));\n\n assertEquals(1, dbInstance.getNthIdentifyId(0));\n assertEquals(1, dbInstance.getNthIdentifyId(1));\n assertEquals(2, dbInstance.getNthIdentifyId(2));\n assertEquals(3, dbInstance.getNthIdentifyId(3));\n assertEquals(4, dbInstance.getNthIdentifyId(4));\n assertEquals(5, dbInstance.getNthIdentifyId(5));\n\n dbInstance.removeIdentify(1);\n assertEquals(2, dbInstance.getNthIdentifyId(1));\n\n dbInstance.removeIdentifys(3);\n assertEquals(4, dbInstance.getNthIdentifyId(1));\n\n dbInstance.removeIdentifys(10);\n assertEquals(-1, dbInstance.getNthIdentifyId(1));\n }\n\n @Test\n public void testNoConflictBetweenEventsAndIdentifys() {\n assertEquals(1, addEvent(\"test_add_event_id_1\"));\n assertEquals(2, addEvent(\"test_add_event_id_2\"));\n assertEquals(3, addEvent(\"test_add_event_id_3\"));\n assertEquals(4, addEvent(\"test_add_event_id_4\"));\n assertEquals(4, dbInstance.getEventCount());\n assertEquals(0, dbInstance.getIdentifyCount());\n\n assertEquals(1, addIdentify(\"test_add_identify_id_1\"));\n assertEquals(2, addIdentify(\"test_add_identify_id_2\"));\n assertEquals(4, dbInstance.getEventCount());\n assertEquals(2, dbInstance.getIdentifyCount());\n\n dbInstance.removeEvent(1);\n assertEquals(3, dbInstance.getEventCount());\n assertEquals(2, dbInstance.getIdentifyCount());\n\n dbInstance.removeIdentify(1);\n assertEquals(3, dbInstance.getEventCount());\n assertEquals(1, dbInstance.getIdentifyCount());\n\n dbInstance.removeEvents(4);\n assertEquals(0, dbInstance.getEventCount());\n assertEquals(1, dbInstance.getIdentifyCount());\n }\n\n @Test\n public void testNullEventString() throws JSONException {\n dbInstance.addEvent(null);\n List events = dbInstance.getEvents(-1, -1);\n assertTrue(events.isEmpty());\n }\n\n @Test\n public void testGetDatabaseHelper() {\n assertEquals(DatabaseHelper.instances.size(), 1);\n DatabaseHelper oldDbHelper = DatabaseHelper.getDatabaseHelper(context);\n assertEquals(oldDbHelper.getEventCount(), 0); // run query to initialize db file\n\n assertSame(oldDbHelper, DatabaseHelper.getDatabaseHelper(context, null));\n assertSame(oldDbHelper, DatabaseHelper.getDatabaseHelper(context, \"\"));\n assertSame(\n oldDbHelper, DatabaseHelper.getDatabaseHelper(context, Constants.DEFAULT_INSTANCE)\n );\n DatabaseHelper a = DatabaseHelper.getDatabaseHelper(context, \"a\");\n DatabaseHelper b = DatabaseHelper.getDatabaseHelper(context, \"b\");\n assertNotSame(oldDbHelper, a);\n assertNotSame(oldDbHelper, b);\n assertNotSame(a, b);\n assertSame(a, DatabaseHelper.getDatabaseHelper(context, \"a\"));\n assertSame(b, DatabaseHelper.getDatabaseHelper(context, \"b\"));\n\n assertEquals(DatabaseHelper.instances.size(), 3);\n assertTrue(DatabaseHelper.instances.containsKey(Constants.DEFAULT_INSTANCE));\n assertTrue(DatabaseHelper.instances.containsKey(\"a\"));\n assertTrue(DatabaseHelper.instances.containsKey(\"b\"));\n\n // test for instance name case insensitivity\n assertSame(a, DatabaseHelper.getDatabaseHelper(context, \"A\"));\n assertSame(b, DatabaseHelper.getDatabaseHelper(context, \"B\"));\n assertSame(\n oldDbHelper,\n DatabaseHelper.getDatabaseHelper(context, Constants.DEFAULT_INSTANCE.toUpperCase())\n );\n\n // assert defaultInstance maintains old database filename while new instances have new names\n a.addEvent(\"testEvent1\");\n b.addEvent(\"testEvent2\");\n assertTrue(context.getDatabasePath(Constants.DATABASE_NAME).exists());\n assertTrue(context.getDatabasePath(Constants.DATABASE_NAME + \"_a\").exists());\n assertTrue(context.getDatabasePath(Constants.DATABASE_NAME + \"_b\").exists());\n }\n\n @Test\n public void testSeparateInstances() {\n DatabaseHelper dbHelper1 = DatabaseHelper.getDatabaseHelper(context, \"a\");\n DatabaseHelper dbHelper2 = DatabaseHelper.getDatabaseHelper(context, \"b\");\n DatabaseHelper dbHelper3 = DatabaseHelper.getDatabaseHelper(context, \"c\");\n\n dbHelper1.insertOrReplaceKeyValue(\"device_id\", \"testDeviceId\");\n assertEquals(dbHelper1.getValue(\"device_id\"), \"testDeviceId\");\n assertNull(dbHelper2.getValue(\"device_id\"));\n assertNull(dbHelper3.getValue(\"device_id\"));\n\n dbHelper2.addEvent(\"test_event\");\n assertEquals(dbHelper1.getEventCount(), 0);\n assertEquals(dbHelper2.getEventCount(), 1);\n assertEquals(dbHelper3.getEventCount(), 0);\n\n dbHelper3.addIdentify(\"test_identify_1\");\n assertEquals(dbHelper1.getIdentifyCount(), 0);\n assertEquals(dbHelper2.getIdentifyCount(), 0);\n assertEquals(dbHelper3.getIdentifyCount(), 1);\n }\n\n @Test\n public void testGetLastIdentifyInterceptorId() {\n assertEquals(1, addIdentifyInterceptor(\"test_get_last_identify_id_1\"));\n assertEquals(2, addIdentifyInterceptor(\"test_get_last_identify_id_2\"));\n\n assertEquals(2, dbInstance.getLastIdentifyInterceptorId());\n dbInstance.removeIdentifyInterceptors(2);\n assertEquals(-1, dbInstance.getLastIdentifyInterceptorId());\n\n assertEquals(3, addIdentifyInterceptor(\"test_get_last_identify_id_3\"));\n assertEquals(3, dbInstance.getLastIdentifyInterceptorId());\n }\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/DatabaseRecoveryTest.java",
"content": "package com.amplitude.api;\n\nimport android.content.ContentValues;\nimport android.database.sqlite.SQLiteDatabase;\nimport android.database.sqlite.SQLiteException;\n\nimport androidx.test.ext.junit.runners.AndroidJUnit4;\n\nimport org.json.JSONException;\nimport org.json.JSONObject;\nimport org.junit.After;\nimport org.junit.Before;\nimport org.junit.Test;\nimport org.junit.runner.RunWith;\nimport org.mockito.Matchers;\nimport org.powermock.api.mockito.PowerMockito;\nimport org.robolectric.Robolectric;\nimport org.robolectric.Shadows;\nimport org.robolectric.annotation.Config;\nimport org.robolectric.shadows.ShadowLooper;\n\nimport java.io.File;\nimport java.io.IOException;\nimport java.io.RandomAccessFile;\nimport java.util.List;\n\nimport static org.junit.Assert.assertEquals;\nimport static org.junit.Assert.assertNotNull;\nimport static org.junit.Assert.assertNull;\nimport static org.junit.Assert.assertTrue;\nimport static org.mockito.Matchers.anyString;\nimport static org.mockito.Mockito.reset;\n\n@RunWith(AndroidJUnit4.class)\n@Config(manifest = Config.NONE)\npublic class DatabaseRecoveryTest extends BaseTest {\n\n protected DatabaseHelper dbInstance;\n protected ShadowLooper looper;\n protected long startTime;\n\n @Before\n public void setUp() throws Exception {\n super.setUp(false);\n\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n startTime = System.currentTimeMillis();\n\n amplitude.setEventUploadPeriodMillis(10*60*1000);\n amplitude.initialize(context, apiKey, null, null, true);\n\n looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runOneTask();\n dbInstance = DatabaseHelper.getDatabaseHelper(context);\n }\n\n @After\n public void tearDown() throws Exception {\n super.tearDown();\n\n DatabaseHelper.instances.put(Constants.DEFAULT_INSTANCE, null);\n dbInstance = null;\n }\n\n @Test\n public void testRecoverFromDatabaseReset() {\n\n // log an event normally, verify metadata updated in table\n amplitude.logEvent(\"test\");\n looper.runToEndOfTasks();\n\n // metadata: deviceId, sessionId, sequence number, last_event_id, last_event_time, previous_session_id\n assertEquals(dbInstance.getEventCount(), 1);\n assertEquals(dbInstance.getTotalEventCount(), 1);\n assertEquals(dbInstance.getNthEventId(1), 1);\n\n String deviceId = dbInstance.getValue(AmplitudeClient.DEVICE_ID_KEY);\n long previousSessionId = dbInstance.getLongValue(AmplitudeClient.PREVIOUS_SESSION_ID_KEY);\n long sequenceNumber = dbInstance.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY);\n long lastEventId = dbInstance.getLongValue(AmplitudeClient.LAST_EVENT_ID_KEY);\n long lastEventTime = dbInstance.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY);\n Long lastIdentifyId = dbInstance.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY);\n\n assertNotNull(deviceId);\n assertTrue(deviceId.endsWith(\"R\"));\n assertTrue(previousSessionId > 0);\n assertEquals(sequenceNumber, 1);\n assertEquals(lastEventId, 1);\n assertTrue(lastEventTime >= startTime);\n assertNull(lastIdentifyId);\n\n // difficult to mock out the SQLiteDatabase object inside DatabaseHelper since it's private\n // add helper method specifically for mocking / testing\n DatabaseHelper mockDbHelper = PowerMockito.spy(dbInstance);\n PowerMockito.doThrow(new SQLiteException(\"test\")).when(mockDbHelper).insertEventContentValuesIntoTable(Matchers.any(SQLiteDatabase.class), anyString(), Matchers.any(ContentValues.class));\n amplitude.dbHelper = mockDbHelper;\n\n // log an event to trigger SQLException that we set up with mocks\n amplitude.logEvent(\"test\");\n looper.runToEndOfTasks();\n\n // verify that the metadata has been persisted back into database\n String newDeviceId = dbInstance.getValue(AmplitudeClient.DEVICE_ID_KEY);\n long newPreviousSessionId = dbInstance.getLongValue(AmplitudeClient.PREVIOUS_SESSION_ID_KEY);\n long newLastEventTime = dbInstance.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY);\n\n // these do not get persisted and should be reset\n Long newSequenceNumber = dbInstance.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY);\n Long newLastEventId = dbInstance.getLongValue(AmplitudeClient.LAST_EVENT_ID_KEY);\n Long newLastIdentifyId = dbInstance.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY);\n\n assertEquals(newDeviceId, deviceId);\n assertEquals(newPreviousSessionId, previousSessionId);\n assertTrue(newLastEventTime >= lastEventTime);\n\n assertNull(newSequenceNumber);\n assertEquals(newLastEventId, Long.valueOf(-1)); // insert event fails, and returns -1\n assertNull(newLastIdentifyId);\n }\n\n @Test\n public void testDatabaseResetAvoidStackOverflow() {\n\n // log an event normally, verify metadata updated in table\n amplitude.logEvent(\"test\");\n looper.runToEndOfTasks();\n\n // metadata: deviceId, sessionId, sequence number, last_event_id, last_event_time, previous_session_id\n assertEquals(dbInstance.getEventCount(), 1);\n assertEquals(dbInstance.getTotalEventCount(), 1);\n assertEquals(dbInstance.getNthEventId(1), 1);\n\n String deviceId = dbInstance.getValue(AmplitudeClient.DEVICE_ID_KEY);\n long previousSessionId = dbInstance.getLongValue(AmplitudeClient.PREVIOUS_SESSION_ID_KEY);\n long sequenceNumber = dbInstance.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY);\n long lastEventId = dbInstance.getLongValue(AmplitudeClient.LAST_EVENT_ID_KEY);\n long lastEventTime = dbInstance.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY);\n Long lastIdentifyId = dbInstance.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY);\n\n assertNotNull(deviceId);\n assertTrue(deviceId.endsWith(\"R\"));\n assertTrue(previousSessionId > 0);\n assertEquals(sequenceNumber, 1);\n assertEquals(lastEventId, 1);\n assertTrue(lastEventTime >= startTime);\n assertNull(lastIdentifyId);\n\n // difficult to mock out the SQLiteDatabase object inside DatabaseHelper since it's private\n // add helper method specifically for mocking / testing\n DatabaseHelper mockDbHelper = PowerMockito.spy(dbInstance);\n PowerMockito.doThrow(new SQLiteException(\"test\")).when(mockDbHelper).insertKeyValueContentValuesIntoTable(Matchers.any(SQLiteDatabase.class), anyString(), Matchers.any(ContentValues.class));\n amplitude.dbHelper = mockDbHelper;\n\n // log an event to trigger SQLException that we set up with mocks\n amplitude.logEvent(\"test\");\n looper.runToEndOfTasks();\n\n // the reset callback handler will keep retriggering the exception, so make sure we guard against stack overflows\n // verify that the metadata has not been persisted back to database\n String newDeviceId = dbInstance.getValue(AmplitudeClient.DEVICE_ID_KEY);\n Long newPreviousSessionId = dbInstance.getLongValue(AmplitudeClient.PREVIOUS_SESSION_ID_KEY);\n Long newLastEventTime = dbInstance.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY);\n Long newSequenceNumber = dbInstance.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY);\n Long newLastEventId = dbInstance.getLongValue(AmplitudeClient.LAST_EVENT_ID_KEY);\n Long newLastIdentifyId = dbInstance.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY);\n\n assertNull(newDeviceId);\n assertNull(newPreviousSessionId);\n assertNull(newLastEventTime);\n assertNull(newSequenceNumber);\n assertNull(newLastEventId); // insert event fails, and returns -1\n assertNull(newLastIdentifyId);\n }\n\n @Test\n public void testCorruptingDatabaseFile() throws IOException, JSONException {\n\n // log an event normally, verify metadata updated in table\n amplitude.logEvent(\"test\");\n looper.runToEndOfTasks();\n\n // metadata: deviceId, sessionId, sequence number, last_event_id, last_event_time, previous_session_id\n assertEquals(dbInstance.getEventCount(), 1);\n assertEquals(dbInstance.getTotalEventCount(), 1);\n assertEquals(dbInstance.getNthEventId(1), 1);\n\n String deviceId = dbInstance.getValue(AmplitudeClient.DEVICE_ID_KEY);\n long previousSessionId = dbInstance.getLongValue(AmplitudeClient.PREVIOUS_SESSION_ID_KEY);\n long sequenceNumber = dbInstance.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY);\n long lastEventId = dbInstance.getLongValue(AmplitudeClient.LAST_EVENT_ID_KEY);\n long lastEventTime = dbInstance.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY);\n Long lastIdentifyId = dbInstance.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY);\n\n assertNotNull(deviceId);\n assertTrue(deviceId.endsWith(\"R\"));\n assertTrue(previousSessionId > 0);\n assertEquals(sequenceNumber, 1);\n assertEquals(lastEventId, 1);\n assertTrue(lastEventTime >= startTime);\n assertNull(lastIdentifyId);\n\n // try to corrupt database file and then log another event\n File file = dbInstance.file;\n RandomAccessFile writer = new RandomAccessFile(file.getAbsolutePath(), \"rw\");\n writer.seek(2);\n writer.writeChars(\"corrupt database file with random string\");\n writer.close();\n\n amplitude.logEvent(\"test\");\n looper.runToEndOfTasks();\n\n // since events table recreated, the event id should have been reset back to 1\n List events = dbInstance.getEvents(5, 5);\n assertEquals(events.size(), 1);\n assertEquals(events.get(0).optInt(\"event_id\"), 1);\n\n // verify metadata is re-inserted into database\n String newDeviceId = dbInstance.getValue(AmplitudeClient.DEVICE_ID_KEY);\n long newPreviousSessionId = dbInstance.getLongValue(AmplitudeClient.PREVIOUS_SESSION_ID_KEY);\n long newLastEventTime = dbInstance.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY);\n long newSequenceNumber = dbInstance.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY);\n long newLastEventId = dbInstance.getLongValue(AmplitudeClient.LAST_EVENT_ID_KEY);\n Long newLastIdentifyId = dbInstance.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY);\n\n assertEquals(newDeviceId, deviceId);\n assertEquals(newPreviousSessionId, previousSessionId);\n assertTrue(newLastEventTime >= lastEventTime);\n assertEquals(newSequenceNumber, 2);\n assertEquals(newLastEventId, 1);\n assertNull(newLastIdentifyId);\n }\n\n @Test\n public void testDeletedDatabaseFile() throws IOException, JSONException {\n\n // log an event normally, verify metadata updated in table\n amplitude.logEvent(\"test\");\n looper.runToEndOfTasks();\n\n // metadata: deviceId, sessionId, sequence number, last_event_id, last_event_time, previous_session_id\n assertEquals(dbInstance.getEventCount(), 1);\n assertEquals(dbInstance.getTotalEventCount(), 1);\n assertEquals(dbInstance.getNthEventId(1), 1);\n\n String deviceId = dbInstance.getValue(AmplitudeClient.DEVICE_ID_KEY);\n long previousSessionId = dbInstance.getLongValue(AmplitudeClient.PREVIOUS_SESSION_ID_KEY);\n long sequenceNumber = dbInstance.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY);\n long lastEventId = dbInstance.getLongValue(AmplitudeClient.LAST_EVENT_ID_KEY);\n long lastEventTime = dbInstance.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY);\n Long lastIdentifyId = dbInstance.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY);\n\n assertNotNull(deviceId);\n assertTrue(deviceId.endsWith(\"R\"));\n assertTrue(previousSessionId > 0);\n assertEquals(sequenceNumber, 1);\n assertEquals(lastEventId, 1);\n assertTrue(lastEventTime >= startTime);\n assertNull(lastIdentifyId);\n\n // try to delete database file and test logging event\n File file = dbInstance.file;\n context.deleteDatabase(file.getName());\n amplitude.logEvent(\"test\");\n looper.runToEndOfTasks();\n\n // since events table recreated, the event id should have been reset back to 1\n List events = dbInstance.getEvents(5, 5);\n assertEquals(events.size(), 1);\n assertEquals(events.get(0).optInt(\"event_id\"), 1);\n\n // verify metadata is re-inserted into database\n String newDeviceId = dbInstance.getValue(AmplitudeClient.DEVICE_ID_KEY);\n long newPreviousSessionId = dbInstance.getLongValue(AmplitudeClient.PREVIOUS_SESSION_ID_KEY);\n long newLastEventTime = dbInstance.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY);\n long newSequenceNumber = dbInstance.getLongValue(AmplitudeClient.SEQUENCE_NUMBER_KEY);\n long newLastEventId = dbInstance.getLongValue(AmplitudeClient.LAST_EVENT_ID_KEY);\n Long newLastIdentifyId = dbInstance.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY);\n\n assertEquals(newDeviceId, deviceId);\n assertEquals(newPreviousSessionId, previousSessionId);\n assertTrue(newLastEventTime >= lastEventTime);\n assertEquals(newSequenceNumber, 2);\n assertEquals(newLastEventId, 1);\n assertNull(newLastIdentifyId);\n }\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/DeviceInfoTest.java",
"content": "package com.amplitude.api;\n\nimport android.content.ContentResolver;\nimport android.content.Context;\nimport android.content.res.Configuration;\nimport android.content.res.Resources;\nimport android.location.Location;\nimport android.os.Build;\nimport android.os.LocaleList;\nimport android.provider.Settings.Secure;\nimport android.telephony.TelephonyManager;\n\nimport androidx.test.ext.junit.runners.AndroidJUnit4;\n\nimport com.google.android.gms.ads.identifier.AdvertisingIdClient;\nimport com.google.android.gms.common.ConnectionResult;\nimport com.google.android.gms.common.GooglePlayServicesUtil;\nimport com.google.android.gms.tasks.Task;\nimport com.google.android.gms.tasks.Tasks;\n\nimport org.json.JSONObject;\nimport org.junit.After;\nimport org.junit.Assert;\nimport org.junit.Before;\nimport org.junit.Rule;\nimport org.junit.Test;\nimport org.junit.runner.RunWith;\nimport org.mockito.Mock;\nimport org.mockito.Mockito;\nimport org.mockito.internal.stubbing.answers.CallsRealMethods;\nimport org.powermock.api.mockito.PowerMockito;\nimport org.powermock.core.classloader.annotations.PowerMockIgnore;\nimport org.powermock.core.classloader.annotations.PrepareForTest;\nimport org.powermock.modules.junit4.PowerMockRunner;\nimport org.powermock.modules.junit4.rule.PowerMockRule;\nimport org.robolectric.Robolectric;\nimport org.robolectric.Shadows;\nimport org.robolectric.annotation.Config;\nimport org.robolectric.shadows.ShadowLooper;\nimport org.robolectric.shadows.ShadowTelephonyManager;\nimport org.robolectric.util.ReflectionHelpers;\n\nimport java.util.Locale;\nimport java.util.UUID;\n\nimport static org.junit.Assert.assertEquals;\nimport static org.junit.Assert.assertFalse;\nimport static org.junit.Assert.assertNotEquals;\nimport static org.junit.Assert.assertNotNull;\nimport static org.junit.Assert.assertNull;\nimport static org.junit.Assert.assertTrue;\nimport static org.junit.Assert.fail;\n\n\n@RunWith(AndroidJUnit4.class)\n@PowerMockIgnore({ \"org.mockito.*\", \"org.robolectric.*\", \"android.*\", \"androidx.*\", \"javax.net.ssl.*\", \"jdk.internal.reflect.*\" })\n@PrepareForTest({AdvertisingIdClient.class, GooglePlayServicesUtil.class})\n@Config(manifest = Config.NONE)\npublic class DeviceInfoTest extends BaseTest {\n private DeviceInfo deviceInfo;\n private static final String TEST_VERSION_NAME = \"test_version\";\n private static final String TEST_BRAND = \"brand\";\n private static final String TEST_MANUFACTURER = \"manufacturer\";\n private static final String TEST_MODEL = \"model\";\n private static final String TEST_CARRIER = \"carrier\";\n private static final Locale TEST_LOCALE = Locale.FRANCE;\n private static final String TEST_COUNTRY = \"FR\";\n private static final String TEST_LANGUAGE = \"fr\";\n private static final String TEST_NETWORK_COUNTRY = \"GB\";\n private static final double TEST_LOCATION_LAT = 37.7749295;\n private static final double TEST_LOCATION_LNG = -122.4194155;\n private static final String TEST_GEO_COUNTRY = \"US\";\n\n private static Location makeLocation(String provider, double lat, double lng) {\n Location l = new Location(provider);\n l.setLatitude(lat);\n l.setLongitude(lng);\n l.setTime(System.currentTimeMillis());\n return l;\n }\n\n @Rule\n public PowerMockRule rule = new PowerMockRule();\n\n @Before\n public void setUp() throws Exception {\n super.setUp(false);\n\n ReflectionHelpers.setStaticField(Build.class, \"BRAND\", TEST_BRAND);\n ReflectionHelpers.setStaticField(Build.class, \"MANUFACTURER\", TEST_MANUFACTURER);\n ReflectionHelpers.setStaticField(Build.class, \"MODEL\", TEST_MODEL);\n\n Configuration c = context.getResources().getConfiguration();\n Resources.getSystem().getConfiguration().setLocales(LocaleList.forLanguageTags(TEST_LOCALE.toLanguageTag()));\n\n ShadowTelephonyManager manager = Shadows.shadowOf((TelephonyManager) context\n .getSystemService(Context.TELEPHONY_SERVICE));\n manager.setNetworkOperatorName(TEST_CARRIER);\n deviceInfo = new DeviceInfo(context, true, false);\n }\n\n @After\n public void tearDown() throws Exception {\n super.tearDown();\n }\n\n @Test\n public void testGetVersionName() {\n assertEquals(TEST_VERSION_NAME, deviceInfo.getVersionName());\n }\n\n @Test\n public void testGetBrand() {\n assertEquals(TEST_BRAND, deviceInfo.getBrand());\n }\n\n @Test\n public void testGetManufacturer() {\n assertEquals(TEST_MANUFACTURER, deviceInfo.getManufacturer());\n }\n\n @Test\n public void testGetModel() {\n assertEquals(TEST_MODEL, deviceInfo.getModel());\n }\n\n @Test\n public void testGetCarrier() {\n assertEquals(TEST_CARRIER, deviceInfo.getCarrier());\n }\n\n @Test\n public void testGetCountry() {\n assertEquals(TEST_COUNTRY, deviceInfo.getCountry());\n }\n\n @Test\n public void testGetCountryFromNetwork() {\n ShadowTelephonyManager manager = Shadows.shadowOf((TelephonyManager) context\n .getSystemService(Context.TELEPHONY_SERVICE));\n manager.setNetworkCountryIso(TEST_NETWORK_COUNTRY);\n\n DeviceInfo deviceInfo = new DeviceInfo(context, true, false);\n assertEquals(TEST_NETWORK_COUNTRY, deviceInfo.getCountry());\n }\n\n // TODO: Consider move this test to android specific tests.\n// @Test\n// @Config(shadows = {MockGeocoder.class})\n// public void testGetCountryFromLocation() {\n// ShadowTelephonyManager telephonyManager = Shadows.shadowOf((TelephonyManager) context\n// .getSystemService(Context.TELEPHONY_SERVICE));\n// telephonyManager.setNetworkCountryIso(TEST_NETWORK_COUNTRY);\n// ShadowLocationManager locationManager = Shadows.shadowOf((LocationManager) context\n// .getSystemService(Context.LOCATION_SERVICE));\n// locationManager.simulateLocation(makeLocation(LocationManager.NETWORK_PROVIDER,\n// TEST_LOCATION_LAT, TEST_LOCATION_LNG));\n// locationManager.setProviderEnabled(LocationManager.NETWORK_PROVIDER, true);\n//\n// DeviceInfo deviceInfo = new DeviceInfo(context) {\n// @Override\n// protected Geocoder getGeocoder() {\n// Geocoder geocoder = new Geocoder(context, Locale.ENGLISH);\n// ShadowGeocoder shadowGeocoder = Shadow.extract(geocoder);\n// shadowGeocoder.setSimulatedResponse(\"1 Dr Carlton B Goodlett Pl\", \"San Francisco\",\n// \"CA\", \"94506\", TEST_GEO_COUNTRY);\n// return geocoder;\n// }\n// };\n//\n// assertEquals(TEST_GEO_COUNTRY, deviceInfo.getCountry());\n// }\n\n @Test\n public void testGetLanguage() {\n assertEquals(TEST_LANGUAGE, deviceInfo.getLanguage());\n }\n\n @Test\n public void testGetAdvertisingIdFromGoogleDevice() {\n PowerMockito.mockStatic(AdvertisingIdClient.class);\n String advertisingId = \"advertisingId\";\n AdvertisingIdClient.Info info = new AdvertisingIdClient.Info(\n advertisingId,\n false\n );\n\n try {\n Mockito.when(AdvertisingIdClient.getAdvertisingIdInfo(context)).thenReturn(info);\n } catch (Exception e) {\n fail(e.toString());\n }\n DeviceInfo deviceInfo = new DeviceInfo(context, true, true);\n\n // still get advertisingId even if limit ad tracking disabled\n assertEquals(advertisingId, deviceInfo.getAdvertisingId());\n assertFalse(deviceInfo.isLimitAdTrackingEnabled());\n }\n\n @Test\n public void testGetAdvertisingIdFromGoogleDeviceDisabledTrackAdid() {\n PowerMockito.mockStatic(AdvertisingIdClient.class);\n\n DeviceInfo deviceInfo = new DeviceInfo(context, true, false);\n\n assertNull(deviceInfo.getAdvertisingId());\n assertFalse(deviceInfo.isLimitAdTrackingEnabled());\n\n PowerMockito.verifyStatic(Mockito.never());\n }\n\n @Test\n public void testGetAdvertisingIdFromAmazonDevice() {\n ReflectionHelpers.setStaticField(Build.class, \"MANUFACTURER\", \"Amazon\");\n\n String advertisingId = \"advertisingId\";\n ContentResolver cr = context.getContentResolver();\n\n Secure.putInt(cr, \"limit_ad_tracking\", 1);\n Secure.putString(cr, \"advertising_id\", advertisingId);\n\n DeviceInfo deviceInfo = new DeviceInfo(context, true, true);\n\n // still get advertisingID even if limit ad tracking enabled\n assertEquals(advertisingId, deviceInfo.getAdvertisingId());\n assertTrue(deviceInfo.isLimitAdTrackingEnabled());\n }\n\n @Test\n public void testGetAdvertisingIdFromAmazonDeviceDisabledTrackAdid() {\n ReflectionHelpers.setStaticField(Build.class, \"MANUFACTURER\", \"Amazon\");\n\n String advertisingId = \"advertisingId\";\n ContentResolver cr = context.getContentResolver();\n\n Secure.putInt(cr, \"limit_ad_tracking\", 1);\n Secure.putString(cr, \"advertising_id\", advertisingId);\n\n DeviceInfo deviceInfo = new DeviceInfo(context, true, false);\n\n assertNull(deviceInfo.getAdvertisingId());\n assertFalse(deviceInfo.isLimitAdTrackingEnabled());\n }\n\n @Test\n public void testGPSDisabled() {\n // GPS not enabled\n DeviceInfo deviceInfo = new DeviceInfo(context, true, false);\n assertFalse(deviceInfo.isGooglePlayServicesEnabled());\n\n // GPS bundled but not enabled, GooglePlayUtils.isAvailable returns non-0 value\n PowerMockito.mockStatic(GooglePlayServicesUtil.class);\n try {\n Mockito.when(GooglePlayServicesUtil.isGooglePlayServicesAvailable(context))\n .thenReturn(1);\n } catch (Exception e) {\n fail(e.toString());\n }\n assertFalse(deviceInfo.isGooglePlayServicesEnabled());\n }\n\n @Test\n public void testGPSEnabled() {\n PowerMockito.mockStatic(GooglePlayServicesUtil.class);\n try {\n Mockito.when(GooglePlayServicesUtil.isGooglePlayServicesAvailable(context))\n .thenReturn(ConnectionResult.SUCCESS);\n } catch (Exception e) {\n fail(e.toString());\n }\n assertTrue(deviceInfo.isGooglePlayServicesEnabled());\n }\n// TODO: Consider move this test to android specific tests.\n// @Test\n// public void testGetMostRecentLocation() {\n// DeviceInfo deviceInfo = new DeviceInfo(context);\n// ShadowLocationManager locationManager = Shadows.shadowOf((LocationManager) context\n// .getSystemService(Context.LOCATION_SERVICE));\n// Location loc = makeLocation(LocationManager.NETWORK_PROVIDER, TEST_LOCATION_LAT,\n// TEST_LOCATION_LNG);\n// locationManager.simulateLocation(loc);\n// locationManager.setProviderEnabled(LocationManager.NETWORK_PROVIDER, true);\n// assertEquals(loc, deviceInfo.getMostRecentLocation());\n// }\n\n @Test\n public void testNoLocation() {\n DeviceInfo deviceInfo = new DeviceInfo(context, true, false);\n Location recent = deviceInfo.getMostRecentLocation();\n assertNull(recent);\n }\n\n @Test\n public void testUseAdvertisingIdAsDeviceId() {\n PowerMockito.mockStatic(AdvertisingIdClient.class);\n String advertisingId = \"advertisingId\";\n AdvertisingIdClient.Info info = new AdvertisingIdClient.Info(\n advertisingId,\n false\n );\n\n try {\n Mockito.when(AdvertisingIdClient.getAdvertisingIdInfo(context)).thenReturn(info);\n } catch (Exception e) {\n fail(e.toString());\n }\n\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n AmplitudeClient client = Amplitude.getInstance(\"ADID\");\n client.useAdvertisingIdForDeviceId();\n client.initialize(context, \"1cc2c1978ebab0f6451112a8f5df4f4e\");\n ShadowLooper looper = Shadows.shadowOf(client.logThread.getLooper());\n looper.runToEndOfTasks();\n\n assertEquals(advertisingId, client.getDeviceId());\n }\n\n @Test\n public void testDontUseAdvertisingIdAsDeviceId() {\n PowerMockito.mockStatic(AdvertisingIdClient.class);\n String advertisingId = \"advertisingId\";\n AdvertisingIdClient.Info info = new AdvertisingIdClient.Info(\n advertisingId,\n true\n );\n\n try {\n Mockito.when(AdvertisingIdClient.getAdvertisingIdInfo(context)).thenReturn(info);\n } catch (Exception e) {\n fail(e.toString());\n }\n\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n AmplitudeClient client = Amplitude.getInstance(\"NoADID\");\n client.useAdvertisingIdForDeviceId();\n client.initialize(context, \"1cc2c1978ebab0f6451112a8f5df4f4e\");\n ShadowLooper looper = Shadows.shadowOf(client.logThread.getLooper());\n looper.runToEndOfTasks();\n\n assertNotEquals(advertisingId, client.getDeviceId());\n assertTrue(client.getDeviceId().endsWith(\"R\"));\n }\n\n @Test\n public void testDeviceIdEqualsToAppSetId() {\n //Set Advertising ID to be invalid to fallback to app set id\n PowerMockito.mockStatic(AdvertisingIdClient.class);\n String advertisingId = \"00000000-0000-0000-0000-000000000000\";\n AdvertisingIdClient.Info info = new AdvertisingIdClient.Info(\n advertisingId,\n true\n );\n try {\n Mockito.when(AdvertisingIdClient.getAdvertisingIdInfo(context)).thenReturn(info);\n } catch (Exception e) {\n fail(e.toString());\n }\n\n String mockAppSetId = \"5a8f0fd1-31a9-4a1f-bfad-cd5439ce533b\";\n DeviceInfoAmplitudeClient client = Mockito.spy(new DeviceInfoAmplitudeClient(\"AppSetId\"));\n DeviceInfo mockDeviceInfo = Mockito.mock(DeviceInfo.class, Mockito.CALLS_REAL_METHODS);\n try {\n Mockito.when(mockDeviceInfo.getAppSetId()).thenReturn(mockAppSetId);\n Mockito.when(client.publicInitializeDeviceInfo()).thenReturn(mockDeviceInfo);\n } catch (Exception e) {\n Assert.fail(e.toString());\n }\n\n client.useAdvertisingIdForDeviceId();\n client.useAppSetIdForDeviceId();\n\n final String[] deviceIdCallbackResult = new String[1];\n client.setDeviceIdCallback(deviceId -> deviceIdCallbackResult[0] = deviceId);\n\n client.initialize(context, \"1cc2c1978ebab0f6451112a8f5df4f4e\");\n ShadowLooper looper = Shadows.shadowOf(client.logThread.getLooper());\n looper.runToEndOfTasks();\n\n assertEquals(mockAppSetId + \"S\", client.getDeviceId());\n assertEquals(mockAppSetId + \"S\", deviceIdCallbackResult[0]);\n }\n\n @Test\n public void testToggleAppSetIdInEvents() {\n String mockAppSetId = \"5a8f0fd1-31a9-4a1f-bfad-cd5439ce533b\";\n amplitude = new DeviceInfoAmplitudeClient(\"\");\n DeviceInfoAmplitudeClient client = Mockito.spy((DeviceInfoAmplitudeClient) amplitude);\n DeviceInfo mockDeviceInfo = Mockito.mock(DeviceInfo.class, Mockito.CALLS_REAL_METHODS);\n try {\n Mockito.when(mockDeviceInfo.getAppSetId()).thenReturn(mockAppSetId);\n Mockito.when(client.publicInitializeDeviceInfo()).thenReturn(mockDeviceInfo);\n } catch (Exception e) {\n Assert.fail(e.toString());\n }\n\n ShadowLooper looper = Shadows.shadowOf(client.logThread.getLooper());\n\n client.useAppSetIdForDeviceId();\n client.initialize(context, apiKey);\n looper.runToEndOfTasks();\n assertEquals(mockAppSetId + \"S\", client.getDeviceId());\n\n client.logEvent(\"testSendAppSetIdInJson\");\n looper.runToEndOfTasks();\n\n JSONObject event = getLastEvent();\n assertNotNull(event);\n try {\n assertEquals(\"testSendAppSetIdInJson\", event.getString(\"event_type\"));\n JSONObject apiProps = event.getJSONObject(\"api_properties\");\n String appSetId = apiProps.getString(\"android_app_set_id\");\n assertEquals(mockAppSetId, appSetId);\n } catch (Exception e) {\n Assert.fail(e.toString());\n }\n\n TrackingOptions options = new TrackingOptions();\n options.disableAppSetId();\n client.setTrackingOptions(options);\n client.logEvent(\"testSendAppSetIdInJson-2\");\n looper.runToEndOfTasks();\n\n event = getLastEvent();\n assertNotNull(event);\n try {\n assertEquals(\"testSendAppSetIdInJson-2\", event.getString(\"event_type\"));\n JSONObject apiProps = event.getJSONObject(\"api_properties\");\n assertFalse(apiProps.has(\"android_app_set_id\"));\n } catch (Exception e) {\n Assert.fail(e.toString());\n }\n }\n\n private class DeviceInfoAmplitudeClient extends AmplitudeClient {\n protected DeviceInfo initializeDeviceInfo() {\n return this.publicInitializeDeviceInfo();\n }\n public DeviceInfo publicInitializeDeviceInfo() {\n return new DeviceInfo(context, true, false);\n }\n public DeviceInfoAmplitudeClient(String instance) {\n super(instance);\n }\n }\n\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/IdentifyTest.java",
"content": "package com.amplitude.api;\n\nimport androidx.test.ext.junit.runners.AndroidJUnit4;\n\nimport org.json.JSONArray;\nimport org.json.JSONException;\nimport org.json.JSONObject;\nimport org.junit.After;\nimport org.junit.Before;\nimport org.junit.Test;\nimport org.junit.runner.RunWith;\nimport org.robolectric.annotation.Config;\n\nimport static org.junit.Assert.assertTrue;\n\n@RunWith(AndroidJUnit4.class)\n@Config(manifest = Config.NONE)\npublic class IdentifyTest extends BaseTest {\n\n @Before\n public void setUp() throws Exception { setUp(false); }\n\n @After\n public void tearDown() throws Exception {}\n\n @Test\n public void testUnsetProperty() throws JSONException {\n String property1 = \"testProperty1\";\n String property2 = \"testProperty2\";\n Identify identify = new Identify().unset(property1).unset(property2).unset(property1);\n\n JSONObject expected = new JSONObject();\n JSONObject expectedOperations = new JSONObject().put(property1, \"-\").put(property2, \"-\");\n expected.put(Constants.AMP_OP_UNSET, expectedOperations);\n assertTrue(Utils.compareJSONObjects(expected, identify.userPropertiesOperations));\n }\n\n @Test\n public void testSetProperty() throws JSONException {\n String property1 = \"string value\";\n String value1 = \"testValue\";\n\n String property2 = \"double value\";\n double value2 = 0.123;\n\n String property3 = \"boolean value\";\n boolean value3 = true;\n\n String property4 = \"json value\";\n JSONObject value4 = new JSONObject();\n\n String property5 = \"boolean array\";\n boolean[] value5 = new boolean[]{true, true, false};\n JSONArray value5Expected = new JSONArray();\n for (boolean value : value5) value5Expected.put(value);\n\n Identify identify = new Identify().set(property1, value1).set(property2, value2);\n identify.set(property3, value3).set(property4, value4).set(property5, value5);\n\n // identify should ignore this since duplicate key\n identify.set(property1, value3);\n\n JSONObject expected = new JSONObject();\n JSONObject expectedOperations = new JSONObject().put(property1, value1);\n expectedOperations.put(property2, value2).put(property3, value3).put(property4, value4);\n expectedOperations.put(property5, value5Expected);\n expected.put(Constants.AMP_OP_SET, expectedOperations);\n assertTrue(Utils.compareJSONObjects(expected, identify.userPropertiesOperations));\n }\n\n @Test\n public void testSetOnceProperty() throws JSONException {\n String property1 = \"string value\";\n String value1 = \"testValue\";\n\n String property2 = \"double value\";\n double value2 = 0.123;\n\n String property3 = \"boolean value\";\n boolean value3 = true;\n\n String property4 = \"json value\";\n JSONObject value4 = new JSONObject();\n\n String property5 = \"double array\";\n double[] value5 = new double[]{1.2, 2.3, 3.4};\n JSONArray value5Expected = new JSONArray();\n for (double value : value5) value5Expected.put(value);\n\n Identify identify = new Identify().setOnce(property1, value1).setOnce(property2, value2);\n identify.setOnce(property3, value3).setOnce(property4, value4).setOnce(property5, value5);\n\n // identify should ignore this since duplicate key\n identify.setOnce(property1, value3);\n\n JSONObject expected = new JSONObject();\n JSONObject expectedOperations = new JSONObject().put(property1, value1);\n expectedOperations.put(property2, value2).put(property3, value3).put(property4, value4);\n expectedOperations.put(property5, value5Expected);\n expected.put(Constants.AMP_OP_SET_ONCE, expectedOperations);\n assertTrue(Utils.compareJSONObjects(expected, identify.userPropertiesOperations));\n }\n\n @Test\n public void testAddProperty() throws JSONException {\n String property1 = \"int value\";\n int value1 = 5;\n\n String property2 = \"double value\";\n double value2 = 0.123;\n\n String property3 = \"float value\";\n double value3 = 0.625; // floats are actually promoted to long in JSONObject\n\n String property4 = \"long value\";\n long value4 = 18l;\n\n String property5 = \"string value\";\n String value5 = \"19\";\n\n Identify identify = new Identify().add(property1, value1).add(property2, value2);\n identify.add(property3, value3).add(property4, value4).add(property5, value5);\n\n // identify should ignore this since duplicate key\n identify.add(property1, value3);\n\n JSONObject expected = new JSONObject();\n JSONObject expectedOperations = new JSONObject().put(property1, value1);\n expectedOperations.put(property2, value2).put(property3, value3).put(property4, value4);\n expectedOperations.put(property5, value5);\n expected.put(Constants.AMP_OP_ADD, expectedOperations);\n assertTrue(Utils.compareJSONObjects(expected, identify.userPropertiesOperations));\n }\n\n @Test\n public void testAppendProperty() throws JSONException {\n String property1 = \"int value\";\n int value1 = 5;\n\n String property2 = \"double value\";\n double value2 = 0.123;\n\n String property3 = \"float value\";\n double value3 = 0.625; // floats are actually promoted to long in JSONObject\n\n String property4 = \"long value\";\n long value4 = 18l;\n\n String property5 = \"array value\";\n JSONArray value5 = new JSONArray();\n value5.put(1);\n value5.put(2);\n value5.put(3);\n\n String property6 = \"float array\";\n float[] value6 = new float[]{(float)1.2, (float)2.3, (float)3.4, (float)4.5};\n JSONArray value6Expected = new JSONArray();\n for (float value : value6) value6Expected.put(value);\n\n String property7 = \"int array\";\n int[] value7 = new int[]{10, 12, 14, 17};\n JSONArray value7Expected = new JSONArray();\n for (int value : value7) value7Expected.put(value);\n\n String property8 = \"long array\";\n long[] value8 = new long[]{20, 22, 24, 27};\n JSONArray value8Expected = new JSONArray();\n for (long value : value8) value8Expected.put(value);\n\n String property9 = \"string array\";\n String[] value9 = new String[]{\"test1\", \"test2\", \"test3\"};\n JSONArray value9Expected = new JSONArray();\n for (String value : value9) value9Expected.put(value);\n\n Identify identify = new Identify().append(property1, value1).append(property2, value2);\n identify.append(property3, value3).append(property4, value4).append(property5, value5);\n identify.append(property6, value6).append(property7, value7).append(property8, value8);\n identify.append(property9, value9);\n\n // identify should ignore this since duplicate key\n identify.add(property1, value3);\n\n JSONObject expected = new JSONObject();\n JSONObject expectedOperations = new JSONObject().put(property1, value1);\n expectedOperations.put(property2, value2).put(property3, value3).put(property4, value4);\n expectedOperations.put(property5, value5).put(property6, value6Expected);\n expectedOperations.put(property7, value7Expected).put(property8, value8Expected);\n expectedOperations.put(property9, value9Expected);\n expected.put(Constants.AMP_OP_APPEND, expectedOperations);\n assertTrue(Utils.compareJSONObjects(expected, identify.userPropertiesOperations));\n }\n\n @Test\n public void testPrependProperty() throws JSONException {\n String property1 = \"int value\";\n int value1 = 5;\n\n String property2 = \"double value\";\n double value2 = 0.123;\n\n String property3 = \"float value\";\n double value3 = 0.625; // floats are actually promoted to long in JSONObject\n\n String property4 = \"long value\";\n long value4 = 18l;\n\n String property5 = \"array value\";\n JSONArray value5 = new JSONArray();\n value5.put(1);\n value5.put(2);\n value5.put(3);\n\n String property6 = \"float array\";\n float[] value6 = new float[]{(float)1.2, (float)2.3, (float)3.4, (float)4.5};\n JSONArray value6Expected = new JSONArray();\n for (float value : value6) value6Expected.put(value);\n\n String property7 = \"int array\";\n int[] value7 = new int[]{10, 12, 14, 17};\n JSONArray value7Expected = new JSONArray();\n for (int value : value7) value7Expected.put(value);\n\n String property8 = \"long array\";\n long[] value8 = new long[]{20, 22, 24, 27};\n JSONArray value8Expected = new JSONArray();\n for (long value : value8) value8Expected.put(value);\n\n String property9 = \"string array\";\n String[] value9 = new String[]{\"test1\", \"test2\", \"test3\"};\n JSONArray value9Expected = new JSONArray();\n for (String value : value9) value9Expected.put(value);\n\n Identify identify = new Identify().prepend(property1, value1).prepend(property2, value2);\n identify.prepend(property3, value3).prepend(property4, value4).prepend(property5, value5);\n identify.prepend(property6, value6).prepend(property7, value7).prepend(property8, value8);\n identify.prepend(property9, value9);\n\n // identify should ignore this since duplicate key\n identify.add(property1, value3);\n\n JSONObject expected = new JSONObject();\n JSONObject expectedOperations = new JSONObject().put(property1, value1);\n expectedOperations.put(property2, value2).put(property3, value3).put(property4, value4);\n expectedOperations.put(property5, value5).put(property6, value6Expected);\n expectedOperations.put(property7, value7Expected).put(property8, value8Expected);\n expectedOperations.put(property9, value9Expected);\n expected.put(Constants.AMP_OP_PREPEND, expectedOperations);\n assertTrue(Utils.compareJSONObjects(expected, identify.userPropertiesOperations));\n }\n\n @Test\n public void testPreInsertProperty() throws JSONException {\n String property1 = \"int value\";\n int value1 = 5;\n\n String property2 = \"double value\";\n double value2 = 0.123;\n\n String property3 = \"float value\";\n double value3 = 0.625; // floats are actually promoted to long in JSONObject\n\n String property4 = \"long value\";\n long value4 = 18l;\n\n String property5 = \"array value\";\n JSONArray value5 = new JSONArray();\n value5.put(1);\n value5.put(2);\n value5.put(3);\n\n String property6 = \"float array\";\n float[] value6 = new float[]{(float)1.2, (float)2.3, (float)3.4, (float)4.5};\n JSONArray value6Expected = new JSONArray();\n for (float value : value6) value6Expected.put(value);\n\n String property7 = \"int array\";\n int[] value7 = new int[]{10, 12, 14, 17};\n JSONArray value7Expected = new JSONArray();\n for (int value : value7) value7Expected.put(value);\n\n String property8 = \"long array\";\n long[] value8 = new long[]{20, 22, 24, 27};\n JSONArray value8Expected = new JSONArray();\n for (long value : value8) value8Expected.put(value);\n\n String property9 = \"string array\";\n String[] value9 = new String[]{\"test1\", \"test2\", \"test3\"};\n JSONArray value9Expected = new JSONArray();\n for (String value : value9) value9Expected.put(value);\n\n Identify identify = new Identify().preInsert(property1, value1).preInsert(property2, value2);\n identify.preInsert(property3, value3).preInsert(property4, value4).preInsert(property5, value5);\n identify.preInsert(property6, value6).preInsert(property7, value7).preInsert(property8, value8);\n identify.preInsert(property9, value9);\n\n // identify should ignore this since duplicate key\n identify.add(property1, value3);\n\n JSONObject expected = new JSONObject();\n JSONObject expectedOperations = new JSONObject().put(property1, value1);\n expectedOperations.put(property2, value2).put(property3, value3).put(property4, value4);\n expectedOperations.put(property5, value5).put(property6, value6Expected);\n expectedOperations.put(property7, value7Expected).put(property8, value8Expected);\n expectedOperations.put(property9, value9Expected);\n expected.put(Constants.AMP_OP_PREINSERT, expectedOperations);\n assertTrue(Utils.compareJSONObjects(expected, identify.userPropertiesOperations));\n }\n\n @Test\n public void testPostInsertProperty() throws JSONException {\n String property1 = \"int value\";\n int value1 = 5;\n\n String property2 = \"double value\";\n double value2 = 0.123;\n\n String property3 = \"float value\";\n double value3 = 0.625; // floats are actually promoted to long in JSONObject\n\n String property4 = \"long value\";\n long value4 = 18l;\n\n String property5 = \"array value\";\n JSONArray value5 = new JSONArray();\n value5.put(1);\n value5.put(2);\n value5.put(3);\n\n String property6 = \"float array\";\n float[] value6 = new float[]{(float)1.2, (float)2.3, (float)3.4, (float)4.5};\n JSONArray value6Expected = new JSONArray();\n for (float value : value6) value6Expected.put(value);\n\n String property7 = \"int array\";\n int[] value7 = new int[]{10, 12, 14, 17};\n JSONArray value7Expected = new JSONArray();\n for (int value : value7) value7Expected.put(value);\n\n String property8 = \"long array\";\n long[] value8 = new long[]{20, 22, 24, 27};\n JSONArray value8Expected = new JSONArray();\n for (long value : value8) value8Expected.put(value);\n\n String property9 = \"string array\";\n String[] value9 = new String[]{\"test1\", \"test2\", \"test3\"};\n JSONArray value9Expected = new JSONArray();\n for (String value : value9) value9Expected.put(value);\n\n Identify identify = new Identify().postInsert(property1, value1).postInsert(property2, value2);\n identify.postInsert(property3, value3).postInsert(property4, value4).postInsert(property5, value5);\n identify.postInsert(property6, value6).postInsert(property7, value7).postInsert(property8, value8);\n identify.postInsert(property9, value9);\n\n // identify should ignore this since duplicate key\n identify.add(property1, value3);\n\n JSONObject expected = new JSONObject();\n JSONObject expectedOperations = new JSONObject().put(property1, value1);\n expectedOperations.put(property2, value2).put(property3, value3).put(property4, value4);\n expectedOperations.put(property5, value5).put(property6, value6Expected);\n expectedOperations.put(property7, value7Expected).put(property8, value8Expected);\n expectedOperations.put(property9, value9Expected);\n expected.put(Constants.AMP_OP_POSTINSERT, expectedOperations);\n assertTrue(Utils.compareJSONObjects(expected, identify.userPropertiesOperations));\n }\n\n @Test\n public void testRemoveProperty() throws JSONException {\n String property1 = \"int value\";\n int value1 = 5;\n\n String property2 = \"double value\";\n double value2 = 0.123;\n\n String property3 = \"float value\";\n double value3 = 0.625; // floats are actually promoted to long in JSONObject\n\n String property4 = \"long value\";\n long value4 = 18l;\n\n String property5 = \"array value\";\n JSONArray value5 = new JSONArray();\n value5.put(1);\n value5.put(2);\n value5.put(3);\n\n String property6 = \"float array\";\n float[] value6 = new float[]{(float)1.2, (float)2.3, (float)3.4, (float)4.5};\n JSONArray value6Expected = new JSONArray();\n for (float value : value6) value6Expected.put(value);\n\n String property7 = \"int array\";\n int[] value7 = new int[]{10, 12, 14, 17};\n JSONArray value7Expected = new JSONArray();\n for (int value : value7) value7Expected.put(value);\n\n String property8 = \"long array\";\n long[] value8 = new long[]{20, 22, 24, 27};\n JSONArray value8Expected = new JSONArray();\n for (long value : value8) value8Expected.put(value);\n\n String property9 = \"string array\";\n String[] value9 = new String[]{\"test1\", \"test2\", \"test3\"};\n JSONArray value9Expected = new JSONArray();\n for (String value : value9) value9Expected.put(value);\n\n Identify identify = new Identify().remove(property1, value1).remove(property2, value2);\n identify.remove(property3, value3).remove(property4, value4).remove(property5, value5);\n identify.remove(property6, value6).remove(property7, value7).remove(property8, value8);\n identify.remove(property9, value9);\n\n // identify should ignore this since duplicate key\n identify.add(property1, value3);\n\n JSONObject expected = new JSONObject();\n JSONObject expectedOperations = new JSONObject().put(property1, value1);\n expectedOperations.put(property2, value2).put(property3, value3).put(property4, value4);\n expectedOperations.put(property5, value5).put(property6, value6Expected);\n expectedOperations.put(property7, value7Expected).put(property8, value8Expected);\n expectedOperations.put(property9, value9Expected);\n expected.put(Constants.AMP_OP_REMOVE, expectedOperations);\n assertTrue(Utils.compareJSONObjects(expected, identify.userPropertiesOperations));\n }\n\n @Test\n public void testMultipleOperations() throws JSONException {\n String property1 = \"string value\";\n String value1 = \"testValue\";\n\n String property2 = \"double value\";\n double value2 = 0.123;\n\n String property3 = \"boolean value\";\n boolean value3 = true;\n\n String property4 = \"json value\";\n\n String property5 = \"array value\";\n JSONArray value5 = new JSONArray();\n value5.put(15);\n value5.put(25);\n\n String property6 = \"int value\";\n int value6 = 100;\n\n String property7 = \"string value2\";\n String value7 = \"testValue2\";\n\n String property8 = \"double value2\";\n double value8 = 0.123;\n\n String property9 = \"boolean value2\";\n boolean value9 = true; \n\n Identify identify = new Identify().setOnce(property1, value1).add(property2, value2);\n identify.set(property3, value3).unset(property4).append(property5, value5);\n identify.prepend(property6, value6).preInsert(property7, value7);\n identify.postInsert(property8, value8).remove(property9, value9);\n\n // identify should ignore this since duplicate key\n identify.set(property4, value3);\n\n JSONObject expected = new JSONObject();\n expected.put(Constants.AMP_OP_SET_ONCE, new JSONObject().put(property1, value1));\n expected.put(Constants.AMP_OP_ADD, new JSONObject().put(property2, value2));\n expected.put(Constants.AMP_OP_SET, new JSONObject().put(property3, value3));\n expected.put(Constants.AMP_OP_UNSET, new JSONObject().put(property4, \"-\"));\n expected.put(Constants.AMP_OP_APPEND, new JSONObject().put(property5, value5));\n expected.put(Constants.AMP_OP_PREPEND, new JSONObject().put(property6, value6));\n expected.put(Constants.AMP_OP_PREINSERT, new JSONObject().put(property7, value7));\n expected.put(Constants.AMP_OP_POSTINSERT, new JSONObject().put(property8, value8));\n expected.put(Constants.AMP_OP_REMOVE, new JSONObject().put(property9, value9));\n assertTrue(Utils.compareJSONObjects(expected, identify.userPropertiesOperations));\n }\n\n @Test\n public void testDisallowDuplicateProperties() throws JSONException {\n String property = \"testProperty\";\n String value1 = \"testValue\";\n double value2 = 0.123;\n boolean value3 = true;\n\n Identify identify = new Identify().setOnce(property, value1).add(property, value2);\n identify.set(property, value3).unset(property);\n\n JSONObject expected = new JSONObject();\n expected.put(Constants.AMP_OP_SET_ONCE, new JSONObject().put(property, value1));\n assertTrue(Utils.compareJSONObjects(expected, identify.userPropertiesOperations));\n }\n\n @Test\n public void testDisallowOtherOperationsOnClearAllIdentify() throws JSONException {\n String property = \"testProperty\";\n String value1 = \"testValue\";\n double value2 = 0.123;\n boolean value3 = true;\n\n Identify identify = new Identify().clearAll().setOnce(property, value1);\n identify.add(property, value2).set(property, value3).unset(property);\n\n JSONObject expected = new JSONObject();\n expected.put(Constants.AMP_OP_CLEAR_ALL, \"-\");\n assertTrue(Utils.compareJSONObjects(expected, identify.userPropertiesOperations));\n }\n\n @Test\n public void testDisallowClearAllOnIdentifysWithOtherOperations() throws JSONException {\n String property = \"testProperty\";\n String value1 = \"testValue\";\n double value2 = 0.123;\n boolean value3 = true;\n\n Identify identify = new Identify().setOnce(property, value1).add(property, value2);\n identify.set(property, value3).unset(property).clearAll();\n\n JSONObject expected = new JSONObject();\n expected.put(Constants.AMP_OP_SET_ONCE, new JSONObject().put(property, value1));\n assertTrue(Utils.compareJSONObjects(expected, identify.userPropertiesOperations));\n }\n\n @Test\n public void testGetUserPropertyOperations() throws JSONException {\n String property1 = \"string value\";\n String value1 = \"testValue\";\n\n String property2 = \"double value\";\n double value2 = 0.123;\n\n String property3 = \"boolean value\";\n boolean value3 = true;\n\n String property4 = \"json value\";\n\n String property5 = \"array value\";\n JSONArray value5 = new JSONArray();\n value5.put(15);\n value5.put(25);\n\n String property6 = \"int value\";\n int value6 = 100;\n\n String property7 = \"string value2\";\n String value7 = \"testValue2\";\n\n String property8 = \"double value2\";\n double value8 = 0.123;\n\n String property9 = \"boolean value2\";\n boolean value9 = true; \n\n Identify identify = new Identify().setOnce(property1, value1).add(property2, value2);\n identify.set(property3, value3).unset(property4).append(property5, value5);\n identify.prepend(property6, value6).preInsert(property7, value7);\n identify.postInsert(property8, value8).remove(property9, value9);\n\n // identify should ignore this since duplicate key\n identify.set(property4, value3);\n\n JSONObject expected = new JSONObject();\n expected.put(Constants.AMP_OP_SET_ONCE, new JSONObject().put(property1, value1));\n expected.put(Constants.AMP_OP_ADD, new JSONObject().put(property2, value2));\n expected.put(Constants.AMP_OP_SET, new JSONObject().put(property3, value3));\n expected.put(Constants.AMP_OP_UNSET, new JSONObject().put(property4, \"-\"));\n expected.put(Constants.AMP_OP_APPEND, new JSONObject().put(property5, value5));\n expected.put(Constants.AMP_OP_PREPEND, new JSONObject().put(property6, value6));\n expected.put(Constants.AMP_OP_PREINSERT, new JSONObject().put(property7, value7));\n expected.put(Constants.AMP_OP_POSTINSERT, new JSONObject().put(property8, value8));\n expected.put(Constants.AMP_OP_REMOVE, new JSONObject().put(property9, value9));\n assertTrue(Utils.compareJSONObjects(expected, identify.getUserPropertiesOperations()));\n }\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/IngestionMetadataTest.java",
"content": "package com.amplitude.api;\n\nimport static org.junit.Assert.assertEquals;\n\nimport androidx.test.ext.junit.runners.AndroidJUnit4;\n\nimport org.json.JSONException;\nimport org.json.JSONObject;\nimport org.junit.Test;\nimport org.junit.runner.RunWith;\nimport org.robolectric.annotation.Config;\n\n@RunWith(AndroidJUnit4.class)\n@Config(manifest=Config.NONE)\npublic class IngestionMetadataTest {\n\n @Test\n public void testToJSONObject() throws JSONException {\n IngestionMetadata ingestionMetadata = new IngestionMetadata();\n String sourceName = \"ampli\";\n String sourceVersion = \"1.0.0\";\n ingestionMetadata.setSourceName(sourceName)\n .setSourceVersion(sourceVersion);\n JSONObject result = ingestionMetadata.toJSONObject();\n assertEquals(sourceName, result.getString(Constants.AMP_INGESTION_METADATA_SOURCE_NAME));\n assertEquals(sourceVersion, result.getString(Constants.AMP_INGESTION_METADATA_SOURCE_VERSION));\n }\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/InitializeTest.java",
"content": "package com.amplitude.api;\n\nimport android.content.Context;\n\nimport androidx.test.ext.junit.runners.AndroidJUnit4;\n\nimport org.json.JSONArray;\nimport org.json.JSONException;\nimport org.junit.After;\nimport org.junit.Before;\nimport org.junit.Test;\nimport org.junit.runner.RunWith;\nimport org.robolectric.Shadows;\nimport org.robolectric.annotation.Config;\nimport org.robolectric.shadows.ShadowLooper;\n\nimport okhttp3.mockwebserver.RecordedRequest;\n\nimport static org.junit.Assert.assertEquals;\nimport static org.junit.Assert.assertFalse;\nimport static org.junit.Assert.assertNotNull;\nimport static org.junit.Assert.assertNull;\nimport static org.junit.Assert.assertTrue;\n\n@RunWith(AndroidJUnit4.class)\n@Config(manifest = Config.NONE)\npublic class InitializeTest extends BaseTest {\n\n @Before\n public void setUp() throws Exception {\n super.setUp();\n }\n\n @After\n public void tearDown() throws Exception {\n super.tearDown();\n }\n\n @Test\n public void testInitializeUserId() {\n // the userId passed to initialize should override any existing values\n\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n dbHelper.insertOrReplaceKeyValue(AmplitudeClient.USER_ID_KEY, \"oldUserId\");\n\n String userId = \"newUserId\";\n amplitude.initialize(context, apiKey, userId);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runOneTask();\n\n // Test that the user id is set.\n assertEquals(userId, amplitude.userId);\n assertEquals(userId, dbHelper.getValue(AmplitudeClient.USER_ID_KEY));\n\n // Test that events are logged.\n RecordedRequest request = sendEvent(amplitude, \"init_test_event\", null);\n assertNotNull(request);\n }\n\n @Test\n public void testInitializeUserIdFromDb() {\n // since user id already exists in database, ignore old value in shared prefs\n String userId = \"testUserId\";\n\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n dbHelper.insertOrReplaceKeyValue(AmplitudeClient.USER_ID_KEY, userId);\n\n amplitude.initialize(context, apiKey);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runOneTask();\n\n // Test that the user id is set.\n assertEquals(amplitude.userId, userId);\n assertEquals(userId, dbHelper.getValue(AmplitudeClient.USER_ID_KEY));\n }\n\n @Test\n public void testInitializeOptOut() {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n assertNull(dbHelper.getLongValue(AmplitudeClient.OPT_OUT_KEY));\n\n amplitude.initialize(context, apiKey);\n looper.runOneTask();\n\n assertFalse(amplitude.isOptedOut());\n\n amplitude.setOptOut(true);\n looper.runOneTask();\n\n assertTrue(amplitude.isOptedOut());\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.OPT_OUT_KEY), 1L);\n }\n\n @Test\n public void testInitializeOptOutFromDB() {\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n dbHelper.insertOrReplaceKeyLongValue(AmplitudeClient.OPT_OUT_KEY, 0L);\n\n amplitude.initialize(context, apiKey);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runOneTask();\n\n assertFalse(amplitude.isOptedOut());\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.OPT_OUT_KEY), 0L);\n }\n\n\n @Test\n public void testInitializeLastEventId() throws JSONException {\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n\n amplitude.initialize(context, apiKey);\n amplitude.setLastEventId(3L);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runOneTask();\n Shadows.shadowOf(amplitude.logThread.getLooper()).runOneTask();\n\n assertEquals(amplitude.lastEventId, 3L);\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_ID_KEY), 3L);\n\n amplitude.logEvent(\"testEvent\");\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n\n assertEquals(events.getJSONObject(0).getLong(\"event_id\"), 1L);\n\n assertEquals(amplitude.lastEventId, 1L);\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_ID_KEY), 1L);\n }\n\n @Test\n public void testInitializePreviousSessionId() {\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n\n amplitude.initialize(context, apiKey);\n amplitude.setPreviousSessionId(4000L);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.PREVIOUS_SESSION_ID_KEY), 4000L);\n }\n\n @Test\n public void testInitializeLastEventTime() {\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n dbHelper.insertOrReplaceKeyLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY, 5000L);\n\n amplitude.initialize(context, apiKey);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runOneTask();\n\n assertEquals(amplitude.lastEventTime, 5000L);\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY), 5000L);\n }\n\n @Test\n public void testSkipSharedPrefsToDb() {\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n dbHelper.insertOrReplaceKeyValue(AmplitudeClient.DEVICE_ID_KEY, \"testDeviceId\");\n dbHelper.insertOrReplaceKeyLongValue(AmplitudeClient.PREVIOUS_SESSION_ID_KEY, 1000L);\n dbHelper.insertOrReplaceKeyLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY, 2000L);\n\n assertNull(dbHelper.getValue(AmplitudeClient.USER_ID_KEY));\n assertNull(dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_ID_KEY));\n assertNull(dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY));\n assertNull(dbHelper.getLongValue(AmplitudeClient.OPT_OUT_KEY));\n\n amplitude.initialize(context, apiKey);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runOneTask();\n looper.runToEndOfTasks();\n\n assertEquals(dbHelper.getValue(AmplitudeClient.DEVICE_ID_KEY), \"testDeviceId\");\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.PREVIOUS_SESSION_ID_KEY), 1000L);\n assertEquals((long) dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY), 2000L);\n assertNull(dbHelper.getValue(AmplitudeClient.USER_ID_KEY));\n assertNull(dbHelper.getLongValue(AmplitudeClient.LAST_EVENT_ID_KEY));\n assertNull(dbHelper.getLongValue(AmplitudeClient.LAST_IDENTIFY_ID_KEY));\n assertNull(dbHelper.getLongValue(AmplitudeClient.OPT_OUT_KEY));\n\n // after upgrade, pref values still there since they weren't deleted\n assertEquals(amplitude.deviceId, \"testDeviceId\");\n assertEquals(amplitude.previousSessionId, 1000L);\n assertEquals(amplitude.lastEventTime, 2000L);\n assertNull(amplitude.userId);\n }\n\n @Test\n public void testInitializePreviousSessionIdLastEventTime() {\n // set a previous session id & last event time\n // log an event with timestamp such that same session is continued\n // log second event with timestamp such that new session is started\n\n amplitude.setSessionTimeoutMillis(5000); // 5s\n\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n dbHelper.insertOrReplaceKeyLongValue(AmplitudeClient.LAST_EVENT_TIME_KEY, 7000L);\n\n long [] timestamps = {8000, 14000};\n clock.setTimestamps(timestamps);\n\n amplitude.initialize(context, apiKey);\n amplitude.setPreviousSessionId(6000);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n\n assertNull(amplitude.userId);\n\n // log first event\n amplitude.logEvent(\"testEvent1\");\n looper.runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, 6000L);\n assertEquals(amplitude.lastEventTime, 8000L);\n\n // log second event\n amplitude.logEvent(\"testEvent2\");\n looper.runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, 14000L);\n assertEquals(amplitude.lastEventTime, 14000L);\n }\n\n @Test\n public void testReloadDeviceIdFromDatabase() {\n String deviceId = \"test_device_id_from_database\";\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n DatabaseHelper.getDatabaseHelper(context).insertOrReplaceKeyValue(\n AmplitudeClient.DEVICE_ID_KEY, deviceId\n );\n\n amplitude.initialize(context, apiKey);\n looper.runToEndOfTasks();\n assertEquals(deviceId, amplitude.getDeviceId());\n }\n\n @Test\n public void testInitializeDeviceIdWithRandomUUID() {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n amplitude.initialize(context, apiKey);\n looper.runToEndOfTasks();\n\n String deviceId = amplitude.getDeviceId();\n assertEquals(37, deviceId.length());\n assertTrue(deviceId.endsWith(\"R\"));\n DatabaseHelper dbHelper = DatabaseHelper.getDatabaseHelper(context);\n assertEquals(deviceId, dbHelper.getValue(AmplitudeClient.DEVICE_ID_KEY));\n }\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/MiddlewareRunnerTest.java",
"content": "package com.amplitude.api;\n\nimport androidx.test.ext.junit.runners.AndroidJUnit4;\n\nimport org.json.JSONException;\nimport org.json.JSONObject;\nimport org.junit.Test;\nimport org.junit.runner.RunWith;\nimport org.robolectric.annotation.Config;\n\nimport static org.junit.Assert.assertEquals;\nimport static org.junit.Assert.assertFalse;\nimport static org.junit.Assert.assertTrue;\n\nimport java.util.concurrent.atomic.AtomicInteger;\n\n@RunWith(AndroidJUnit4.class)\n@Config(manifest= Config.NONE)\npublic class MiddlewareRunnerTest {\n MiddlewareRunner middlewareRunner = new MiddlewareRunner();\n\n @Test\n public void testMiddlewareRun() throws JSONException {\n String middlewareDevice = \"middleware_device\";\n Middleware updateDeviceIdMiddleware = new Middleware() {\n @Override\n public void run(MiddlewarePayload payload, MiddlewareNext next) {\n try {\n payload.event.put(\"device_model\", middlewareDevice);\n } catch (JSONException e) {\n e.printStackTrace();\n }\n next.run(payload);\n }\n };\n middlewareRunner.add(updateDeviceIdMiddleware);\n\n JSONObject event = new JSONObject().put(\"device_model\", \"sample_device\");\n boolean middlewareCompleted = middlewareRunner.run(new MiddlewarePayload(event, new MiddlewareExtra()));\n\n assertTrue(middlewareCompleted);\n assertEquals(event.getString(\"device_model\"), middlewareDevice);\n }\n\n @Test\n public void testRunWithNotPassMiddleware() throws JSONException {\n // first middleware\n String middlewareDevice = \"middleware_device\";\n Middleware updateDeviceIdMiddleware = new Middleware() {\n @Override\n public void run(MiddlewarePayload payload, MiddlewareNext next) {\n try {\n payload.event.put(\"device_model\", middlewareDevice);\n } catch (JSONException e) {\n e.printStackTrace();\n }\n next.run(payload);\n }\n };\n\n // swallow middleware\n String middlewareUser = \"middleware_user\";\n Middleware swallowMiddleware = new Middleware() {\n @Override\n public void run(MiddlewarePayload payload, MiddlewareNext next) {\n try {\n payload.event.put(\"user_id\", middlewareUser);\n } catch (JSONException e) {\n e.printStackTrace();\n }\n }\n };\n middlewareRunner.add(updateDeviceIdMiddleware);\n middlewareRunner.add(swallowMiddleware);\n\n JSONObject event = new JSONObject().put(\"device_model\", \"sample_device\").put(\"user_id\", \"sample_user\");\n boolean middlewareCompleted = middlewareRunner.run(new MiddlewarePayload(event));\n\n assertFalse(middlewareCompleted);\n assertEquals(event.getString(\"device_model\"), middlewareDevice);\n assertEquals(event.getString(\"user_id\"), middlewareUser);\n }\n\n @Test\n public void testMiddlewareFlush() throws JSONException {\n AtomicInteger runCount = new AtomicInteger(0);\n AtomicInteger flushCount = new AtomicInteger(0);\n\n MiddlewareExtended flushMiddleware = new MiddlewareExtended() {\n @Override\n public void run(MiddlewarePayload payload, MiddlewareNext next) {\n runCount.incrementAndGet();\n }\n\n @Override\n public void flush() {\n flushCount.incrementAndGet();\n }\n };\n\n middlewareRunner.add(flushMiddleware);\n\n middlewareRunner.flush();\n\n assertEquals(flushCount.get(), 1);\n assertEquals(runCount.get(), 0);\n }\n\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/MockGeocoder.java",
"content": "package com.amplitude.api;\n\nimport android.location.Geocoder;\n\nimport org.robolectric.annotation.Implementation;\nimport org.robolectric.annotation.Implements;\nimport org.robolectric.shadows.maps.ShadowGeocoder;\n\n// mock for static Geocoder method\n@Implements(Geocoder.class)\npublic class MockGeocoder extends ShadowGeocoder {\n @Implementation\n public static boolean isPresent() {\n return true;\n }\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/PinningTest.java",
"content": "package com.amplitude.api;\n\nimport android.os.SystemClock;\n\nimport androidx.test.ext.junit.runners.AndroidJUnit4;\n\nimport org.junit.After;\nimport org.junit.Before;\nimport org.junit.Ignore;\nimport org.junit.Test;\nimport org.junit.runner.RunWith;\nimport org.robolectric.Shadows;\nimport org.robolectric.annotation.Config;\nimport org.robolectric.shadows.ShadowLooper;\n\nimport static org.junit.Assert.assertNotNull;\nimport static org.junit.Assert.assertNull;\n\n@RunWith(AndroidJUnit4.class)\n@Config(manifest = Config.NONE)\npublic class PinningTest extends BaseTest {\n\n @Before\n public void setUp() throws Exception {\n super.setUp(false);\n PinnedAmplitudeClient.instances.clear();\n // need to set clock > 0 so that logThread posts in order\n SystemClock.setCurrentTimeMillis(1000);\n }\n\n @After\n public void tearDown() throws Exception {\n super.tearDown();\n PinnedAmplitudeClient.instances.clear();\n }\n\n @Test\n @Ignore(\"This stopped working in github and skipping this for now since this SDK is \" +\n \"under maintenance\")\n public void testSslPinningUS() {\n amplitude = PinnedAmplitudeClient.getInstance();\n amplitude.initialize(context, \"1cc2c1978ebab0f6451112a8f5df4f4e\");\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runOneTask();\n looper.runOneTask();\n\n amplitude.logEvent(\"us_pinned_test_event\", null);\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n ShadowLooper httplooper = Shadows.shadowOf(amplitude.httpThread.getLooper());\n httplooper.runToEndOfTasks();\n\n assertNull(amplitude.lastError);\n }\n\n @Test\n public void testSslPinningEU() {\n amplitude = PinnedAmplitudeClient.getInstance();\n amplitude.setServerZone(AmplitudeServerZone.EU);\n amplitude.initialize(context, \"361e4558bb359e288ef75d1ae31437a0\");\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runOneTask();\n looper.runOneTask();\n\n amplitude.logEvent(\"eu_pinned_test_event\", null);\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n ShadowLooper httplooper = Shadows.shadowOf(amplitude.httpThread.getLooper());\n httplooper.runToEndOfTasks();\n\n assertNull(amplitude.lastError);\n }\n\n @Test\n @Ignore(\"This stopped working in github and skipping this for now since this SDK is \" +\n \"under maintenance\")\n public void testSslPinningSwitch() {\n amplitude = PinnedAmplitudeClient.getInstance();\n amplitude.initialize(context, \"361e4558bb359e288ef75d1ae31437a0\");\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runOneTask();\n looper.runOneTask();\n\n amplitude.setServerZone(AmplitudeServerZone.EU);\n amplitude.logEvent(\"eu_pinned_test_event\", null);\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n ShadowLooper httplooper = Shadows.shadowOf(amplitude.httpThread.getLooper());\n httplooper.runToEndOfTasks();\n assertNotNull(amplitude.lastError);\n }\n\n @Test\n public void testSslPinningInvalid() {\n amplitude = new InvalidPinnedAmplitudeClient();\n amplitude.initialize(context, \"1cc2c1978ebab0f6451112a8f5df4f4e\");\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n looper.runToEndOfTasks();\n\n amplitude.logEvent(\"pinned_test_event_invalid\", null);\n looper.runToEndOfTasks();\n looper.runToEndOfTasks();\n\n ShadowLooper httplooper = Shadows.shadowOf(amplitude.httpThread.getLooper());\n httplooper.runToEndOfTasks();\n\n assertNotNull(amplitude.lastError);\n }\n\n private static class InvalidPinnedAmplitudeClient extends PinnedAmplitudeClient {\n public static final SSLContextBuilder INVALID_SSL_CONTEXT = new SSLContextBuilder()\n .addCertificate(\"\"\n + \"MIIFVjCCBD6gAwIBAgIRAObsedhCFsMHaYL156gA4XAwDQYJKoZIhvcNAQELBQAwgZ\"\n + \"AxCzAJBgNVBAYTAkdCMRswGQYDVQQIExJHcmVhdGVyIE1hbmNoZXN0ZXIxEDAOBgNV\"\n + \"BAcTB1NhbGZvcmQxGjAYBgNVBAoTEUNPTU9ETyBDQSBMaW1pdGVkMTYwNAYDVQQDEy\"\n + \"1DT01PRE8gUlNBIERvbWFpbiBWYWxpZGF0aW9uIFNlY3VyZSBTZXJ2ZXIgQ0EwHhcN\"\n + \"MTQwODE0MDAwMDAwWhcNMTkwODEzMjM1OTU5WjBcMSEwHwYDVQQLExhEb21haW4gQ2\"\n + \"9udHJvbCBWYWxpZGF0ZWQxHTAbBgNVBAsTFFBvc2l0aXZlU1NMIFdpbGRjYXJkMRgw\"\n + \"FgYDVQQDFA8qLnlpa3lha2FwaS5uZXQwggEiMA0GCSqGSIb3DQEBAQUAA4IBDwAwgg\"\n + \"EKAoIBAQDANfb+9W5g48LTQezWOMdQlL6kE66mqAnR9GAM1Ron31WuHQFn52Y/A6KK\"\n + \"EfUUHIcC/3vgLHRGzWlzPs8ctTHIMH++Tb2eS3uNhyeiQ2ZTALYFslNThZsdoh/kYu\"\n + \"YD5qX55ZKP1DJxm2ftcR38XoyWH1mv/JsT1Hq6/ATsesJJxwzxjI2G4NZyPFm8c2w8\"\n + \"EhfdBgDGyBliGo24TpM7uOVYmC01mAB/pZZS2EuBWjhA6Ny7pTLnjIBAx3jh8Vd3is\"\n + \"cMxq5boq2DKD0rSVQWbWdYbFMvvMmvq8qnuX9IijqbykoHoHGKerE/LaDs3xDZTlpE\"\n + \"AvAt8oFUyAllUNCairq3AgMBAAGjggHcMIIB2DAfBgNVHSMEGDAWgBSQr2o6lFoL2J\"\n + \"DqElZz30O0Oija5zAdBgNVHQ4EFgQUMpYvhtltmd1BhVilRheMd6wqENYwDgYDVR0P\"\n + \"AQH/BAQDAgWgMAwGA1UdEwEB/wQCMAAwHQYDVR0lBBYwFAYIKwYBBQUHAwEGCCsGAQ\"\n + \"UFBwMCMFAGA1UdIARJMEcwOwYMKwYBBAGyMQECAQMEMCswKQYIKwYBBQUHAgEWHWh0\"\n + \"dHBzOi8vc2VjdXJlLmNvbW9kby5uZXQvQ1BTMAgGBmeBDAECATBUBgNVHR8ETTBLME\"\n + \"mgR6BFhkNodHRwOi8vY3JsLmNvbW9kb2NhLmNvbS9DT01PRE9SU0FEb21haW5WYWxp\"\n + \"ZGF0aW9uU2VjdXJlU2VydmVyQ0EuY3JsMIGFBggrBgEFBQcBAQR5MHcwTwYIKwYBBQ\"\n + \"UHMAKGQ2h0dHA6Ly9jcnQuY29tb2RvY2EuY29tL0NPTU9ET1JTQURvbWFpblZhbGlk\"\n + \"YXRpb25TZWN1cmVTZXJ2ZXJDQS5jcnQwJAYIKwYBBQUHMAGGGGh0dHA6Ly9vY3NwLm\"\n + \"NvbW9kb2NhLmNvbTApBgNVHREEIjAggg8qLnlpa3lha2FwaS5uZXSCDXlpa3lha2Fw\"\n + \"aS5uZXQwDQYJKoZIhvcNAQELBQADggEBAEdw5iJwxvXcZlPQEbudu84VI48uwSYcGz\"\n + \"xBVzOsfYPdLUc7HSYT8zwg2d5l89iLiapvHS6gQiASZRi7nzR/oqnARcVjWnvIKPlq\"\n + \"+b3OUtNElnfSFXZnsgpxp3BlcCEfQs7faII89rTzxJqRf0fNo8Y4u3+k79zGF8xbon\"\n + \"O8oXZt0ApxcYmFIQlhCddM20lgHLTeMx4yG5C2lHGJE3iUS7YVAq6ENRrgiVhcuf5R\"\n + \"H1mWAYpFPJ7rOmpCReC6brxCho/7jg+fBqEUfCGyrMtYSRejCc9aZGBQmuz5v5iT6P\"\n + \"XCBeVmjEX3kh4bkRPHJ5vyASNXUkF3nwVAe4cwOoLHN8o=\");\n\n public InvalidPinnedAmplitudeClient() {\n super(Constants.DEFAULT_INSTANCE);\n super.getPinnedCertSslSocketFactory(INVALID_SSL_CONTEXT);\n }\n }\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/PlanTest.java",
"content": "package com.amplitude.api;\n\nimport androidx.test.ext.junit.runners.AndroidJUnit4;\n\nimport org.json.JSONException;\nimport org.json.JSONObject;\nimport org.junit.Test;\nimport org.junit.runner.RunWith;\nimport org.robolectric.annotation.Config;\n\nimport static org.junit.Assert.assertEquals;\n\n@RunWith(AndroidJUnit4.class)\n@Config(manifest=Config.NONE)\npublic class PlanTest {\n\n @Test\n public void testToJSONObject() throws JSONException {\n Plan testPlan = new Plan();\n String branch = \"main\";\n String version = \"1.0.0\";\n String source = \"mobile\";\n String versionId = \"9ec23ba0-275f-468f-80d1-66b88bff9529\";\n testPlan.setBranch(branch)\n .setSource(source)\n .setVersion(version)\n .setVersionId(versionId);\n JSONObject result = testPlan.toJSONObject();\n assertEquals(branch, result.getString(Constants.AMP_PLAN_BRANCH));\n assertEquals(source, result.getString(Constants.AMP_PLAN_SOURCE));\n assertEquals(version, result.getString(Constants.AMP_PLAN_VERSION));\n assertEquals(versionId, result.getString(Constants.AMP_PLAN_VERSION_ID));\n }\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/RevenueTest.java",
"content": "package com.amplitude.api;\n\nimport androidx.test.ext.junit.runners.AndroidJUnit4;\n\nimport org.json.JSONException;\nimport org.json.JSONObject;\nimport org.junit.After;\nimport org.junit.Before;\nimport org.junit.Test;\nimport org.junit.runner.RunWith;\nimport org.robolectric.annotation.Config;\n\nimport static org.junit.Assert.assertEquals;\nimport static org.junit.Assert.assertFalse;\nimport static org.junit.Assert.assertNotEquals;\nimport static org.junit.Assert.assertNull;\nimport static org.junit.Assert.assertTrue;\n\n@RunWith(AndroidJUnit4.class)\n@Config(manifest=Config.NONE)\npublic class RevenueTest extends BaseTest {\n\n @Before\n public void setUp() throws Exception { setUp(false); }\n\n @After\n public void tearDown() throws Exception {}\n\n @Test\n public void testProductId() {\n Revenue revenue = new Revenue();\n assertNull(revenue.productId);\n\n String productId = \"testProductId\";\n revenue.setProductId(productId);\n assertEquals(revenue.productId, productId);\n\n // test that ignore empty inputs\n revenue.setProductId(null);\n assertEquals(revenue.productId, productId);\n revenue.setProductId(\"\");\n assertEquals(revenue.productId, productId);\n\n JSONObject obj = revenue.toJSONObject();\n assertEquals(obj.optString(\"$productId\"), productId);\n }\n\n @Test\n public void testQuantity() {\n Revenue revenue = new Revenue();\n assertEquals(revenue.quantity, 1);\n\n int quantity = 100;\n revenue.setQuantity(quantity);\n assertEquals(revenue.quantity, quantity);\n\n JSONObject obj = revenue.toJSONObject();\n assertEquals(obj.optInt(\"$quantity\"), quantity);\n }\n\n @Test\n public void testPrice() {\n Revenue revenue = new Revenue();\n assertNull(revenue.price);\n\n double price = 10.99;\n revenue.setPrice(price);\n assertEquals(revenue.price.doubleValue(), price, 0);\n\n JSONObject obj = revenue.toJSONObject();\n assertEquals(obj.optDouble(\"$price\"), price, 0);\n }\n\n @Test\n public void testRevenueType() {\n Revenue revenue = new Revenue();\n assertEquals(revenue.revenueType, null);\n\n String revenueType = \"testRevenueType\";\n revenue.setRevenueType(revenueType);\n assertEquals(revenue.revenueType, revenueType);\n\n // verify that null and empty strings allowed\n revenue.setRevenueType(null);\n assertNull(revenue.revenueType);\n revenue.setRevenueType(\"\");\n assertEquals(revenue.revenueType, \"\");\n\n revenue.setRevenueType(revenueType);\n assertEquals(revenue.revenueType, revenueType);\n\n JSONObject obj = revenue.toJSONObject();\n assertEquals(obj.optString(\"$revenueType\"), revenueType);\n }\n\n @Test\n public void testReceipt() {\n Revenue revenue = new Revenue();\n assertNull(revenue.receipt);\n assertNull(revenue.receiptSig);\n\n String receipt = \"testReceipt\";\n String receiptSig = \"testReceiptSig\";\n revenue.setReceipt(receipt, receiptSig);\n assertEquals(revenue.receipt, receipt);\n assertEquals(revenue.receiptSig, receiptSig);\n\n JSONObject obj = revenue.toJSONObject();\n assertEquals(obj.optString(\"$receipt\"), receipt);\n assertEquals(obj.optString(\"$receiptSig\"), receiptSig);\n }\n\n @Test\n public void testRevenueProperties() throws JSONException {\n Revenue revenue = new Revenue();\n assertNull(revenue.properties);\n\n JSONObject properties = new JSONObject().put(\"city\", \"san francisco\");\n revenue.setRevenueProperties(properties);\n assertTrue(Utils.compareJSONObjects(properties, revenue.properties));\n\n JSONObject obj = revenue.toJSONObject();\n assertEquals(obj.optString(\"city\"), \"san francisco\");\n assertEquals(obj.optInt(\"$quantity\"), 1);\n\n // assert original json object was not modified\n assertFalse(properties.has(\"$quantity\"));\n }\n\n @Test\n public void testEventProperties() throws JSONException {\n Revenue revenue = new Revenue();\n assertNull(revenue.properties);\n\n JSONObject properties = new JSONObject().put(\"city\", \"san francisco\");\n revenue.setEventProperties(properties);\n assertTrue(Utils.compareJSONObjects(properties, revenue.properties));\n\n JSONObject obj = revenue.toJSONObject();\n assertEquals(obj.optString(\"city\"), \"san francisco\");\n assertEquals(obj.optInt(\"$quantity\"), 1);\n\n // assert original json object was not modified\n assertFalse(properties.has(\"$quantity\"));\n }\n\n @Test\n public void testValidRevenue() {\n Revenue revenue = new Revenue();\n assertFalse(revenue.isValidRevenue());\n revenue.setProductId(\"testProductId\");\n assertFalse(revenue.isValidRevenue());\n revenue.setPrice(10.99);\n assertTrue(revenue.isValidRevenue());\n\n Revenue revenue2 = new Revenue();\n assertFalse(revenue2.isValidRevenue());\n revenue2.setPrice(10.99);\n revenue2.setQuantity(15);\n assertTrue(revenue2.isValidRevenue());\n revenue2.setProductId(\"testProductId\");\n assertTrue(revenue2.isValidRevenue());\n }\n\n @Test\n public void testToJSONObject() throws JSONException {\n double price = 10.99;\n int quantity = 15;\n String productId = \"testProductId\";\n String receipt = \"testReceipt\";\n String receiptSig = \"testReceiptSig\";\n String revenueType = \"testRevenueType\";\n JSONObject props = new JSONObject().put(\"city\", \"Boston\");\n\n Revenue revenue = new Revenue().setProductId(productId).setPrice(price);\n revenue.setQuantity(quantity).setReceipt(receipt, receiptSig);\n revenue.setRevenueType(revenueType).setRevenueProperties(props);\n\n JSONObject obj = revenue.toJSONObject();\n assertEquals(obj.optDouble(\"$price\"), price, 0);\n assertEquals(obj.optInt(\"$quantity\"), 15);\n assertEquals(obj.optString(\"$productId\"), productId);\n assertEquals(obj.optString(\"$receipt\"), receipt);\n assertEquals(obj.optString(\"$receiptSig\"), receiptSig);\n assertEquals(obj.optString(\"$revenueType\"), revenueType);\n assertEquals(obj.optString(\"city\"), \"Boston\");\n }\n\n @Test\n public void testEquals() throws JSONException {\n Revenue r1 = new Revenue();\n Revenue r2 = new Revenue();\n\n r1.setPrice(10.00).setQuantity(2).setProductId(\"testProductId\");\n r1.setEventProperties(new JSONObject().put(\"testProp\", \"testValue\"));\n r2.setPrice(9.99).setQuantity(2).setProductId(\"testProductId\");\n r2.setEventProperties(new JSONObject().put(\"testProp\", \"testValue\"));\n assertFalse(r1.equals(r2));\n r2.setPrice(10.00);\n assertTrue(r1.equals(r2));\n r2.setEventProperties(new JSONObject().put(\"testProp\", \"fakeValue\"));\n assertFalse(r1.equals(r2));\n }\n\n @Test\n public void testHashCode() {\n Revenue r1 = new Revenue();\n Revenue r2 = new Revenue();\n\n r1.setPrice(10.00).setQuantity(2).setProductId(\"testProductId\");\n r2.setPrice(9.99).setQuantity(2).setProductId(\"testProductId\");\n assertNotEquals(r1.hashCode(), r2.hashCode());\n r2.setPrice(10.00);\n assertEquals(r1.hashCode(), r2.hashCode());\n }\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/SessionTest.java",
"content": "package com.amplitude.api;\n\nimport androidx.test.ext.junit.runners.AndroidJUnit4;\n\nimport org.json.JSONArray;\nimport org.json.JSONException;\nimport org.json.JSONObject;\nimport org.junit.After;\nimport org.junit.Before;\nimport org.junit.Test;\nimport org.junit.runner.RunWith;\nimport org.robolectric.Robolectric;\nimport org.robolectric.Shadows;\nimport org.robolectric.annotation.Config;\nimport org.robolectric.shadows.ShadowLooper;\n\nimport okhttp3.mockwebserver.RecordedRequest;\n\nimport static org.junit.Assert.assertEquals;\nimport static org.junit.Assert.assertFalse;\nimport static org.junit.Assert.assertNull;\nimport static org.junit.Assert.assertTrue;\n\n@RunWith(AndroidJUnit4.class)\n@Config(manifest = Config.NONE)\npublic class SessionTest extends BaseTest {\n\n // allows for control of System.currentTimeMillis\n private class AmplitudeCallbacksWithTime extends AmplitudeCallbacks {\n\n private int index;\n private long [] timestamps = null;\n\n public AmplitudeCallbacksWithTime(AmplitudeClient client, long [] timestamps) {\n super(client);\n this.index = 0;\n this.timestamps = timestamps;\n }\n\n @Override\n protected long getCurrentTimeMillis() {\n return timestamps[index++ % timestamps.length];\n }\n }\n\n @Before\n public void setUp() throws Exception {\n super.setUp(true);\n amplitude.initialize(context, apiKey);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runOneTask();\n }\n\n @After\n public void tearDown() throws Exception {\n super.tearDown();\n }\n\n @Test\n public void testDefaultStartSession() {\n long timestamp = System.currentTimeMillis();\n amplitude.logEventAsync(\"test\", null, null, null, null, null, timestamp, false);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n\n // trackSessionEvents is false, no start_session event added\n assertEquals(getUnsentEventCount(), 1);\n JSONObject event = getLastUnsentEvent();\n assertEquals(event.optString(\"event_type\"), \"test\");\n assertEquals(event.optString(\"session_id\"), String.valueOf(timestamp));\n }\n\n @Test\n public void testDefaultTriggerNewSession() {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n long sessionTimeoutMillis = 5 * 1000; //5s\n amplitude.setSessionTimeoutMillis(sessionTimeoutMillis);\n\n // log 1st event, initialize first session\n long timestamp1 = System.currentTimeMillis();\n amplitude.logEventAsync(\"test1\", null, null, null, null, null, timestamp1, false);\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 1);\n\n // log 2nd event past timeout, verify new session started\n long timestamp2 = timestamp1 + sessionTimeoutMillis;\n amplitude.logEventAsync(\"test2\", null, null, null, null, null, timestamp2, false);\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 2);\n\n JSONArray events = getUnsentEvents(2);\n JSONObject event1 = events.optJSONObject(0);\n JSONObject event2 = events.optJSONObject(1);\n\n assertEquals(event1.optString(\"event_type\"), \"test1\");\n assertEquals(event1.optString(\"session_id\"), String.valueOf(timestamp1));\n assertEquals(event2.optString(\"event_type\"), \"test2\");\n assertEquals(event2.optString(\"session_id\"), String.valueOf(timestamp2));\n\n // also test getSessionId\n assertEquals(amplitude.getSessionId(), timestamp2);\n }\n\n @Test\n public void testDefaultExtendSession() {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n long sessionTimeoutMillis = 5 * 1000; //5s\n amplitude.setSessionTimeoutMillis(sessionTimeoutMillis);\n\n // log 3 events all just within session expiration window, verify all in same session\n long timestamp1 = System.currentTimeMillis();\n amplitude.logEventAsync(\"test1\", null, null, null, null, null, timestamp1, false);\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 1);\n\n long timestamp2 = timestamp1 + sessionTimeoutMillis - 1;\n amplitude.logEventAsync(\"test2\", null, null, null, null, null, timestamp2, false);\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 2);\n\n long timestamp3 = timestamp2 + sessionTimeoutMillis - 1;\n amplitude.logEventAsync(\"test3\", null, null, null, null, null, timestamp3, false);\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 3);\n\n JSONArray events = getUnsentEvents(3);\n JSONObject event1 = events.optJSONObject(0);\n JSONObject event2 = events.optJSONObject(1);\n JSONObject event3 = events.optJSONObject(2);\n\n assertEquals(event1.optString(\"event_type\"), \"test1\");\n assertEquals(event1.optString(\"session_id\"), String.valueOf(timestamp1));\n assertEquals(event1.optString(\"timestamp\"), String.valueOf(timestamp1));\n\n assertEquals(event2.optString(\"event_type\"), \"test2\");\n assertEquals(event2.optString(\"session_id\"), String.valueOf(timestamp1));\n assertEquals(event2.optString(\"timestamp\"), String.valueOf(timestamp2));\n\n assertEquals(event3.optString(\"event_type\"), \"test3\");\n assertEquals(event3.optString(\"session_id\"), String.valueOf(timestamp1));\n assertEquals(event3.optString(\"timestamp\"), String.valueOf(timestamp3));\n }\n\n @Test\n public void testDefaultStartSessionWithTracking() {\n amplitude.trackSessionEvents(true);\n\n long timestamp = System.currentTimeMillis();\n amplitude.logEventAsync(\"test\", null, null, null, null, null, timestamp, false);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n\n // trackSessions is true, start_session event is added\n assertEquals(getUnsentEventCount(), 2);\n JSONArray events = getUnsentEvents(2);\n JSONObject session_event = events.optJSONObject(0);\n JSONObject test_event = events.optJSONObject(1);\n\n assertEquals(session_event.optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT);\n assertEquals(\n session_event.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.START_SESSION_EVENT\n );\n assertEquals(session_event.optString(\"session_id\"), String.valueOf(timestamp));\n\n assertEquals(test_event.optString(\"event_type\"), \"test\");\n assertEquals(test_event.optString(\"session_id\"), String.valueOf(timestamp));\n }\n\n @Test\n public void testDefaultStartSessionWithTrackingSynchronous() {\n amplitude.trackSessionEvents(true);\n\n long timestamp = System.currentTimeMillis();\n amplitude.logEventSync(\"test\", null, null, timestamp, false);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n // trackSessions is true, start_session event is added\n assertEquals(getUnsentEventCount(), 2);\n\n // verify order of synchronous events\n JSONArray events = getUnsentEvents(2);\n JSONObject session_event = events.optJSONObject(0);\n JSONObject test_event = events.optJSONObject(1);\n\n assertEquals(session_event.optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT);\n assertEquals(\n session_event.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.START_SESSION_EVENT\n );\n assertEquals(session_event.optString(\"session_id\"), String.valueOf(timestamp));\n\n assertEquals(test_event.optString(\"event_type\"), \"test\");\n assertEquals(test_event.optString(\"session_id\"), String.valueOf(timestamp));\n }\n\n @Test\n public void testDefaultTriggerNewSessionWithTracking() {\n amplitude.trackSessionEvents(true);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n long sessionTimeoutMillis = 5 * 1000; //5s\n amplitude.setSessionTimeoutMillis(sessionTimeoutMillis);\n\n // log 1st event, initialize first session\n long timestamp1 = System.currentTimeMillis();\n amplitude.logEventAsync(\"test1\", null, null, null, null, null, timestamp1, false);\n looper.runToEndOfTasks();\n // trackSessions is true, start_session event is added\n assertEquals(getUnsentEventCount(), 2);\n\n // log 2nd event past timeout, verify new session started\n long timestamp2 = timestamp1 + sessionTimeoutMillis;\n amplitude.logEventAsync(\"test2\", null, null, null, null, null, timestamp2, false);\n looper.runToEndOfTasks();\n // trackSessions is true, end_session and start_session events are added\n assertEquals(getUnsentEventCount(), 5);\n\n JSONArray events = getUnsentEvents(5);\n JSONObject startSession1 = events.optJSONObject(0);\n JSONObject event1 = events.optJSONObject(1);\n JSONObject endSession = events.optJSONObject(2);\n JSONObject startSession2 = events.optJSONObject(3);\n JSONObject event2 = events.optJSONObject(4);\n\n assertEquals(startSession1.optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT);\n assertEquals(\n startSession1.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.START_SESSION_EVENT\n );\n assertEquals(startSession1.optString(\"session_id\"), String.valueOf(timestamp1));\n\n assertEquals(event1.optString(\"event_type\"), \"test1\");\n assertEquals(event1.optString(\"session_id\"), String.valueOf(timestamp1));\n assertEquals(event1.optString(\"timestamp\"), String.valueOf(timestamp1));\n\n assertEquals(endSession.optString(\"event_type\"), AmplitudeClient.END_SESSION_EVENT);\n assertEquals(\n endSession.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.END_SESSION_EVENT\n );\n assertEquals(endSession.optString(\"session_id\"), String.valueOf(timestamp1));\n\n assertEquals(startSession2.optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT);\n assertEquals(\n startSession2.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.START_SESSION_EVENT\n );\n assertEquals(startSession2.optString(\"session_id\"), String.valueOf(timestamp2));\n\n assertEquals(event2.optString(\"event_type\"), \"test2\");\n assertEquals(event2.optString(\"session_id\"), String.valueOf(timestamp2));\n assertEquals(event2.optString(\"timestamp\"), String.valueOf(timestamp2));\n }\n\n @Test\n public void testDefaultTriggerNewSessionWithTrackingSynchronous() {\n amplitude.trackSessionEvents(true);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n long sessionTimeoutMillis = 5 * 1000; //5s\n amplitude.setSessionTimeoutMillis(sessionTimeoutMillis);\n\n // log 1st event, initialize first session\n long timestamp1 = System.currentTimeMillis();\n amplitude.logEventSync(\"test1\", null, null, timestamp1, false);\n looper.runToEndOfTasks();\n // trackSessions is true, start_session event is added\n assertEquals(getUnsentEventCount(), 2);\n\n // log 2nd event past timeout, verify new session started\n long timestamp2 = timestamp1 + sessionTimeoutMillis;\n amplitude.logEventSync(\"test2\", null, null, timestamp2, false);\n looper.runToEndOfTasks();\n // trackSessions is true, end_session and start_session events are added\n assertEquals(getUnsentEventCount(), 5);\n\n // verify order of synchronous events\n JSONArray events = getUnsentEvents(5);\n JSONObject startSession1 = events.optJSONObject(0);\n JSONObject event1 = events.optJSONObject(1);\n JSONObject endSession = events.optJSONObject(2);\n JSONObject startSession2 = events.optJSONObject(3);\n JSONObject event2 = events.optJSONObject(4);\n\n assertEquals(startSession1.optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT);\n assertEquals(\n startSession1.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.START_SESSION_EVENT\n );\n assertEquals(startSession1.optString(\"session_id\"), String.valueOf(timestamp1));\n\n assertEquals(event1.optString(\"event_type\"), \"test1\");\n assertEquals(event1.optString(\"session_id\"), String.valueOf(timestamp1));\n assertEquals(event1.optString(\"timestamp\"), String.valueOf(timestamp1));\n\n assertEquals(endSession.optString(\"event_type\"), AmplitudeClient.END_SESSION_EVENT);\n assertEquals(\n endSession.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.END_SESSION_EVENT\n );\n assertEquals(endSession.optString(\"session_id\"), String.valueOf(timestamp1));\n\n assertEquals(startSession2.optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT);\n assertEquals(\n startSession2.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.START_SESSION_EVENT\n );\n assertEquals(startSession2.optString(\"session_id\"), String.valueOf(timestamp2));\n\n assertEquals(event2.optString(\"event_type\"), \"test2\");\n assertEquals(event2.optString(\"session_id\"), String.valueOf(timestamp2));\n assertEquals(event2.optString(\"timestamp\"), String.valueOf(timestamp2));\n }\n\n @Test\n public void testDefaultExtendSessionWithTracking() {\n amplitude.trackSessionEvents(true);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n long sessionTimeoutMillis = 5 * 1000; //5s\n amplitude.setSessionTimeoutMillis(sessionTimeoutMillis);\n\n // log 3 events all just within session expiration window, verify all in same session\n long timestamp1 = System.currentTimeMillis();\n amplitude.logEventAsync(\"test1\", null, null, null, null, null, timestamp1, false);\n looper.runToEndOfTasks();\n // trackSessions is true, start_session event is added\n assertEquals(getUnsentEventCount(), 2);\n\n long timestamp2 = timestamp1 + sessionTimeoutMillis - 1;\n amplitude.logEventAsync(\"test2\", null, null, null, null, null, timestamp2, false);\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 3);\n\n long timestamp3 = timestamp2 + sessionTimeoutMillis - 1;\n amplitude.logEventAsync(\"test3\", null, null, null, null, null, timestamp3, false);\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 4);\n\n JSONArray events = getUnsentEvents(4);\n JSONObject startSession = events.optJSONObject(0);\n JSONObject event1 = events.optJSONObject(1);\n JSONObject event2 = events.optJSONObject(2);\n JSONObject event3 = events.optJSONObject(3);\n\n assertEquals(startSession.optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT);\n assertEquals(\n startSession.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.START_SESSION_EVENT\n );\n assertEquals(startSession.optString(\"session_id\"), String.valueOf(timestamp1));\n\n assertEquals(event1.optString(\"event_type\"), \"test1\");\n assertEquals(event1.optString(\"session_id\"), String.valueOf(timestamp1));\n assertEquals(event1.optString(\"timestamp\"), String.valueOf(timestamp1));\n\n assertEquals(event2.optString(\"event_type\"), \"test2\");\n assertEquals(event2.optString(\"session_id\"), String.valueOf(timestamp1));\n assertEquals(event2.optString(\"timestamp\"), String.valueOf(timestamp2));\n\n assertEquals(event3.optString(\"event_type\"), \"test3\");\n assertEquals(event3.optString(\"session_id\"), String.valueOf(timestamp1));\n assertEquals(event3.optString(\"timestamp\"), String.valueOf(timestamp3));\n }\n\n @Test\n public void testDefaultExtendSessionWithTrackingSynchronous() {\n amplitude.trackSessionEvents(true);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n long sessionTimeoutMillis = 5 * 1000; //5s\n amplitude.setSessionTimeoutMillis(sessionTimeoutMillis);\n\n // log 3 events all just within session expiration window, verify all in same session\n long timestamp1 = System.currentTimeMillis();\n amplitude.logEventSync(\"test1\", null, null, timestamp1, false);\n looper.runToEndOfTasks();\n // trackSessions is true, start_session event is added\n assertEquals(getUnsentEventCount(), 2);\n\n long timestamp2 = timestamp1 + sessionTimeoutMillis - 1;\n amplitude.logEventSync(\"test2\", null, null, timestamp2, false);\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 3);\n\n long timestamp3 = timestamp2 + sessionTimeoutMillis - 1;\n amplitude.logEventAsync(\"test3\", null, null, null, null, null, timestamp3, false);\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 4);\n\n // verify order of synchronous events\n JSONArray events = getUnsentEvents(4);\n JSONObject startSession = events.optJSONObject(0);\n JSONObject event1 = events.optJSONObject(1);\n JSONObject event2 = events.optJSONObject(2);\n JSONObject event3 = events.optJSONObject(3);\n\n assertEquals(startSession.optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT);\n assertEquals(\n startSession.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.START_SESSION_EVENT\n );\n assertEquals(startSession.optString(\"session_id\"), String.valueOf(timestamp1));\n\n assertEquals(event1.optString(\"event_type\"), \"test1\");\n assertEquals(event1.optString(\"session_id\"), String.valueOf(timestamp1));\n assertEquals(event1.optString(\"timestamp\"), String.valueOf(timestamp1));\n\n assertEquals(event2.optString(\"event_type\"), \"test2\");\n assertEquals(event2.optString(\"session_id\"), String.valueOf(timestamp1));\n assertEquals(event2.optString(\"timestamp\"), String.valueOf(timestamp2));\n\n assertEquals(event3.optString(\"event_type\"), \"test3\");\n assertEquals(event3.optString(\"session_id\"), String.valueOf(timestamp1));\n assertEquals(event3.optString(\"timestamp\"), String.valueOf(timestamp3));\n }\n\n @Test\n public void testEnableAccurateTracking() {\n assertFalse(amplitude.isUsingForegroundTracking());\n AmplitudeCallbacks callBacks = new AmplitudeCallbacks(amplitude);\n assertTrue(amplitude.isUsingForegroundTracking());\n }\n\n @Test\n public void testAccurateOnResumeStartSession() {\n long timestamp = System.currentTimeMillis();\n long [] timestamps = {timestamp};\n AmplitudeCallbacks callBacks = new AmplitudeCallbacksWithTime(amplitude, timestamps);\n\n assertEquals(amplitude.previousSessionId, -1);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, -1);\n assertFalse(amplitude.isInForeground());\n\n callBacks.onActivityResumed(null);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n assertTrue(amplitude.isInForeground());\n assertEquals(amplitude.previousSessionId, timestamp);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, timestamp);\n }\n\n @Test\n public void testAccurateOnResumeStartSessionWithTracking() {\n amplitude.trackSessionEvents(true);\n long timestamp = System.currentTimeMillis();\n long [] timestamps = {timestamp};\n AmplitudeCallbacks callBacks = new AmplitudeCallbacksWithTime(amplitude, timestamps);\n\n assertEquals(amplitude.previousSessionId, -1);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, -1);\n assertFalse(amplitude.isInForeground());\n assertEquals(getUnsentEventCount(), 0);\n\n callBacks.onActivityResumed(null);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n assertTrue(amplitude.isInForeground());\n assertEquals(amplitude.previousSessionId, timestamp);\n assertEquals(amplitude.lastEventId, 1);\n assertEquals(amplitude.lastEventTime, timestamp);\n\n // verify that start session event sent\n assertEquals(getUnsentEventCount(), 1);\n JSONObject startSession = getLastUnsentEvent();\n assertEquals(startSession.optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT);\n assertEquals(\n startSession.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.START_SESSION_EVENT\n );\n assertEquals(\n startSession.optString(\"session_id\"),\n String.valueOf(timestamp)\n );\n assertEquals(\n startSession.optString(\"timestamp\"),\n String.valueOf(timestamp)\n );\n }\n\n @Test\n public void testAccurateOnPauseRefreshTimestamp() {\n long minTimeBetweenSessionsMillis = 5*1000; //5s\n long timestamp = System.currentTimeMillis();\n long [] timestamps = {timestamp, timestamp + minTimeBetweenSessionsMillis};\n AmplitudeCallbacks callBacks = new AmplitudeCallbacksWithTime(amplitude, timestamps);\n\n assertEquals(amplitude.previousSessionId, -1);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, -1);\n\n callBacks.onActivityResumed(null);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamps[0]);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, timestamps[0]);\n\n callBacks.onActivityPaused(null);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamps[0]);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, timestamps[1]);\n assertFalse(amplitude.isInForeground());\n }\n\n @Test\n public void testAccurateOnPauseRefreshTimestampWithTracking() {\n amplitude.trackSessionEvents(true);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n long minTimeBetweenSessionsMillis = 5*1000; //5s\n amplitude.setMinTimeBetweenSessionsMillis(minTimeBetweenSessionsMillis);\n long timestamp = System.currentTimeMillis();\n long [] timestamps = {timestamp, timestamp + minTimeBetweenSessionsMillis};\n AmplitudeCallbacks callBacks = new AmplitudeCallbacksWithTime(amplitude, timestamps);\n\n assertEquals(amplitude.previousSessionId, -1);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, -1);\n assertEquals(getUnsentEventCount(), 0);\n\n callBacks.onActivityResumed(null);\n looper.runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamps[0]);\n assertEquals(amplitude.lastEventId, 1);\n assertEquals(amplitude.lastEventTime, timestamps[0]);\n assertEquals(getUnsentEventCount(), 1);\n\n // only refresh time, no session checking\n callBacks.onActivityPaused(null);\n looper.runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamps[0]);\n assertEquals(amplitude.lastEventId, 1);\n assertEquals(amplitude.lastEventTime, timestamps[1]);\n assertEquals(getUnsentEventCount(), 1);\n }\n\n @Test\n public void testAccurateOnResumeTriggerNewSession() {\n long minTimeBetweenSessionsMillis = 5*1000; //5s\n amplitude.setMinTimeBetweenSessionsMillis(minTimeBetweenSessionsMillis);\n long timestamp = System.currentTimeMillis();\n long [] timestamps = {\n timestamp,\n timestamp + 1,\n timestamp + 1 + minTimeBetweenSessionsMillis\n };\n AmplitudeCallbacks callBacks = new AmplitudeCallbacksWithTime(amplitude, timestamps);\n\n assertEquals(amplitude.previousSessionId, -1);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, -1);\n assertEquals(getUnsentEventCount(), 0);\n\n callBacks.onActivityResumed(null);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamps[0]);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, timestamps[0]);\n assertEquals(getUnsentEventCount(), 0);\n assertTrue(amplitude.isInForeground());\n\n // only refresh time, no session checking\n callBacks.onActivityPaused(null);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamps[0]);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, timestamps[1]);\n assertEquals(getUnsentEventCount(), 0);\n assertFalse(amplitude.isInForeground());\n\n // resume after min session expired window, verify new session started\n callBacks.onActivityResumed(null);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamps[2]);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, timestamps[2]);\n assertEquals(getUnsentEventCount(), 0);\n assertTrue(amplitude.isInForeground());\n }\n\n @Test\n public void testAccurateOnResumeTriggerNewSessionWithTracking() {\n amplitude.trackSessionEvents(true);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n long minTimeBetweenSessionsMillis = 5*1000; //5s\n amplitude.setMinTimeBetweenSessionsMillis(minTimeBetweenSessionsMillis);\n long timestamp = System.currentTimeMillis();\n long [] timestamps = {\n timestamp,\n timestamp + 1,\n timestamp + 1 + minTimeBetweenSessionsMillis\n };\n AmplitudeCallbacks callBacks = new AmplitudeCallbacksWithTime(amplitude, timestamps);\n\n assertEquals(amplitude.previousSessionId, -1);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, -1);\n assertEquals(getUnsentEventCount(), 0);\n\n callBacks.onActivityResumed(null);\n looper.runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamps[0]);\n assertEquals(amplitude.lastEventId, 1);\n assertEquals(amplitude.lastEventTime, timestamps[0]);\n assertEquals(getUnsentEventCount(), 1);\n assertTrue(amplitude.isInForeground());\n\n // only refresh time, no session checking\n callBacks.onActivityPaused(null);\n looper.runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamps[0]);\n assertEquals(amplitude.lastEventId, 1);\n assertEquals(amplitude.lastEventTime, timestamps[1]);\n assertEquals(getUnsentEventCount(), 1);\n assertFalse(amplitude.isInForeground());\n\n // resume after min session expired window, verify new session started\n callBacks.onActivityResumed(null);\n looper.runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamps[2]);\n assertEquals(amplitude.lastEventId, 3);\n assertEquals(amplitude.lastEventTime, timestamps[2]);\n assertEquals(getUnsentEventCount(), 3);\n assertTrue(amplitude.isInForeground());\n\n JSONArray events = getUnsentEvents(3);\n JSONObject startSession1 = events.optJSONObject(0);\n JSONObject endSession = events.optJSONObject(1);\n JSONObject startSession2 = events.optJSONObject(2);\n\n assertEquals(startSession1.optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT);\n assertEquals(\n startSession1.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.START_SESSION_EVENT\n );\n assertEquals(startSession1.optString(\"session_id\"), String.valueOf(timestamps[0]));\n assertEquals(startSession1.optString(\"timestamp\"), String.valueOf(timestamps[0]));\n\n assertEquals(endSession.optString(\"event_type\"), AmplitudeClient.END_SESSION_EVENT);\n assertEquals(\n endSession.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.END_SESSION_EVENT\n );\n assertEquals(endSession.optString(\"session_id\"), String.valueOf(timestamps[0]));\n assertEquals(endSession.optString(\"timestamp\"), String.valueOf(timestamps[1]));\n\n assertEquals(startSession2.optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT);\n assertEquals(\n startSession2.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.START_SESSION_EVENT\n );\n assertEquals(startSession2.optString(\"session_id\"), String.valueOf(timestamps[2]));\n assertEquals(startSession2.optString(\"timestamp\"), String.valueOf(timestamps[2]));\n }\n\n @Test\n public void testAccurateOnResumeExtendSession() {\n long minTimeBetweenSessionsMillis = 5*1000; //5s\n amplitude.setMinTimeBetweenSessionsMillis(minTimeBetweenSessionsMillis);\n long timestamp = System.currentTimeMillis();\n long [] timestamps = {\n timestamp,\n timestamp + 1,\n timestamp + 1 + minTimeBetweenSessionsMillis - 1 // just inside session exp window\n };\n AmplitudeCallbacks callBacks = new AmplitudeCallbacksWithTime(amplitude, timestamps);\n\n assertEquals(amplitude.previousSessionId, -1);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, -1);\n\n callBacks.onActivityResumed(null);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamps[0]);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, timestamps[0]);\n\n callBacks.onActivityPaused(null);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamps[0]);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, timestamps[1]);\n assertFalse(amplitude.isInForeground());\n\n callBacks.onActivityResumed(null);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamps[0]);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, timestamps[2]);\n assertTrue(amplitude.isInForeground());\n }\n\n @Test\n public void testAccurateOnResumeExtendSessionWithTracking() {\n amplitude.trackSessionEvents(true);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n long minTimeBetweenSessionsMillis = 5*1000; //5s\n amplitude.setMinTimeBetweenSessionsMillis(minTimeBetweenSessionsMillis);\n long timestamp = System.currentTimeMillis();\n long [] timestamps = {\n timestamp,\n timestamp + 1,\n timestamp + 1 + minTimeBetweenSessionsMillis - 1 // just inside session exp window\n };\n AmplitudeCallbacks callBacks = new AmplitudeCallbacksWithTime(amplitude, timestamps);\n\n assertEquals(amplitude.previousSessionId, -1);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, -1);\n assertEquals(getUnsentEventCount(), 0);\n\n callBacks.onActivityResumed(null);\n looper.runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamps[0]);\n assertEquals(amplitude.lastEventId, 1);\n assertEquals(amplitude.lastEventTime, timestamps[0]);\n assertEquals(getUnsentEventCount(), 1);\n\n callBacks.onActivityPaused(null);\n looper.runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamps[0]);\n assertEquals(amplitude.lastEventId, 1);\n assertEquals(amplitude.lastEventTime, timestamps[1]);\n assertFalse(amplitude.isInForeground());\n assertEquals(getUnsentEventCount(), 1);\n\n callBacks.onActivityResumed(null);\n looper.runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamps[0]);\n assertEquals(amplitude.lastEventId, 1);\n assertEquals(amplitude.lastEventTime, timestamps[2]);\n assertTrue(amplitude.isInForeground());\n assertEquals(getUnsentEventCount(), 1);\n\n JSONObject event = getLastUnsentEvent();\n assertEquals(event.optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT);\n assertEquals(\n event.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.START_SESSION_EVENT\n );\n assertEquals(event.optString(\"session_id\"), String.valueOf(timestamps[0]));\n assertEquals(event.optString(\"timestamp\"), String.valueOf(timestamps[0]));\n }\n\n @Test\n public void testAccurateLogAsyncEvent() {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n long minTimeBetweenSessionsMillis = 5*1000; //5s\n amplitude.setMinTimeBetweenSessionsMillis(minTimeBetweenSessionsMillis);\n long timestamp = System.currentTimeMillis();\n long [] timestamps = {timestamp + minTimeBetweenSessionsMillis - 1};\n AmplitudeCallbacks callBacks = new AmplitudeCallbacksWithTime(amplitude, timestamps);\n\n assertEquals(amplitude.previousSessionId, -1);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, -1);\n assertEquals(getUnsentEventCount(), 0);\n assertFalse(amplitude.isInForeground());\n\n // logging an event before onResume will force a session check\n amplitude.logEventAsync(\"test\", null, null, null, null, null, timestamp, false);\n looper.runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamp);\n assertEquals(amplitude.lastEventId, 1);\n assertEquals(amplitude.lastEventTime, timestamp);\n assertEquals(getUnsentEventCount(), 1);\n\n callBacks.onActivityResumed(null);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamp);\n assertEquals(amplitude.lastEventId, 1);\n assertEquals(amplitude.lastEventTime, timestamps[0]);\n assertEquals(getUnsentEventCount(), 1);\n assertTrue(amplitude.isInForeground());\n\n JSONObject event = getLastUnsentEvent();\n assertEquals(event.optString(\"event_type\"), \"test\");\n assertEquals(event.optString(\"session_id\"), String.valueOf(timestamp));\n assertEquals(event.optString(\"timestamp\"), String.valueOf(timestamp));\n }\n\n @Test\n public void testAccurateLogAsyncEventWithTracking() {\n amplitude.trackSessionEvents(true);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n long minTimeBetweenSessionsMillis = 5*1000; //5s\n amplitude.setMinTimeBetweenSessionsMillis(minTimeBetweenSessionsMillis);\n long timestamp = System.currentTimeMillis();\n long [] timestamps = {timestamp + minTimeBetweenSessionsMillis};\n AmplitudeCallbacks callBacks = new AmplitudeCallbacksWithTime(amplitude, timestamps);\n\n assertEquals(amplitude.previousSessionId, -1);\n assertEquals(amplitude.lastEventId, -1);\n assertEquals(amplitude.lastEventTime, -1);\n assertEquals(getUnsentEventCount(), 0);\n assertFalse(amplitude.isInForeground());\n\n // logging an event before onResume will force a session check\n amplitude.logEventAsync(\"test\", null, null, null, null, null, timestamp, false);\n looper.runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamp);\n assertEquals(amplitude.lastEventId, 2);\n assertEquals(amplitude.lastEventTime, timestamp);\n assertEquals(getUnsentEventCount(), 2);\n\n // onResume after session expires will start new session\n callBacks.onActivityResumed(null);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n assertEquals(amplitude.previousSessionId, timestamps[0]);\n assertEquals(amplitude.lastEventId, 4);\n assertEquals(amplitude.lastEventTime, timestamps[0]);\n assertEquals(getUnsentEventCount(), 4);\n assertTrue(amplitude.isInForeground());\n\n JSONArray events = getUnsentEvents(4);\n JSONObject startSession1 = events.optJSONObject(0);\n JSONObject event = events.optJSONObject(1);\n JSONObject endSession = events.optJSONObject(2);\n JSONObject startSession2 = events.optJSONObject(3);\n\n assertEquals(startSession1.optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT);\n assertEquals(\n startSession1.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.START_SESSION_EVENT\n );\n assertEquals(startSession1.optString(\"session_id\"), String.valueOf(timestamp));\n assertEquals(startSession1.optString(\"timestamp\"), String.valueOf(timestamp));\n\n assertEquals(event.optString(\"event_type\"), \"test\");\n assertEquals(event.optString(\"session_id\"), String.valueOf(timestamp));\n assertEquals(event.optString(\"timestamp\"), String.valueOf(timestamp));\n\n assertEquals(endSession.optString(\"event_type\"), AmplitudeClient.END_SESSION_EVENT);\n assertEquals(\n endSession.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.END_SESSION_EVENT\n );\n assertEquals(endSession.optString(\"session_id\"), String.valueOf(timestamp));\n assertEquals(endSession.optString(\"timestamp\"), String.valueOf(timestamp));\n\n assertEquals(startSession2.optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT);\n assertEquals(\n startSession2.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.START_SESSION_EVENT\n );\n assertEquals(startSession2.optString(\"session_id\"), String.valueOf(timestamps[0]));\n assertEquals(startSession2.optString(\"timestamp\"), String.valueOf(timestamps[0]));\n }\n\n\n @Test\n public void testLogOutOfSessionEvent() {\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n long sessionTimeoutMillis = 5*1000; //1s\n amplitude.setSessionTimeoutMillis(sessionTimeoutMillis);\n\n long timestamp1 = System.currentTimeMillis();\n amplitude.logEventAsync(\"test1\", null, null, null, null, null, timestamp1, false);\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 1);\n\n // log out of session event just within session expiration window\n long timestamp2 = timestamp1 + sessionTimeoutMillis - 1;\n amplitude.logEventAsync(\"outOfSession\", null, null, null, null, null, timestamp2, true);\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 2);\n\n // out of session events do not extend session, 2nd event will start new session\n long timestamp3 = timestamp1 + sessionTimeoutMillis;\n amplitude.logEventAsync(\"test2\", null, null, null, null, null, timestamp3, false);\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 3);\n\n JSONArray events = getUnsentEvents(3);\n JSONObject event1 = events.optJSONObject(0);\n JSONObject outOfSessionEvent = events.optJSONObject(1);\n JSONObject event2 = events.optJSONObject(2);\n\n assertEquals(event1.optString(\"event_type\"), \"test1\");\n assertEquals(event1.optString(\"session_id\"), String.valueOf(timestamp1));\n assertEquals(outOfSessionEvent.optString(\"event_type\"), \"outOfSession\");\n assertEquals(outOfSessionEvent.optString(\"session_id\"), String.valueOf(-1));\n assertEquals(event2.optString(\"event_type\"), \"test2\");\n assertEquals(event2.optString(\"session_id\"), String.valueOf(timestamp3));\n }\n\n @Test\n public void testOnPauseFlushEvents() throws JSONException {\n long timestamp = System.currentTimeMillis();\n long [] timestamps = {\n timestamp, timestamp + 1, timestamp + 2,\n timestamp + 3, timestamp + 4, timestamp + 5,\n };\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n AmplitudeCallbacks callBacks = new AmplitudeCallbacksWithTime(amplitude, timestamps);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n // log an event, should not be uploaded\n amplitude.logEventAsync(\"testEvent\", null, null, null, null, null, timestamps[0], false);\n looper.runOneTask();\n looper.runOneTask();\n assertEquals(getUnsentEventCount(), 1);\n\n // force client into background and verify flushing of events\n callBacks.onActivityPaused(null);\n looper.runOneTask(); // run the update server\n RecordedRequest request = runRequest(amplitude);\n JSONArray events = getEventsFromRequest(request);\n assertEquals(events.length(), 1);\n assertEquals(events.getJSONObject(0).optString(\"event_type\"), \"testEvent\");\n\n // verify that events have been cleared from client\n looper.runOneTask();\n assertEquals(getUnsentEventCount(), 0);\n }\n\n @Test\n public void testOnPauseFlushEventsDisabled() throws JSONException {\n long timestamp = System.currentTimeMillis();\n long [] timestamps = {\n timestamp, timestamp + 1, timestamp + 2,\n timestamp + 3, timestamp + 4, timestamp + 5,\n };\n amplitude.setFlushEventsOnClose(false);\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n AmplitudeCallbacks callBacks = new AmplitudeCallbacksWithTime(amplitude, timestamps);\n Robolectric.getForegroundThreadScheduler().advanceTo(1);\n\n // log an event, should not be uploaded\n amplitude.logEventAsync(\"testEvent\", null, null, null, null, null, timestamps[0], false);\n looper.runOneTask();\n assertEquals(getUnsentEventCount(), 1);\n\n // force client into background and verify no flushing of events\n callBacks.onActivityPaused(null);\n looper.runOneTask(); // run the update server\n RecordedRequest request = runRequest(amplitude);\n\n // flushing disabled, so no request should be sent\n assertNull(request);\n assertEquals(getUnsentEventCount(), 1);\n }\n\n @Test\n public void testIdentifyTriggerNewSession() throws JSONException {\n amplitude.trackSessionEvents(true);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n long sessionTimeoutMillis = 5 * 1000; //5s\n amplitude.setSessionTimeoutMillis(sessionTimeoutMillis);\n\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 0);\n\n // log 1st identify, initialize first session\n Identify identify = new Identify().set(\"key\", \"value\");\n amplitude.identify(identify);\n looper.runToEndOfTasks();\n // trackSessions is true, start_session event is added\n assertEquals(getUnsentEventCount(), 1);\n assertEquals(getUnsentIdentifyCount(), 0);\n\n JSONArray events = getUnsentEvents(1);\n assertEquals(\n events.getJSONObject(0).optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT\n );\n looper.runToEndOfTasks();\n JSONArray identifies = getUnsentIdentifys(1);\n JSONObject expected = new JSONObject().put(\"$set\", new JSONObject().put(\"key\", \"value\"));\n assertTrue(Utils.compareJSONObjects(\n identifies.getJSONObject(0).getJSONObject(\"user_properties\"), expected\n ));\n }\n\n @Test\n public void testOutOfSessionIdentifyDoesNotTriggerNewSession() throws JSONException {\n amplitude.trackSessionEvents(true);\n\n ShadowLooper looper = Shadows.shadowOf(amplitude.logThread.getLooper());\n long sessionTimeoutMillis = 5 * 1000; //5s\n amplitude.setSessionTimeoutMillis(sessionTimeoutMillis);\n\n assertEquals(getUnsentEventCount(), 0);\n assertEquals(getUnsentIdentifyCount(), 0);\n\n // log 1st identify, initialize first session\n Identify identify = new Identify().set(\"key\", \"value\");\n amplitude.identify(identify, true);\n looper.runToEndOfTasks();\n assertEquals(getUnsentEventCount(), 0); // out of session, start session is not added\n assertEquals(getUnsentIdentifyCount(), 0);\n looper.runToEndOfTasks();\n assertEquals(getUnsentIdentifyCount(), 1);\n\n JSONArray identifies = getUnsentIdentifys(1);\n JSONObject expected = new JSONObject().put(\"$set\", new JSONObject().put(\"key\", \"value\"));\n assertTrue(Utils.compareJSONObjects(\n identifies.getJSONObject(0).getJSONObject(\"user_properties\"), expected\n ));\n }\n\n @Test\n public void testSetUserIdAndStartNewSessionWithTracking() {\n amplitude.trackSessionEvents(true);\n\n long timestamp = System.currentTimeMillis();\n amplitude.logEventAsync(\"test\", null, null, null, null, null, timestamp, false);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n\n // trackSessions is true, start_session event is added\n assertEquals(getUnsentEventCount(), 2);\n\n // set user id and validate session ended and new session started\n amplitude.setUserId(\"test_new_user\", true);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n\n // total of 4 events, start session, test event, end session, start session\n assertEquals(getUnsentEventCount(), 4);\n JSONArray events = getUnsentEvents(4);\n\n // verify pre setUserId events\n JSONObject session_event = events.optJSONObject(0);\n JSONObject test_event = events.optJSONObject(1);\n assertEquals(session_event.optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT);\n assertEquals(session_event.optString(\"user_id\"), \"null\");\n assertEquals(\n session_event.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.START_SESSION_EVENT\n );\n assertEquals(session_event.optString(\"session_id\"), String.valueOf(timestamp));\n\n assertEquals(test_event.optString(\"event_type\"), \"test\");\n assertEquals(test_event.optString(\"session_id\"), String.valueOf(timestamp));\n assertEquals(test_event.optString(\"user_id\"), \"null\");\n\n // verify post setUserId events\n session_event = events.optJSONObject(2);\n assertEquals(session_event.optString(\"event_type\"), AmplitudeClient.END_SESSION_EVENT);\n assertEquals(session_event.optString(\"user_id\"), \"null\");\n assertEquals(\n session_event.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.END_SESSION_EVENT\n );\n assertEquals(session_event.optString(\"session_id\"), String.valueOf(timestamp));\n\n session_event = events.optJSONObject(3);\n assertEquals(session_event.optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT);\n assertEquals(session_event.optString(\"user_id\"), \"test_new_user\");\n assertEquals(\n session_event.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.START_SESSION_EVENT\n );\n\n // the new event should have a newer session id\n assertTrue(session_event.optLong(\"session_id\") > timestamp);\n }\n\n @Test\n public void testSetUserIdAndDoNotStartNewSessionWithTracking() {\n amplitude.trackSessionEvents(true);\n\n long timestamp = System.currentTimeMillis();\n amplitude.logEventAsync(\"test\", null, null, null, null, null, timestamp, false);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n\n // trackSessions is true, start_session event is added\n assertEquals(getUnsentEventCount(), 2);\n\n // set user id and validate session ended and new session started\n amplitude.setUserId(\"test_new_user\", false);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n\n // still only 2 events, start session, test event\n assertEquals(getUnsentEventCount(), 2);\n JSONArray events = getUnsentEvents(2);\n\n // verify pre setUserId events\n JSONObject session_event = events.optJSONObject(0);\n JSONObject test_event = events.optJSONObject(1);\n assertEquals(session_event.optString(\"event_type\"), AmplitudeClient.START_SESSION_EVENT);\n assertEquals(session_event.optString(\"user_id\"), \"null\");\n assertEquals(\n session_event.optJSONObject(\"api_properties\").optString(\"special\"),\n AmplitudeClient.START_SESSION_EVENT\n );\n assertEquals(session_event.optString(\"session_id\"), String.valueOf(timestamp));\n\n assertEquals(test_event.optString(\"event_type\"), \"test\");\n assertEquals(test_event.optString(\"session_id\"), String.valueOf(timestamp));\n assertEquals(test_event.optString(\"user_id\"), \"null\");\n\n // verify same session id\n assertEquals(amplitude.sessionId, timestamp);\n }\n\n @Test\n public void testSetUserIdAndStartNewSessionWithoutTracking() {\n amplitude.trackSessionEvents(false);\n\n long timestamp = System.currentTimeMillis();\n amplitude.logEventAsync(\"test\", null, null, null, null, null, timestamp, false);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n\n // trackSessions is false, there should only be 1 event\n assertEquals(getUnsentEventCount(), 1);\n\n // set user id and validate session ended and new session started\n amplitude.setUserId(\"test_new_user\", true);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n\n // still only 1 event1, test event\n assertEquals(getUnsentEventCount(), 1);\n JSONArray events = getUnsentEvents(1);\n\n // verify pre setUserId events\n JSONObject session_event = events.optJSONObject(0);\n assertEquals(session_event.optString(\"event_type\"), \"test\");\n assertEquals(session_event.optString(\"user_id\"), \"null\");\n assertEquals(session_event.optString(\"session_id\"), String.valueOf(timestamp));\n\n // log an event with new user id and session\n amplitude.logEventAsync(\"test\", null, null, null, null, null, timestamp, false);\n Shadows.shadowOf(amplitude.logThread.getLooper()).runToEndOfTasks();\n\n // verify post set user id\n assertEquals(getUnsentEventCount(), 2);\n JSONObject test_event = getLastEvent();\n assertEquals(test_event.optString(\"event_type\"), \"test\");\n assertEquals(test_event.optString(\"user_id\"), \"test_new_user\");\n assertEquals(test_event.optLong(\"session_id\"), amplitude.sessionId);\n\n // there should be a new session id at least\n assertTrue(amplitude.sessionId > timestamp);\n assertTrue(test_event.optLong(\"session_id\") > timestamp);\n }\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/TrackingOptionsTest.java",
"content": "package com.amplitude.api;\n\nimport androidx.test.ext.junit.runners.AndroidJUnit4;\n\nimport org.json.JSONException;\nimport org.json.JSONObject;\nimport org.junit.After;\nimport org.junit.Before;\nimport org.junit.Test;\nimport org.junit.runner.RunWith;\nimport org.robolectric.annotation.Config;\n\nimport java.util.HashSet;\nimport java.util.Set;\n\nimport static org.junit.Assert.assertEquals;\nimport static org.junit.Assert.assertFalse;\nimport static org.junit.Assert.assertTrue;\n\n@RunWith(AndroidJUnit4.class)\n@Config(manifest= Config.NONE)\npublic class TrackingOptionsTest extends BaseTest {\n\n @Before\n public void setUp() throws Exception { setUp(false); }\n\n @After\n public void tearDown() throws Exception {}\n\n @Test\n public void testDisableFields() {\n TrackingOptions options = new TrackingOptions().disableCity().disableCountry().disableIpAddress().disableLanguage().disableLatLng();\n\n Set expectedDisabledFields = new HashSet();\n expectedDisabledFields.add(\"city\");\n expectedDisabledFields.add(\"country\");\n expectedDisabledFields.add(\"ip_address\");\n expectedDisabledFields.add(\"language\");\n expectedDisabledFields.add(\"lat_lng\");\n\n assertEquals(options.disabledFields, expectedDisabledFields);\n assertTrue(options.shouldTrackCarrier());\n assertFalse(options.shouldTrackCity());\n assertFalse(options.shouldTrackCountry());\n assertTrue(options.shouldTrackDeviceBrand());\n assertTrue(options.shouldTrackDeviceManufacturer());\n assertTrue(options.shouldTrackDeviceModel());\n assertTrue(options.shouldTrackDma());\n assertFalse(options.shouldTrackIpAddress());\n assertFalse(options.shouldTrackLanguage());\n assertFalse(options.shouldTrackLatLng());\n assertTrue(options.shouldTrackOsName());\n assertTrue(options.shouldTrackOsVersion());\n assertTrue(options.shouldTrackPlatform());\n assertTrue(options.shouldTrackRegion());\n assertTrue(options.shouldTrackVersionName());\n }\n\n @Test\n public void testGetApiPropertiesTrackingOptions() throws JSONException {\n TrackingOptions options = new TrackingOptions().disableCity().disableCountry().disableIpAddress().disableLanguage().disableLatLng();\n\n JSONObject expectedOptions = new JSONObject();\n expectedOptions.put(\"city\", false);\n expectedOptions.put(\"country\", false);\n expectedOptions.put(\"ip_address\", false);\n expectedOptions.put(\"lat_lng\", false);\n\n assertTrue(Utils.compareJSONObjects(options.getApiPropertiesTrackingOptions(), expectedOptions));\n }\n\n @Test\n public void testGetCoppaControlTrackingOptions() {\n TrackingOptions options = TrackingOptions.forCoppaControl();\n assertFalse(options.shouldTrackAdid());\n assertFalse(options.shouldTrackCity());\n assertFalse(options.shouldTrackIpAddress());\n assertFalse(options.shouldTrackLatLng());\n }\n\n @Test\n public void testMerging() {\n TrackingOptions options1 = TrackingOptions.forCoppaControl();\n TrackingOptions options2 = new TrackingOptions().disableCountry().disableLanguage().disableAppSetId();\n options1.mergeIn(options2);\n assertFalse(options1.shouldTrackAdid());\n assertFalse(options1.shouldTrackCity());\n assertFalse(options1.shouldTrackIpAddress());\n assertFalse(options1.shouldTrackCountry());\n assertFalse(options1.shouldTrackLanguage());\n assertFalse(options1.shouldTrackAppSetId());\n }\n\n @Test\n public void testCopyOf() {\n TrackingOptions options = TrackingOptions.copyOf(TrackingOptions.forCoppaControl());\n assertFalse(options.shouldTrackAdid());\n assertFalse(options.shouldTrackCity());\n assertFalse(options.shouldTrackIpAddress());\n assertFalse(options.shouldTrackLatLng());\n }\n\n @Test\n public void testEquals() {\n TrackingOptions options1 = new TrackingOptions();\n options1.disableAdid().disableCarrier().disableAppSetId();\n TrackingOptions options2 = new TrackingOptions();\n options2.disableAdid().disableCarrier().disableAppSetId();\n assertEquals(options1, options2);\n }\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/util/MockHttpURLConnectionHelper.java",
"content": "package com.amplitude.api.util;\n\nimport static org.mockito.Mockito.mock;\nimport static org.mockito.Mockito.when;\n\nimport java.io.ByteArrayInputStream;\nimport java.io.IOException;\nimport java.io.InputStream;\nimport java.io.OutputStream;\nimport java.net.HttpURLConnection;\nimport java.nio.charset.StandardCharsets;\n\npublic class MockHttpURLConnectionHelper {\n\n public static HttpURLConnection getMockHttpURLConnection(int code, String response)\n throws IOException {\n HttpURLConnection connection = mock(HttpURLConnection.class);\n OutputStream outputStream = mock(OutputStream.class);\n when(connection.getOutputStream()).thenReturn(outputStream);\n when(connection.getResponseCode()).thenReturn(code);\n InputStream inputStream = new ByteArrayInputStream(response.getBytes(StandardCharsets.UTF_8));\n if (code == 200) {\n when(connection.getInputStream()).thenReturn(inputStream);\n } else {\n when(connection.getErrorStream()).thenReturn(inputStream);\n }\n return connection;\n }\n}\n"
},
{
"path": "src/test/java/com/amplitude/api/util/MockURLStreamHandler.java",
"content": "package com.amplitude.api.util;\n\nimport java.io.IOException;\nimport java.net.URL;\nimport java.net.URLConnection;\nimport java.net.URLStreamHandler;\nimport java.net.URLStreamHandlerFactory;\nimport java.util.HashMap;\nimport java.util.Map;\n\npublic class MockURLStreamHandler extends URLStreamHandler implements URLStreamHandlerFactory {\n\n private final Map connections = new HashMap<>();\n\n private static final MockURLStreamHandler instance = new MockURLStreamHandler();\n\n public static MockURLStreamHandler getInstance() {\n return instance;\n }\n\n @Override\n protected URLConnection openConnection(URL url) throws IOException {\n return connections.get(url);\n }\n\n public void resetConnections() {\n connections.clear();\n }\n\n public MockURLStreamHandler setConnection(URL url, URLConnection urlConnection) {\n connections.put(url, urlConnection);\n return this;\n }\n\n @Override\n public URLStreamHandler createURLStreamHandler(String protocol) {\n return getInstance();\n }\n}\n"
}
]
![](\"https://static.amplitude.com/lightning/46c85bfd91905de8047f1ee65c7c93d6fa9ee6ea/static/media/amplitude-logo-with-text.4fb9e463.svg\"){kind=logo}
\n \n