packages/in_app_purchase/in_app_purchase_android/android/src/test/java/io/flutter/plugins/inapppurchase/TranslatorTest.java

// Copyright 2013 The Flutter Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package io.flutter.plugins.inapppurchase;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;

import androidx.annotation.NonNull;
import com.android.billingclient.api.AccountIdentifiers;
import com.android.billingclient.api.BillingClient;
import com.android.billingclient.api.BillingResult;
import com.android.billingclient.api.Purchase;
import com.android.billingclient.api.PurchaseHistoryRecord;
import com.android.billingclient.api.SkuDetails;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import org.json.JSONException;
import org.junit.Before;
import org.junit.Test;

public class TranslatorTest {
  private static final String SKU_DETAIL_EXAMPLE_JSON =
      "{\"productId\":\"example\",\"type\":\"inapp\",\"price\":\"$0.99\",\"price_amount_micros\":990000,\"price_currency_code\":\"USD\",\"title\":\"Example title\",\"description\":\"Example description.\",\"original_price\":\"$0.99\",\"original_price_micros\":990000}";
  private static final String PURCHASE_EXAMPLE_JSON =
      "{\"orderId\":\"foo\",\"packageName\":\"bar\",\"productId\":\"consumable\",\"purchaseTime\":11111111,\"purchaseState\":0,\"purchaseToken\":\"baz\",\"developerPayload\":\"dummy payload\",\"isAcknowledged\":\"true\", \"obfuscatedAccountId\":\"Account101\", \"obfuscatedProfileId\": \"Profile105\"}";

  @Before
  public void setup() {
    Locale locale = new Locale("en", "us");
    Locale.setDefault(locale);
  }

  @Test
  public void fromSkuDetail() throws JSONException {
    final SkuDetails expected = new SkuDetails(SKU_DETAIL_EXAMPLE_JSON);

    Map<String, Object> serialized = Translator.fromSkuDetail(expected);

    assertSerialized(expected, serialized);
  }

  @Test
  public void fromSkuDetailsList() throws JSONException {
    final String SKU_DETAIL_EXAMPLE_2_JSON =
        "{\"productId\":\"example2\",\"type\":\"inapp\",\"price\":\"$0.99\",\"price_amount_micros\":990000,\"price_currency_code\":\"USD\",\"title\":\"Example title\",\"description\":\"Example description.\",\"original_price\":\"$0.99\",\"original_price_micros\":990000}";
    final List<SkuDetails> expected =
        Arrays.asList(
            new SkuDetails(SKU_DETAIL_EXAMPLE_JSON), new SkuDetails(SKU_DETAIL_EXAMPLE_2_JSON));

    final List<HashMap<String, Object>> serialized = Translator.fromSkuDetailsList(expected);

    assertEquals(expected.size(), serialized.size());
    assertSerialized(expected.get(0), serialized.get(0));
    assertSerialized(expected.get(1), serialized.get(1));
  }

  @Test
  public void fromSkuDetailsList_null() {
    assertEquals(Collections.emptyList(), Translator.fromSkuDetailsList(null));
  }

  @Test
  public void fromPurchase() throws JSONException {
    final Purchase expected = new Purchase(PURCHASE_EXAMPLE_JSON, "signature");
    assertSerialized(expected, Translator.fromPurchase(expected));
  }

  @Test
  public void fromPurchaseWithoutAccountIds() throws JSONException {
    final Purchase expected =
        new PurchaseWithoutAccountIdentifiers(PURCHASE_EXAMPLE_JSON, "signature");
    Map<String, Object> serialized = Translator.fromPurchase(expected);
    assertNotNull(serialized.get("orderId"));
    assertNull(serialized.get("obfuscatedProfileId"));
    assertNull(serialized.get("obfuscatedAccountId"));
  }

  @Test
  public void fromPurchaseHistoryRecord() throws JSONException {
    final PurchaseHistoryRecord expected =
        new PurchaseHistoryRecord(PURCHASE_EXAMPLE_JSON, "signature");
    assertSerialized(expected, Translator.fromPurchaseHistoryRecord(expected));
  }

  @Test
  public void fromPurchasesHistoryRecordList() throws JSONException {
    final String purchase2Json =
        "{\"orderId\":\"foo2\",\"packageName\":\"bar\",\"productId\":\"consumable\",\"purchaseTime\":11111111,\"purchaseState\":0,\"purchaseToken\":\"baz\",\"developerPayload\":\"dummy payload\",\"isAcknowledged\":\"true\"}";
    final String signature = "signature";
    final List<PurchaseHistoryRecord> expected =
        Arrays.asList(
            new PurchaseHistoryRecord(PURCHASE_EXAMPLE_JSON, signature),
            new PurchaseHistoryRecord(purchase2Json, signature));

    final List<HashMap<String, Object>> serialized =
        Translator.fromPurchaseHistoryRecordList(expected);

    assertEquals(expected.size(), serialized.size());
    assertSerialized(expected.get(0), serialized.get(0));
    assertSerialized(expected.get(1), serialized.get(1));
  }

  @Test
  public void fromPurchasesHistoryRecordList_null() {
    assertEquals(Collections.emptyList(), Translator.fromPurchaseHistoryRecordList(null));
  }

  @Test
  public void fromPurchasesList() throws JSONException {
    final String purchase2Json =
        "{\"orderId\":\"foo2\",\"packageName\":\"bar\",\"productId\":\"consumable\",\"purchaseTime\":11111111,\"purchaseState\":0,\"purchaseToken\":\"baz\",\"developerPayload\":\"dummy payload\",\"isAcknowledged\":\"true\"}";
    final String signature = "signature";
    final List<Purchase> expected =
        Arrays.asList(
            new Purchase(PURCHASE_EXAMPLE_JSON, signature), new Purchase(purchase2Json, signature));

    final List<HashMap<String, Object>> serialized = Translator.fromPurchasesList(expected);

    assertEquals(expected.size(), serialized.size());
    assertSerialized(expected.get(0), serialized.get(0));
    assertSerialized(expected.get(1), serialized.get(1));
  }

  @Test
  public void fromPurchasesList_null() {
    assertEquals(Collections.emptyList(), Translator.fromPurchasesList(null));
  }

  @Test
  public void fromBillingResult() throws JSONException {
    BillingResult newBillingResult =
        BillingResult.newBuilder()
            .setDebugMessage("dummy debug message")
            .setResponseCode(BillingClient.BillingResponseCode.OK)
            .build();
    Map<String, Object> billingResultMap = Translator.fromBillingResult(newBillingResult);

    assertEquals(billingResultMap.get("responseCode"), newBillingResult.getResponseCode());
    assertEquals(billingResultMap.get("debugMessage"), newBillingResult.getDebugMessage());
  }

  @Test
  public void fromBillingResult_debugMessageNull() throws JSONException {
    BillingResult newBillingResult =
        BillingResult.newBuilder().setResponseCode(BillingClient.BillingResponseCode.OK).build();
    Map<String, Object> billingResultMap = Translator.fromBillingResult(newBillingResult);

    assertEquals(billingResultMap.get("responseCode"), newBillingResult.getResponseCode());
    assertEquals(billingResultMap.get("debugMessage"), newBillingResult.getDebugMessage());
  }

  @Test
  public void currencyCodeFromSymbol() {
    assertEquals("$", Translator.currencySymbolFromCode("USD"));
    try {
      Translator.currencySymbolFromCode("EUROPACOIN");
      fail("Translator should throw an exception");
    } catch (Exception e) {
      assertTrue(e instanceof IllegalArgumentException);
    }
  }

  private void assertSerialized(SkuDetails expected, Map<String, Object> serialized) {
    assertEquals(expected.getDescription(), serialized.get("description"));
    assertEquals(expected.getFreeTrialPeriod(), serialized.get("freeTrialPeriod"));
    assertEquals(expected.getIntroductoryPrice(), serialized.get("introductoryPrice"));
    assertEquals(
        expected.getIntroductoryPriceAmountMicros(),
        serialized.get("introductoryPriceAmountMicros"));
    assertEquals(expected.getIntroductoryPriceCycles(), serialized.get("introductoryPriceCycles"));
    assertEquals(expected.getIntroductoryPricePeriod(), serialized.get("introductoryPricePeriod"));
    assertEquals(expected.getPrice(), serialized.get("price"));
    assertEquals(expected.getPriceAmountMicros(), serialized.get("priceAmountMicros"));
    assertEquals(expected.getPriceCurrencyCode(), serialized.get("priceCurrencyCode"));
    assertEquals("$", serialized.get("priceCurrencySymbol"));
    assertEquals(expected.getSku(), serialized.get("sku"));
    assertEquals(expected.getSubscriptionPeriod(), serialized.get("subscriptionPeriod"));
    assertEquals(expected.getTitle(), serialized.get("title"));
    assertEquals(expected.getType(), serialized.get("type"));
    assertEquals(expected.getOriginalPrice(), serialized.get("originalPrice"));
    assertEquals(
        expected.getOriginalPriceAmountMicros(), serialized.get("originalPriceAmountMicros"));
  }

  private void assertSerialized(Purchase expected, Map<String, Object> serialized) {
    assertEquals(expected.getOrderId(), serialized.get("orderId"));
    assertEquals(expected.getPackageName(), serialized.get("packageName"));
    assertEquals(expected.getPurchaseTime(), serialized.get("purchaseTime"));
    assertEquals(expected.getPurchaseToken(), serialized.get("purchaseToken"));
    assertEquals(expected.getSignature(), serialized.get("signature"));
    assertEquals(expected.getOriginalJson(), serialized.get("originalJson"));
    assertEquals(expected.getSkus(), serialized.get("skus"));
    assertEquals(expected.getDeveloperPayload(), serialized.get("developerPayload"));
    assertEquals(expected.isAcknowledged(), serialized.get("isAcknowledged"));
    assertEquals(expected.getPurchaseState(), serialized.get("purchaseState"));
    assertNotNull(expected.getAccountIdentifiers().getObfuscatedAccountId());
    assertEquals(
        expected.getAccountIdentifiers().getObfuscatedAccountId(),
        serialized.get("obfuscatedAccountId"));
    assertNotNull(expected.getAccountIdentifiers().getObfuscatedProfileId());
    assertEquals(
        expected.getAccountIdentifiers().getObfuscatedProfileId(),
        serialized.get("obfuscatedProfileId"));
  }

  private void assertSerialized(PurchaseHistoryRecord expected, Map<String, Object> serialized) {
    assertEquals(expected.getPurchaseTime(), serialized.get("purchaseTime"));
    assertEquals(expected.getPurchaseToken(), serialized.get("purchaseToken"));
    assertEquals(expected.getSignature(), serialized.get("signature"));
    assertEquals(expected.getOriginalJson(), serialized.get("originalJson"));
    assertEquals(expected.getSkus(), serialized.get("skus"));
    assertEquals(expected.getDeveloperPayload(), serialized.get("developerPayload"));
  }
}

class PurchaseWithoutAccountIdentifiers extends Purchase {
  public PurchaseWithoutAccountIdentifiers(@NonNull String s, @NonNull String s1)
      throws JSONException {
    super(s, s1);
  }

  @Override
  public AccountIdentifiers getAccountIdentifiers() {
    return null;
  }
}