// 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
}