diff --git a/broken-site/broken-site-impl/src/main/java/com/duckduckgo/brokensite/impl/RealBrokenSitePrompt.kt b/broken-site/broken-site-impl/src/main/java/com/duckduckgo/brokensite/impl/RealBrokenSitePrompt.kt index c6f7cba73b79..ad1f52bc6506 100644 --- a/broken-site/broken-site-impl/src/main/java/com/duckduckgo/brokensite/impl/RealBrokenSitePrompt.kt +++ b/broken-site/broken-site-impl/src/main/java/com/duckduckgo/brokensite/impl/RealBrokenSitePrompt.kt @@ -48,6 +48,7 @@ class RealBrokenSitePrompt @Inject constructor( override suspend fun userAcceptedPrompt() { if (!_featureEnabled) return brokenSiteReportRepository.resetDismissStreak() + // TODO (cbarreiro) Set next shown date based on limiting logic: https://app.asana.com/0/0/1208572901396846/f brokenSiteReportRepository.setNextShownDate(null) } diff --git a/broken-site/broken-site-impl/src/test/java/com/duckduckgo/brokensite/impl/RealBrokenSitePromptTest.kt b/broken-site/broken-site-impl/src/test/java/com/duckduckgo/brokensite/impl/RealBrokenSitePromptTest.kt index e0f4b978b3fa..4385eb24c8f8 100644 --- a/broken-site/broken-site-impl/src/test/java/com/duckduckgo/brokensite/impl/RealBrokenSitePromptTest.kt +++ b/broken-site/broken-site-impl/src/test/java/com/duckduckgo/brokensite/impl/RealBrokenSitePromptTest.kt @@ -90,9 +90,10 @@ class RealBrokenSitePromptTest { @Test fun whenUserAcceptedPromptThenResetDismissStreakAndSetNextShownDateToNull() = runTest { - testee.userAcceptedPrompt() fakeBrokenSitePromptRCFeature.self().setRawStoredState(State(true)) + testee.userAcceptedPrompt() + verify(mockBrokenSiteReportRepository).resetDismissStreak() verify(mockBrokenSiteReportRepository).setNextShownDate(null) }