09b0c2a630
* init marketing service Fix linting error Create offerdb implementation Create offers service Add update method Create offer table and migration Fix linting error fix conflicts Insert new data Change duration to have clear indication to be based on days add error wrapper Change from using uuid to int for id field * Create Marketing service * make error virable name more readable * add condition in update service method to check offer status * generate lock file Change get to listAllOffers * Add method for getting current offer wip * add check for expires_at in update method * Fix conflicts * add copyright header * Fix linting error * only allow update to active offers * add isDefault argument to GetCurrent * Update lock file * add migration file * finish migrate for adding credit_in_cents for both award and invitee * save 100 years as expiration date for default offers * create crud test for offers * add GetCurrent test * modify doc * Fix GetCurrent to work with default offer * fix linting issue * add more tests and address feedbacks * fix migration file * add type column back to match with mockup design * add type column back to match with mockup design * move doc changes to new pr * add comments * change GetCurrent to GetCurrentByType * fix typo
106 lines
3.2 KiB
Go
106 lines
3.2 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information
|
|
|
|
package marketing_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"storj.io/storj/internal/testcontext"
|
|
"storj.io/storj/internal/testplanet"
|
|
"storj.io/storj/satellite/marketing"
|
|
)
|
|
|
|
func TestOffer_Database(t *testing.T) {
|
|
testplanet.Run(t, testplanet.Config{
|
|
SatelliteCount: 1,
|
|
}, func(t *testing.T, ctx *testcontext.Context, planet *testplanet.Planet) {
|
|
// Happy path
|
|
validOffers := []marketing.NewOffer{
|
|
{
|
|
Name: "test",
|
|
Description: "test offer 1",
|
|
AwardCreditInCents: 100,
|
|
InviteeCreditInCents: 50,
|
|
AwardCreditDurationDays: 60,
|
|
InviteeCreditDurationDays: 30,
|
|
RedeemableCap: 50,
|
|
ExpiresAt: time.Now().UTC().Add(time.Hour * 1),
|
|
Status: marketing.Active,
|
|
Type: marketing.Referral,
|
|
},
|
|
{
|
|
Name: "test",
|
|
Description: "test offer 2",
|
|
AwardCreditInCents: 100,
|
|
InviteeCreditInCents: 50,
|
|
AwardCreditDurationDays: 60,
|
|
InviteeCreditDurationDays: 30,
|
|
RedeemableCap: 50,
|
|
ExpiresAt: time.Now().UTC().Add(time.Hour * 1),
|
|
Status: marketing.Default,
|
|
Type: marketing.FreeCredit,
|
|
},
|
|
}
|
|
|
|
for i := range validOffers {
|
|
new, err := planet.Satellites[0].DB.Marketing().Offers().Create(ctx, &validOffers[i])
|
|
require.NoError(t, err)
|
|
|
|
all, err := planet.Satellites[0].DB.Marketing().Offers().ListAll(ctx)
|
|
require.NoError(t, err)
|
|
require.Contains(t, all, *new)
|
|
|
|
c, err := planet.Satellites[0].DB.Marketing().Offers().GetCurrentByType(ctx, new.Type)
|
|
require.NoError(t, err)
|
|
require.Equal(t, new, c)
|
|
|
|
update := &marketing.UpdateOffer{
|
|
ID: new.ID,
|
|
Status: marketing.Done,
|
|
NumRedeemed: new.NumRedeemed,
|
|
ExpiresAt: time.Now(),
|
|
}
|
|
err = planet.Satellites[0].DB.Marketing().Offers().Update(ctx, update)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
// create with expired offer
|
|
expiredOffers := []marketing.NewOffer{
|
|
{
|
|
Name: "test",
|
|
Description: "test offer",
|
|
AwardCreditInCents: 100,
|
|
InviteeCreditInCents: 50,
|
|
AwardCreditDurationDays: 60,
|
|
InviteeCreditDurationDays: 30,
|
|
RedeemableCap: 50,
|
|
ExpiresAt: time.Now().UTC().Add(time.Hour * -1),
|
|
Status: marketing.Active,
|
|
Type: marketing.FreeCredit,
|
|
},
|
|
{
|
|
Name: "test",
|
|
Description: "test offer",
|
|
AwardCreditInCents: 100,
|
|
InviteeCreditInCents: 50,
|
|
AwardCreditDurationDays: 60,
|
|
InviteeCreditDurationDays: 30,
|
|
RedeemableCap: 50,
|
|
ExpiresAt: time.Now().UTC().Add(time.Hour * -1),
|
|
Status: marketing.Default,
|
|
Type: marketing.Referral,
|
|
},
|
|
}
|
|
|
|
for i := range expiredOffers {
|
|
output, err := planet.Satellites[0].DB.Marketing().Offers().Create(ctx, &expiredOffers[i])
|
|
require.Error(t, err)
|
|
require.Nil(t, output)
|
|
}
|
|
})
|
|
}
|