@@ -272,7 +272,10 @@ class RealSubscriptionsTest {
272
272
subscriptions.launchPrivacyPro(context, " https://duckduckgo.com/pro?featurePage=duckai" .toUri())
273
273
274
274
verify(globalActivityStarter, times(2 )).startIntent(eq(context), captor.capture())
275
- assertEquals(" ${subscriptionsUrlProvider.buyUrl} ?featurePage=duckai" , (captor.lastValue as SubscriptionsWebViewActivityWithParams ).url)
275
+ assertEquals(
276
+ subscriptionsUrlProvider.buyUrl.appendQueryParams(" featurePage=duckai" ),
277
+ (captor.lastValue as SubscriptionsWebViewActivityWithParams ).url,
278
+ )
276
279
}
277
280
278
281
@Test
@@ -285,7 +288,7 @@ class RealSubscriptionsTest {
285
288
286
289
verify(globalActivityStarter, times(2 )).startIntent(eq(context), captor.capture())
287
290
assertEquals(
288
- " ${ subscriptionsUrlProvider.buyUrl} ? usePaidDuckAi=true&featurePage=duckai" ,
291
+ subscriptionsUrlProvider.buyUrl.appendQueryParams( " usePaidDuckAi=true&featurePage=duckai" ) ,
289
292
(captor.lastValue as SubscriptionsWebViewActivityWithParams ).url,
290
293
)
291
294
}
@@ -299,7 +302,10 @@ class RealSubscriptionsTest {
299
302
subscriptions.launchPrivacyPro(context, " https://duckduckgo.com/subscriptions?featurePage=duckai" .toUri())
300
303
301
304
verify(globalActivityStarter, times(2 )).startIntent(eq(context), captor.capture())
302
- assertEquals(" ${subscriptionsUrlProvider.buyUrl} ?featurePage=duckai" , (captor.lastValue as SubscriptionsWebViewActivityWithParams ).url)
305
+ assertEquals(
306
+ subscriptionsUrlProvider.buyUrl.appendQueryParams(" featurePage=duckai" ),
307
+ (captor.lastValue as SubscriptionsWebViewActivityWithParams ).url,
308
+ )
303
309
}
304
310
305
311
@Test
@@ -312,7 +318,7 @@ class RealSubscriptionsTest {
312
318
313
319
verify(globalActivityStarter, times(2 )).startIntent(eq(context), captor.capture())
314
320
assertEquals(
315
- " ${ subscriptionsUrlProvider.buyUrl} ? usePaidDuckAi=true&featurePage=duckai" ,
321
+ subscriptionsUrlProvider.buyUrl.appendQueryParams( " usePaidDuckAi=true&featurePage=duckai" ) ,
316
322
(captor.lastValue as SubscriptionsWebViewActivityWithParams ).url,
317
323
)
318
324
}
@@ -342,4 +348,9 @@ class RealSubscriptionsTest {
342
348
private fun fakeIntent (): Intent {
343
349
return Intent ().also { it.addFlags(FLAG_ACTIVITY_NEW_TASK ) }
344
350
}
351
+
352
+ private fun String.appendQueryParams (queryParams : String ): String {
353
+ val separator = if (this .contains(" ?" )) " &" else " ?"
354
+ return this + separator + queryParams
355
+ }
345
356
}
0 commit comments