@@ -10,15 +10,15 @@ import com.mparticle.MParticle
1010import com.mparticle.MParticle.IdentityType
1111import com.mparticle.MParticleOptions
1212import com.mparticle.MParticleOptions.DataplanOptions
13+ import com.mparticle.WrapperSdk
14+ import com.mparticle.WrapperSdkVersion
1315import com.mparticle.identity.IdentityApi
1416import com.mparticle.internal.CoreCallbacks
1517import com.mparticle.internal.CoreCallbacks.KitListener
16- import com.mparticle.kits.mocks.MockKitConfiguration
18+ import com.rokt.roktsdk.Rokt
1719import io.mockk.*
1820import org.json.JSONArray
19- import org.json.JSONObject
2021import org.junit.Assert.assertEquals
21- import org.junit.Assert.assertFalse
2222import org.junit.Assert.assertTrue
2323import org.junit.Before
2424import org.junit.Test
@@ -45,14 +45,14 @@ class RoktKitTests {
4545 every { context.applicationContext } returns context
4646 roktKit = RoktKit ()
4747 kitManager = TestKitManager ()
48- MParticle .setInstance(Mockito . mock(MParticle ::class .java))
48+ MParticle .setInstance(mock(MParticle ::class .java))
4949 Mockito .`when `(MParticle .getInstance()?.Identity ()).thenReturn(
50- Mockito . mock(
50+ mock(
5151 IdentityApi ::class .java
5252 )
5353 )
5454 val kitManager = KitManagerImpl (
55- Mockito . mock(
55+ mock(
5656 Context ::class .java
5757 ), null , emptyCoreCallbacks, mock(MParticleOptions ::class .java)
5858 )
@@ -100,7 +100,7 @@ class RoktKitTests {
100100
101101 @Test
102102 fun test_addIdentityAttributes_When_userIdentities_IS_Null (){
103- val mockFilterUser = Mockito . mock(FilteredMParticleUser ::class .java)
103+ val mockFilterUser = mock(FilteredMParticleUser ::class .java)
104104 val userIdentities = HashMap <IdentityType , String >()
105105 Mockito .`when `(mockFilterUser.userIdentities).thenReturn(userIdentities)
106106 val attributes: Map <String , String > = mapOf (
@@ -120,12 +120,11 @@ class RoktKitTests {
120120 assertTrue(result.containsKey(" key1" ))
121121 assertTrue(result.containsKey(" key2" ))
122122 assertTrue(result.containsKey(" key3" ))
123-
124123 }
125124
126125 @Test
127126 fun test_addIdentityAttributes_When_userIdentities_Contain_value (){
128- val mockFilterUser = Mockito . mock(FilteredMParticleUser ::class .java)
127+ val mockFilterUser = mock(FilteredMParticleUser ::class .java)
129128 val userIdentities = HashMap <IdentityType , String >()
130129 userIdentities.put(
IdentityType .
Email ,
" [email protected] " )
131130 Mockito .`when `(mockFilterUser.userIdentities).thenReturn(userIdentities)
@@ -152,7 +151,7 @@ class RoktKitTests {
152151
153152 @Test
154153 fun test_addIdentityAttributes_When_userIdentities_And_attributes_contains_same_key (){
155- val mockFilterUser = Mockito . mock(FilteredMParticleUser ::class .java)
154+ val mockFilterUser = mock(FilteredMParticleUser ::class .java)
156155 val userIdentities = HashMap <IdentityType , String >()
157156 userIdentities.put(
IdentityType .
Email ,
" [email protected] " )
158157 Mockito .`when `(mockFilterUser.userIdentities).thenReturn(userIdentities)
@@ -200,7 +199,7 @@ class RoktKitTests {
200199
201200 @Test
202201 fun testAddIdentityAttributes_nullAttributes_validUser () {
203- val mockFilterUser = Mockito . mock(FilteredMParticleUser ::class .java)
202+ val mockFilterUser = mock(FilteredMParticleUser ::class .java)
204203 val userIdentities = HashMap <IdentityType , String >()
205204 userIdentities.put(
IdentityType .
Email ,
" [email protected] " )
206205 Mockito .`when `(mockFilterUser.userIdentities).thenReturn(userIdentities)
@@ -235,7 +234,27 @@ class RoktKitTests {
235234 assertTrue(result.containsKey(" key3" ))
236235 }
237236
237+ @Test
238+ fun testSetSdkWrapper_correctlySetsRoktFramework () {
239+ mockkObject(Rokt )
240+ every { Rokt .setFrameworkType(any()) } just runs
241+
242+ roktKit.setWrapperSdkVersion(WrapperSdkVersion (WrapperSdk .WrapperFlutter , " 1.0.0" ))
243+ roktKit.setWrapperSdkVersion(WrapperSdkVersion (WrapperSdk .WrapperSdkReactNative , " 1.0.0" ))
244+ roktKit.setWrapperSdkVersion(WrapperSdkVersion (WrapperSdk .WrapperSdkCordova , " 1.0.0" ))
245+ roktKit.setWrapperSdkVersion(WrapperSdkVersion (WrapperSdk .WrapperNone , " 1.0.0" ))
246+ roktKit.setWrapperSdkVersion(WrapperSdkVersion (WrapperSdk .WrapperXamarin , " 1.0.0" ))
247+
248+ verifyOrder {
249+ Rokt .setFrameworkType(Rokt .SdkFrameworkType .Flutter )
250+ Rokt .setFrameworkType(Rokt .SdkFrameworkType .ReactNative )
251+ Rokt .setFrameworkType(Rokt .SdkFrameworkType .Cordova )
252+ Rokt .setFrameworkType(Rokt .SdkFrameworkType .Android )
253+ Rokt .setFrameworkType(Rokt .SdkFrameworkType .Android )
254+ }
238255
256+ unmockkObject(Rokt )
257+ }
239258
240259 internal inner class TestCoreCallbacks : CoreCallbacks {
241260 override fun isBackgrounded (): Boolean = false
0 commit comments