Skip to content

Commit

Permalink
Add unit tests for PaymentRepository
Browse files Browse the repository at this point in the history
  • Loading branch information
Pururun committed Sep 18, 2023
1 parent 4dfdd6d commit 1d926dc
Show file tree
Hide file tree
Showing 3 changed files with 124 additions and 17 deletions.
7 changes: 7 additions & 0 deletions android/lib/payment/build.gradle.kts
Original file line number Diff line number Diff line change
Expand Up @@ -44,4 +44,11 @@ dependencies {

// Billing
implementation(project(Dependencies.Mullvad.billingLib))

// Test dependencies
testImplementation(project(Dependencies.Mullvad.commonTestLib))
testImplementation(Dependencies.Kotlin.test)
testImplementation(Dependencies.KotlinX.coroutinesTest)
testImplementation(Dependencies.MockK.core)
testImplementation(Dependencies.junit)
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,117 @@
package net.mullvad.mullvadvpn.lib.payment

import io.mockk.coEvery
import io.mockk.every
import io.mockk.mockk
import kotlin.test.assertEquals
import kotlin.test.assertFalse
import kotlin.test.assertIs
import kotlinx.coroutines.flow.MutableSharedFlow
import kotlinx.coroutines.test.runBlockingTest
import kotlinx.coroutines.test.runTest
import net.mullvad.mullvadvpn.lib.billing.BillingRepository
import net.mullvad.mullvadvpn.lib.billing.model.BillingProduct
import net.mullvad.mullvadvpn.lib.billing.model.BillingPurchase
import net.mullvad.mullvadvpn.lib.billing.model.PurchaseEvent
import net.mullvad.mullvadvpn.lib.billing.model.PurchaseFlowResult
import net.mullvad.mullvadvpn.lib.billing.model.QueryProductResult
import net.mullvad.mullvadvpn.lib.common.test.TestCoroutineRule
import org.junit.After
import org.junit.Before
import org.junit.Rule
import org.junit.Test

class PaymentRepositoryTest {
@get:Rule val testCoroutineRule = TestCoroutineRule()

private val mockBillingRepository: BillingRepository = mockk()

private val purchaseEventFlow =
MutableSharedFlow<PurchaseEvent>(replay = 1, extraBufferCapacity = 1)

private lateinit var paymentRepository: PaymentRepository

@Before
fun setUp() {
every { mockBillingRepository.purchaseEvents } returns purchaseEventFlow
}

@After fun tearDown() {}

@Test
fun testQueryAvailablePaymentShowWebPayment() = runTest {
// Arrange
paymentRepository =
PaymentRepository(billingRepository = mockBillingRepository, showWebPayment = true)
coEvery { mockBillingRepository.queryProducts(any()) } returns
QueryProductResult.ItemUnavailable

// Act
val result = paymentRepository.queryAvailablePaymentTypes()

// Assert
assert(result.webPaymentAvailable)
}

@Test
fun testQueryAvailablePaymentHideWebPayment() = runTest {
// Arrange
paymentRepository =
PaymentRepository(billingRepository = mockBillingRepository, showWebPayment = false)
coEvery { mockBillingRepository.queryProducts(any()) } returns
QueryProductResult.ItemUnavailable

// Act
val result = paymentRepository.queryAvailablePaymentTypes()

// Assert
assertFalse(result.webPaymentAvailable)
}

@Test
fun testQueryAvailablePaymentProductsAvailable() = runTest {
// Arrange
paymentRepository =
PaymentRepository(billingRepository = mockBillingRepository, showWebPayment = false)
val mockProduct: BillingProduct = mockk()
coEvery { mockBillingRepository.queryProducts(any()) } returns
QueryProductResult.Ok(products = listOf(mockProduct))

// Act
val result = paymentRepository.queryAvailablePaymentTypes().billingPaymentAvailability

// Assert
assertIs<BillingPaymentAvailability.ProductsAvailable>(result)
assertEquals(mockProduct, result.products.first())
}

@Test
fun testQueryAvailablePaymentProductsUnavailable() = runTest {
// Arrange
paymentRepository =
PaymentRepository(billingRepository = mockBillingRepository, showWebPayment = false)
coEvery { mockBillingRepository.queryProducts(any()) } returns
QueryProductResult.ItemUnavailable

// Act
val result = paymentRepository.queryAvailablePaymentTypes().billingPaymentAvailability

// Assert
assertIs<BillingPaymentAvailability.ProductsUnavailable>(result)
}

@Test
fun testQueryAvailablePaymentBillingUnavailableError() = runTest {
// Arrange
paymentRepository =
PaymentRepository(billingRepository = mockBillingRepository, showWebPayment = false)
coEvery { mockBillingRepository.queryProducts(any()) } returns
QueryProductResult.BillingUnavailable

// Act
val result = paymentRepository.queryAvailablePaymentTypes().billingPaymentAvailability

// Assert
assertIs<BillingPaymentAvailability.Error.BillingUnavailable>(result)
}
}

This file was deleted.

0 comments on commit 1d926dc

Please sign in to comment.