82 lines
2.0 KiB
Go
82 lines
2.0 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package rollup
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"go.uber.org/zap"
|
|
|
|
"storj.io/storj/internal/testcontext"
|
|
"storj.io/storj/internal/teststorj"
|
|
"storj.io/storj/pkg/storj"
|
|
"storj.io/storj/satellite"
|
|
"storj.io/storj/satellite/satellitedb"
|
|
)
|
|
|
|
func TestQueryOneDay(t *testing.T) {
|
|
ctx, r, _, nodeData, cleanup := createRollup(t)
|
|
defer cleanup()
|
|
|
|
now := time.Now().UTC()
|
|
later := now.Add(time.Hour * 24)
|
|
|
|
err := r.db.SaveAtRestRaw(ctx, now, true, nodeData)
|
|
assert.NoError(t, err)
|
|
|
|
// test should return error because we delete latest day's rollup
|
|
err = r.Query(ctx)
|
|
assert.NoError(t, err)
|
|
|
|
rows, err := r.db.QueryPaymentInfo(ctx, now, later)
|
|
assert.Equal(t, 0, len(rows))
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestQueryTwoDays(t *testing.T) {
|
|
ctx, r, _, nodeData, cleanup := createRollup(t)
|
|
defer cleanup()
|
|
|
|
now := time.Now().UTC()
|
|
then := now.Add(time.Hour * -24)
|
|
|
|
err := r.db.SaveAtRestRaw(ctx, now, true, nodeData)
|
|
assert.NoError(t, err)
|
|
|
|
// db.db.Exec("UPDATE accounting_raws SET created_at= WHERE ")
|
|
// err = r.Query(ctx)
|
|
// assert.NoError(t, err)
|
|
|
|
_, err = r.db.QueryPaymentInfo(ctx, then, now)
|
|
//assert.Equal(t, 10, len(rows))
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func createRollup(t *testing.T) (*testcontext.Context, *rollup, satellite.DB, map[storj.NodeID]float64, func()) {
|
|
ctx := testcontext.New(t)
|
|
db, err := satellitedb.NewInMemory()
|
|
assert.NoError(t, err)
|
|
|
|
assert.NoError(t, db.CreateTables())
|
|
cleanup := func() {
|
|
defer ctx.Cleanup()
|
|
defer ctx.Check(db.Close)
|
|
}
|
|
statdb := db.StatDB()
|
|
// generate nodeData
|
|
nodeData := make(map[storj.NodeID]float64)
|
|
for i := 1; i <= 10; i++ {
|
|
id := teststorj.NodeIDFromString(string(i))
|
|
nodeData[id] = float64(i * 100)
|
|
_, err := statdb.Create(ctx, id, nil)
|
|
assert.NoError(t, err)
|
|
_, err = statdb.Get(ctx, id)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
return ctx, newRollup(zap.NewNop(), db.Accounting(), time.Second), db, nodeData, cleanup
|
|
}
|