135 lines
4.3 KiB
Go
135 lines
4.3 KiB
Go
package transactions
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestQueries_createRangeQuery(t *testing.T) {
|
|
var tests = []struct {
|
|
tm time.Time
|
|
ts string
|
|
wantStartLastVaa, wantStartAggregatesVaa string
|
|
}{
|
|
{
|
|
//2023-05-04T12:25:48.112233445Z
|
|
tm: time.Date(2023, 5, 4, 12, 25, 48, 112233445, time.UTC),
|
|
ts: "1d",
|
|
wantStartLastVaa: "2023-05-04T12:00:00Z",
|
|
wantStartAggregatesVaa: "2023-05-03T12:00:00Z",
|
|
},
|
|
{
|
|
//2023-05-04T20:59:17.992233445Z
|
|
tm: time.Date(2023, 5, 4, 20, 59, 17, 992233445, time.UTC),
|
|
ts: "1w",
|
|
wantStartLastVaa: "2023-05-04T00:00:00Z",
|
|
wantStartAggregatesVaa: "2023-04-27T00:00:00Z",
|
|
},
|
|
{
|
|
//2023-05-04T17:09:33.987654321Z
|
|
tm: time.Date(2023, 5, 4, 17, 9, 33, 987654321, time.UTC),
|
|
ts: "1mo",
|
|
wantStartLastVaa: "2023-05-04T00:00:00Z",
|
|
wantStartAggregatesVaa: "2023-04-04T00:00:00Z",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
startLastVaa, startAggregatesVaa := createRangeQuery(tt.tm, tt.ts)
|
|
assert.Equal(t, tt.wantStartLastVaa, startLastVaa)
|
|
assert.Equal(t, tt.wantStartAggregatesVaa, startAggregatesVaa)
|
|
}
|
|
}
|
|
|
|
func TestQueries_buildLastTrxQuery1d1h(t *testing.T) {
|
|
|
|
expected := `
|
|
lastVaaCount = from(bucket: "wormscan-1month")
|
|
|> range(start: 2023-05-04T18:00:00Z)
|
|
|> filter(fn: (r) => r["_measurement"] == "vaa_count")
|
|
|> group()
|
|
|> aggregateWindow(every: 1h, fn: count, createEmpty: true)
|
|
aggregatesVaaCount = from(bucket: "wormscan")
|
|
|> range(start: 2023-05-03T18:00:00Z)
|
|
|> filter(fn: (r) => r["_measurement"] == "vaa_count_1h")
|
|
union(tables: [aggregatesVaaCount, lastVaaCount])
|
|
|> group()
|
|
|> aggregateWindow(every: 1h, fn: sum, createEmpty: true)
|
|
|> sort(columns: ["_time"], desc: true)
|
|
`
|
|
//2023-05-04T18:39:10.985Z
|
|
tm := time.Date(2023, 5, 4, 18, 39, 10, 985, time.UTC)
|
|
actual := buildLastTrxQuery("wormscan-1month", "wormscan", tm, &TransactionCountQuery{TimeSpan: "1d", SampleRate: "1h"})
|
|
assert.Equal(t, expected, actual)
|
|
}
|
|
|
|
func TestQueries_buildLastTrxQuery1w1d(t *testing.T) {
|
|
|
|
expected := `
|
|
lastVaaCount = from(bucket: "wormscan-1month")
|
|
|> range(start: 2023-05-04T00:00:00Z)
|
|
|> filter(fn: (r) => r["_measurement"] == "vaa_count")
|
|
|> group()
|
|
aggregatesVaaCount = from(bucket: "wormscan")
|
|
|> range(start: 2023-04-27T00:00:00Z)
|
|
|> filter(fn: (r) => r["_measurement"] == "vaa_count_1h")
|
|
|> aggregateWindow(every: 1h, fn: sum, createEmpty: true)
|
|
union(tables: [aggregatesVaaCount, lastVaaCount])
|
|
|> group()
|
|
|> aggregateWindow(every: 1d, fn: sum, createEmpty: true)
|
|
|> sort(columns: ["_time"], desc: true)
|
|
`
|
|
//2023-05-04T18:39:10.985Z
|
|
tm := time.Date(2023, 5, 4, 18, 39, 10, 985, time.UTC)
|
|
actual := buildLastTrxQuery("wormscan-1month", "wormscan", tm, &TransactionCountQuery{TimeSpan: "1w", SampleRate: "1d"})
|
|
assert.Equal(t, expected, actual)
|
|
}
|
|
|
|
func TestQueries_buildTotalTrxCountQuery(t *testing.T) {
|
|
|
|
expected := `
|
|
current = from(bucket: "bucket-forever")
|
|
|> range(start: 2023-05-12T00:00:00Z)
|
|
|> filter(fn: (r) => r["_measurement"] == "vaa_volume")
|
|
|> filter(fn: (r) => r["_field"] == "volume")
|
|
|> group()
|
|
|> count()
|
|
last = from(bucket: "bucket-30days")
|
|
|> range(start: -1mo)
|
|
|> filter(fn: (r) => r["_measurement"] == "total_tx_count")
|
|
|> last()
|
|
union(tables: [current, last])
|
|
|> group()
|
|
|> sum()
|
|
`
|
|
//2023-05-04T18:39:10.985Z
|
|
tm := time.Date(2023, 5, 12, 16, 53, 10, 985, time.UTC)
|
|
actual := buildTotalTrxCountQuery("bucket-forever", "bucket-30days", tm)
|
|
assert.Equal(t, expected, actual)
|
|
}
|
|
|
|
func TestQueries_buildTotalTrxVolumeQuery(t *testing.T) {
|
|
|
|
expected := `
|
|
current = from(bucket: "bucket-forever")
|
|
|> range(start: 2023-05-10T00:00:00Z)
|
|
|> filter(fn: (r) => r["_measurement"] == "vaa_volume")
|
|
|> filter(fn: (r) => r["_field"] == "volume")
|
|
|> group()
|
|
|> sum()
|
|
last = from(bucket: "bucket-30days")
|
|
|> range(start: -1mo)
|
|
|> filter(fn: (r) => r["_measurement"] == "total_tx_volume")
|
|
|> last()
|
|
union(tables: [current, last])
|
|
|> group()
|
|
|> sum()
|
|
`
|
|
//2023-05-04T18:39:10.985Z
|
|
tm := time.Date(2023, 5, 10, 16, 53, 10, 985, time.UTC)
|
|
actual := buildTotalTrxVolumeQuery("bucket-forever", "bucket-30days", tm)
|
|
assert.Equal(t, expected, actual)
|
|
}
|