event_test.go

package revcatgo

import (
	"encoding/json"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
	"gopkg.in/guregu/null.v4"
)

func TestNewEvent(t *testing.T) {
	b := []byte(`{"store":"APP_STORE", "type":"INITIAL_PURCHASE"}`)

	var event Event
	err := json.Unmarshal(b, &event)
	assert.Nil(t, err)

	assert.Equal(t, "APP_STORE", event.Store.String())
	assert.Equal(t, "INITIAL_PURCHASE", event.Type.String())

	b = []byte(`{"store":1}`)
	err = json.Unmarshal(b, &event)
	assert.Error(t, err)

	b = []byte(`{"store":null}`)
	err = json.Unmarshal(b, &event)
	assert.EqualError(t, err, "store is a required field")
}

func TestEventIsExpired(t *testing.T) {
	// 2020-12-10:10:00
	expirationAt, _ := newMilliseconds(null.IntFrom(1607595010000))
	event := &Event{ExpirationAt: *expirationAt}

	// 2020-12-15:10:00
	futureTime := time.Unix(1608027010000/1000, 0)
	// 2020-12-05:10:00
	pastTime := time.Unix(1607163010000/1000, 0)

	assert.True(t, event.IsExpired(0, &futureTime))
	assert.False(t, event.IsExpired(0, &pastTime))
	assert.False(t, event.IsExpired(time.Hour*168, &futureTime))
	assert.True(t, event.IsExpired(time.Hour*48, &futureTime))
}

const initialPurchaseRawJSON = `
{
   "product_id": "my.subscription.sandbox",
   "event_timestamp_ms": 1605256336738,
   "original_app_user_id": "$RCAnonymousID:0000000000000000000000000000000b",
   "expiration_at_ms": 1605256730251,
   "presented_offering_id": "default",
   "environment": "SANDBOX",
   "app_user_id": "$RCAnonymousID:0000000000000000000000000000000b",
   "subscriber_attributes": {
     "$gpsAdId": {
       "updated_at_ms": 1605236044746,
       "value": "0000-0b0a-4d05-b3fc-000000000000"
     },
     "$ip": {
       "value": "127.0.0.1",
       "updated_at_ms": 1605236044746
     }
   },
   "type": "INITIAL_PURCHASE",
   "currency": "USD",
   "entitlement_id": null,
   "purchased_at_ms": 1605256330385,
   "original_transaction_id": "GPA.0000-4204-5621-00000",
   "entitlement_ids": [
     "premium"
   ],
   "price_in_purchased_currency": 550,
   "takehome_percentage": 0.7,
   "store": "PLAY_STORE",
   "price": 5.233,
   "transaction_id": "GPA.0000-4204-5621-00000",
   "period_type": "NORMAL",
   "id": "00A23FAE-0DB8-42E2-A8DC-00000BCDF0D6",
   "aliases": [
     "$RCAnonymousID:0000000000000000000000000000000b"
   ]
}
`

func TestUnmarshalInitialPurchaseEvent(t *testing.T) {
	b := []byte(initialPurchaseRawJSON)

	var event Event
	err := json.Unmarshal(b, &event)
	assert.Nil(t, err)

	assert.Equal(t, "my.subscription.sandbox", event.ProductID)
	assert.Equal(t, "PLAY_STORE", event.Store.String())
	assert.Equal(t, "INITIAL_PURCHASE", event.Type.String())
	assert.Equal(t, "SANDBOX", event.Environment.String())
	assert.Equal(t, "$RCAnonymousID:0000000000000000000000000000000b", event.AppUserID)
	assert.Equal(t, float32(550), event.PriceInPurchasedCurrency)
	assert.Equal(t, int64(1605256730251), event.ExpirationAt.Int64())
	assert.True(t, event.HasEntitlementID("premium"))
	assert.False(t, event.HasEntitlementID("invalid_entitlement_id"))
}

const cancellationRawJSON = `
{
   "price": 0,
   "entitlement_ids": [
     "premium"
   ],
   "currency": "USD",
   "store": "PLAY_STORE",
   "product_id": "my.subscription.sandbox",
   "app_user_id": "$RCAnonymousID:0000000000000000000000000000000b",
   "original_transaction_id": "GPA.0000-4204-5621-00000",
   "type": "CANCELLATION",
   "presented_offering_id": "default",
   "purchased_at_ms": 1605258350251,
   "original_app_user_id": "$RCAnonymousID:0000000000000000000000000000000b",
   "id": "00A23FAE-0DB8-42E2-A8DC-00000BCDF0D6",
   "environment": "SANDBOX",
   "transaction_id": "GPA.0000-4204-5621-00000",
   "period_type": "NORMAL",
   "price_in_purchased_currency": 0,
   "subscriber_attributes": {
     "$gpsAdId": {
       "updated_at_ms": 1605236044746,
       "value": "0000-0b0a-4d05-b3fc-000000000000"
     },
     "$ip": {
       "value": "127.0.0.1",
       "updated_at_ms": 1605236044746
     }
   },
   "aliases": [
     "$RCAnonymousID:0000000000000000000000000000000b"
   ],
   "cancel_reason": "BILLING_ERROR",
   "entitlement_id": null,
   "event_timestamp_ms": 1605258535218,
   "takehome_percentage": 0.7,
   "expiration_at_ms": 1605258533373
}
`

func TestUnmarshalCancellationEvent(t *testing.T) {
	b := []byte(cancellationRawJSON)

	var event Event
	err := json.Unmarshal(b, &event)
	assert.Nil(t, err)

	assert.Equal(t, "BILLING_ERROR", event.CancelReason.String())
}

func TestEvent_GetAllRelatedUserID(t *testing.T) {
	e := &Event{AppUserID: "one", OriginalAppUserID: "two", Aliases: []string{"one", "two"}}
	assert.Equal(t, []string{"one", "two"}, e.GetAllRelatedUserID())

	e = &Event{AppUserID: "one", OriginalAppUserID: "one", Aliases: []string{"one"}}
	assert.Equal(t, []string{"one"}, e.GetAllRelatedUserID())

	e = &Event{AppUserID: "one"}
	assert.Equal(t, []string{"one"}, e.GetAllRelatedUserID())
}