3b751a35c5
Removes database tables and functionality related to our custom coupon implementation because it has been superseded by the Stripe coupon and promo code system. Requires implementations of the payments Invoices interface to return coupon usages along with invoices. Change-Id: Iac52d2ff64afca8cc4dbb2d1f20e6ad4b39ddfde
132 lines
3.6 KiB
Go
132 lines
3.6 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package stripecoinpayments_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"storj.io/common/testcontext"
|
|
"storj.io/common/uuid"
|
|
"storj.io/storj/satellite"
|
|
"storj.io/storj/satellite/payments/stripecoinpayments"
|
|
"storj.io/storj/satellite/satellitedb/satellitedbtest"
|
|
)
|
|
|
|
func TestProjectRecords(t *testing.T) {
|
|
satellitedbtest.Run(t, func(ctx *testcontext.Context, t *testing.T, db satellite.DB) {
|
|
utc := time.Now().UTC()
|
|
|
|
prjID, err := uuid.New()
|
|
require.NoError(t, err)
|
|
|
|
start := time.Date(utc.Year(), utc.Month(), 1, 0, 0, 0, 0, time.UTC)
|
|
end := time.Date(utc.Year(), utc.Month()+1, 1, 0, 0, 0, 0, time.UTC)
|
|
|
|
projectRecordsDB := db.StripeCoinPayments().ProjectRecords()
|
|
|
|
t.Run("create", func(t *testing.T) {
|
|
err = projectRecordsDB.Create(ctx,
|
|
[]stripecoinpayments.CreateProjectRecord{
|
|
{
|
|
ProjectID: prjID,
|
|
Storage: 1,
|
|
Egress: 2,
|
|
Objects: 3,
|
|
},
|
|
},
|
|
start, end,
|
|
)
|
|
require.NoError(t, err)
|
|
})
|
|
|
|
t.Run("check", func(t *testing.T) {
|
|
err = projectRecordsDB.Check(ctx, prjID, start, end)
|
|
require.Error(t, err)
|
|
assert.Equal(t, stripecoinpayments.ErrProjectRecordExists, err)
|
|
})
|
|
|
|
page, err := projectRecordsDB.ListUnapplied(ctx, 0, 1, start, end)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(page.Records))
|
|
|
|
t.Run("consume", func(t *testing.T) {
|
|
err = projectRecordsDB.Consume(ctx, page.Records[0].ID)
|
|
require.NoError(t, err)
|
|
})
|
|
|
|
page, err = projectRecordsDB.ListUnapplied(ctx, 0, 1, start, end)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 0, len(page.Records))
|
|
})
|
|
}
|
|
|
|
func TestProjectRecordsList(t *testing.T) {
|
|
satellitedbtest.Run(t, func(ctx *testcontext.Context, t *testing.T, db satellite.DB) {
|
|
now := time.Now()
|
|
|
|
start := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, now.Location())
|
|
end := time.Date(now.Year(), now.Month()+1, 1, 0, 0, 0, 0, now.Location())
|
|
|
|
projectRecordsDB := db.StripeCoinPayments().ProjectRecords()
|
|
|
|
const limit = 5
|
|
const recordsLen = limit * 4
|
|
|
|
var createProjectRecords []stripecoinpayments.CreateProjectRecord
|
|
for i := 0; i < recordsLen; i++ {
|
|
projID, err := uuid.New()
|
|
require.NoError(t, err)
|
|
|
|
createProjectRecords = append(createProjectRecords,
|
|
stripecoinpayments.CreateProjectRecord{
|
|
ProjectID: projID,
|
|
Storage: float64(i) + 1,
|
|
Egress: int64(i) + 2,
|
|
Objects: float64(i) + 3,
|
|
},
|
|
)
|
|
}
|
|
|
|
err := projectRecordsDB.Create(ctx, createProjectRecords, start, end)
|
|
require.NoError(t, err)
|
|
|
|
page, err := projectRecordsDB.ListUnapplied(ctx, 0, limit, start, end)
|
|
require.NoError(t, err)
|
|
|
|
records := page.Records
|
|
|
|
for page.Next {
|
|
page, err = projectRecordsDB.ListUnapplied(ctx, page.NextOffset, limit, start, end)
|
|
require.NoError(t, err)
|
|
|
|
records = append(records, page.Records...)
|
|
}
|
|
|
|
require.Equal(t, recordsLen, len(records))
|
|
assert.False(t, page.Next)
|
|
assert.Equal(t, int64(0), page.NextOffset)
|
|
|
|
for _, record := range page.Records {
|
|
for _, createRecord := range createProjectRecords {
|
|
if record.ProjectID != createRecord.ProjectID {
|
|
continue
|
|
}
|
|
|
|
assert.NotNil(t, record.ID)
|
|
assert.Equal(t, 16, len(record.ID))
|
|
assert.Equal(t, createRecord.ProjectID, record.ProjectID)
|
|
assert.Equal(t, createRecord.Storage, record.Storage)
|
|
assert.Equal(t, createRecord.Egress, record.Egress)
|
|
assert.Equal(t, createRecord.Objects, record.Objects)
|
|
assert.True(t, start.Equal(record.PeriodStart))
|
|
assert.True(t, end.Equal(record.PeriodEnd))
|
|
}
|
|
}
|
|
})
|
|
}
|