odbtest/test/unit/mql/batch/batch_insert.go

package batch

import (
	"fmt"
	"testing"
	"time"

	//"fmt"
	. "git.wecise.com/wecise/odbserver/odb"
	"git.wecise.com/wecise/odbserver/odb/test"
	"gitee.com/wecisecode/util/logger"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

type BatchInsertTests struct {
	Test *testing.T
	g    *Gutil
}

func BatchInsertTest(t *testing.T) {

	g := test.TestG()

	t.Run("InitG", func(t *testing.T) {
		test := &BatchInsertTests{Test: t, g: g}
		test.InitG()
	})

	t.Run("Class", func(t *testing.T) {
		test := &BatchInsertTests{Test: t, g: g}
		test.Class()
	})

	t.Run("Data", func(t *testing.T) {
		test := &BatchInsertTests{Test: t, g: g}
		test.Data()
	})

	time.Sleep(time.Duration(2) * time.Second)

	t.Run("Valid", func(t *testing.T) {
		test := &BatchInsertTests{Test: t, g: g}
		test.Valid()
	})
}

func (t *BatchInsertTests) InitG() {
	require.NotNil(t.Test, t.g)
}

func (t *BatchInsertTests) Class() {

	_, _, err := t.g.Query(`
	create class if not exists /test/batch (
		
		v_varchar		varchar,
		v_text			text,
		v_int			int,
		v_bigint		bigint,
		v_smalldouble	double,
		v_double		double,
		v_float			float,
		v_bool			bool,
		
		indexes(v_varchar, v_text, v_int, v_bigint, v_double, v_float, v_smalldouble, v_bool),
		keys(v_varchar , v_text)
	) with ckeys=v_bool`)

	if err != nil {
		logger.Errorf("%v", err)
	}

	require.Nil(t.Test, err, fmt.Sprint(err))
}

func (t *BatchInsertTests) Data() {
	var err error
	_, _, err = t.g.Query(`
	
	BEGIN BATCH
	
	insert into /test/batch (v_varchar, v_text, v_int, v_bigint, v_float, v_double, v_smalldouble, v_bool) values ('4006', 'hello word 2003', 4006, 4006, 1.1, 1.1234567890123456, 1.123456, true  );
	
	insert into /test/batch (v_varchar, v_text, v_int, v_bigint, v_float, v_double, v_smalldouble, v_bool) values ('40', 'hello word 2003', 40, 40, 1.1, 1.1234567890123456, 1.123456, false  );
	
	insert into /test/batch (v_varchar, v_text, v_int, v_bigint, v_float, v_double, v_smalldouble, v_bool) values ('400', 'hello word 2003', 400, 400, 1.1, 1.1234567890123456, 1.123456, true  );
	
	insert into /test/batch (v_varchar, v_text, v_int, v_bigint, v_float, v_double, v_smalldouble, v_bool) values ('74006', 'hello word 2003', -4006, 4006, 1.1, 1.1234567890123456, 1.123456, false  );
	
	insert into /test/batch (v_varchar, v_text, v_int, v_bigint, v_float, v_double, v_smalldouble, v_bool) values ('409', 'hello word 2003', -40, 40, 1.1, 1.1234567890123456, 1.123456, true  );
	
	insert into /test/batch (v_varchar, v_text, v_int, v_bigint, v_float, v_double, v_smalldouble, v_bool) values ('500', 'hello word 2003', -400, 400, 1.1, 1.1234567890123456, 1.123456, false  );
	
	insert into /test/batch (v_varchar, v_text, v_int, v_bigint, v_float, v_double, v_smalldouble, v_bool) values ('600', 'hello word 2003', 370, 370, 1.1, 1.1234567890123456, 1.123456, true  );
	
	insert into /test/batch (v_varchar, v_text, v_int, v_bigint, v_float, v_double, v_smalldouble, v_bool) values ('700', 'hello word 2003', 580, 580, 1.1, 1.1234567890123456, 1.123456, true  );
	
	insert into /test/batch (v_varchar, v_text, v_int, v_bigint, v_float, v_double, v_smalldouble, v_bool) values ('602', 'hello word 2003', -370, -370, 1.1, 1.1234567890123456, 1.123456, true  );
	
	insert into /test/batch (v_varchar, v_text, v_int, v_bigint, v_float, v_double, v_smalldouble, v_bool) values ('708', 'hello word 2003', -580, -580, 1.1, 1.1234567890123456, 1.123456, true  );
	
	END
	
	`)

	if err != nil {
		logger.Errorf("%v", err)
	}
	require.Nil(t.Test, err, fmt.Sprint(err))
}

func (t *BatchInsertTests) Valid() {

	rtn, _, err := t.g.Query(`select * from /test/batch`)

	if err != nil {
		logger.Errorf("%v", err)
	}

	if assert.Nil(t.Test, err) {
		if assert.NotNil(t.Test, rtn) {
			assert.Equal(t.Test, 10, len(rtn), "they should be equal")
		}
	}
}