idb_test.go

package gha2db

import (
	"encoding/json"
	"testing"
	"time"

	lib "gha2db"
)

func TestInfluxDB(t *testing.T) {
	// Environment context parse
	var ctx lib.Ctx
	ctx.Init()

	// Do not allow to run tests in "gha" database
	if ctx.IDBDB == "gha" {
		t.Errorf("tests cannot be run on \"gha\" database")
		return
	}

	// Connect to InfluxDB
	con := lib.IDBConn(&ctx)

	// Drop & create DB, ignore errors (we start with fresh DB)
	// On fatal errors, lib.QueryIDB calls os.Exit, so test will fail too
	lib.QueryIDB(con, &ctx, "drop database "+ctx.IDBDB)
	lib.QueryIDB(con, &ctx, "create database "+ctx.IDBDB)

	// Drop database and close connection at the end
	defer func() {
		// Drop database at the end of test
		lib.QueryIDB(con, &ctx, "drop database "+ctx.IDBDB)

		// Close IDB connection
		con.Close()
	}()

	// Get BatchPoints
	bp := lib.IDBBatchPoints(&ctx, &con)

	// Add batch points
	hourAgo := lib.HourStart(time.Now().Add(-time.Hour))
	hourFromNow := lib.HourStart(time.Now().Add(time.Hour))
	pt := lib.IDBNewPointWithErr("test", nil, map[string]interface{}{"value": 2}, hourFromNow)
	bp.AddPoint(pt)
	pt = lib.IDBNewPointWithErr("test", nil, map[string]interface{}{"value": 1}, hourAgo)
	bp.AddPoint(pt)

	// Write the batch
	err := con.Write(bp)
	if err != nil {
		t.Errorf(err.Error())
	}

	// Get newest value
	res := lib.QueryIDB(con, &ctx, "select last(value) from test")
	series := res[0].Series
	if len(series) != 1 {
		t.Errorf("expected exactly one row of data, got %+v", series)
	}
	row := series[0]
	if row.Name != "test" {
		t.Errorf("expected last series name \"test\", got %+v", row.Name)
	}
	dt := lib.TimeParseIDB(row.Values[0][0].(string))
	value := row.Values[0][1].(json.Number)
	if dt != hourFromNow {
		t.Errorf("expected last series date %v, got %v", hourFromNow, dt)
	}
	expected := json.Number("2")
	if value != expected {
		t.Errorf("expected last series value %v, got %v", expected, value)
	}

	// Get oldest value
	res = lib.QueryIDB(con, &ctx, "select first(value) from test")
	series = res[0].Series
	if len(series) != 1 {
		t.Errorf("expected exactly one row of data, got %+v", series)
	}
	row = series[0]
	if row.Name != "test" {
		t.Errorf("expected first series name \"test\", got %+v", row.Name)
	}
	dt = lib.TimeParseIDB(row.Values[0][0].(string))
	value = row.Values[0][1].(json.Number)
	if dt != hourAgo {
		t.Errorf("expected first series date %v, got %v", hourAgo, dt)
	}
	expected = json.Number("1")
	if value != expected {
		t.Errorf("expected first series value %v, got %v", expected, value)
	}
}