connection_test.go

package gomaxcompute

import (
	"database/sql"
	"fmt"
	"math/rand"
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestQuery(t *testing.T) {
	a := assert.New(t)
	db, err := sql.Open("maxcompute", cfg4test.FormatDSN())
	a.NoError(err)

	queries := []string{
		`SELECT * FROM gomaxcompute_test LIMIT 1`,
		`SELECT * FROM gomaxcompute_test LIMIT 1;`,
	}
	for _, query := range queries {
		_, err = db.Query(query)
		a.NoError(err)
	}
}

func TestExec(t *testing.T) {
	a := assert.New(t)
	db, err := sql.Open("maxcompute", cfg4test.FormatDSN())
	a.NoError(err)

	tn := fmt.Sprintf("unitest%d", rand.Int())
	_, err = db.Exec(fmt.Sprintf("CREATE TABLE IF NOT EXISTS %s(shop_name STRING);", tn))
	a.NoError(err)
	_, err = db.Exec(fmt.Sprintf("DROP TABLE %s;", tn))
	a.NoError(err)
}

func TestQueryBase64(t *testing.T) {
	a := assert.New(t)
	db, err := sql.Open("maxcompute", cfg4test.FormatDSN())
	a.NoError(err)

	row, err := db.Query(`SELECT CAST("\001" AS string) AS a;`)
	a.NoError(err)
	for row.Next() {
		var s string
		row.Scan(&s)
		a.Equal("\001", s)
	}
}

func TestBadQuery(t *testing.T) {
	a := assert.New(t)
	db, err := sql.Open("maxcompute", cfg4test.FormatDSN())
	a.NoError(err)

	// Table not found
	tn := fmt.Sprintf("unitest%d", rand.Int())
	_, err = db.Query(fmt.Sprintf("SELECT * FROM %s;", tn))
	a.Error(err)
	a.True(strings.Contains(err.Error(), "Table not found"))
}

func TestBadExec(t *testing.T) {
	a := assert.New(t)
	db, err := sql.Open("maxcompute", cfg4test.FormatDSN())
	a.NoError(err)

	// Table not found
	tn := fmt.Sprintf("unitest%d", rand.Int())
	_, err = db.Exec(fmt.Sprintf("DROP TABLE %s;", tn))
	a.Error(err)
	a.True(strings.Contains(err.Error(), "Table not found"))
}

func TestInvalidSyntax(t *testing.T) {
	a := assert.New(t)
	db, e := sql.Open("maxcompute", cfg4test.FormatDSN())
	a.NoError(e)

	{
		_, err := db.Query(`SLEECT CAST("\001" AS string) AS a;`)
		a.NotNil(err)
		me, ok := err.(*MaxcomputeError)
		a.True(ok)
		a.Equal(`0130161`, me.Code)
		a.Equal(`ODPS-0130161:[1,1] Parse exception - invalid token 'SLEECT'`, me.Error())
	}

	{
		_, err := db.Exec(`DROP ;`)
		a.NotNil(err)
		me, ok := err.(*MaxcomputeError)
		a.True(ok)
		a.Equal(`0130161`, me.Code)
		a.Equal(`ODPS-0130161:[1,6] Parse exception - invalid token ';'`, me.Error())
	}

	{
		_, err := db.Query(`SELECT * FROM i_dont_exist;`)
		a.NotNil(err)
		me, ok := err.(*MaxcomputeError)
		a.True(ok)
		a.Equal(`0130131`, me.Code)
		a.Equal(`ODPS-0130131:[1,15] Table not found - table gomaxcompute_driver_w7u.i_dont_exist cannot be resolved`, me.Error())
	}
}