Function TestExecGroupBy

exec/exec_test.go:140–202

(t *testing.T)

}

func TestExecGroupBy(t *testing.T) {

	sqlText := `
		select 
	        user_id, count(user_id), avg(price)
	    FROM orders
	    GROUP BY user_id
	`
	ctx := td.TestContext(sqlText)
	job, err := exec.BuildSqlJob(ctx)
	assert.True(t, err == nil, "no error %v", err)

	msgs := make([]schema.Message, 0)
	resultWriter := exec.NewResultBuffer(ctx, &msgs)
	job.RootTask.Add(resultWriter)

	err = job.Setup()
	assert.True(t, err == nil)
	err = job.Run()
	time.Sleep(time.Millisecond * 10)
	assert.True(t, err == nil, "no error %v", err)
	assert.True(t, len(msgs) == 2, "should have grouped orders into 2 users %v", len(msgs))
	var row []driver.Value
	for _, msg := range msgs {
		r := msg.(*datasource.SqlDriverMessageMap).Values()
		if r[0].(string) == "9Ip1aKbeZe2njCDM" {
			row = r
		}
	}

	u.Debugf("row: %#v", row)
	assert.True(t, len(row) == 3, "expects 3 cols but got %v", len(row))
	assert.True(t, row[0] == "9Ip1aKbeZe2njCDM", "%#v", row)
	// I really don't like this float behavior?
	assert.True(t, int(row[1].(int64)) == 2, "expected 2 orders for %v", row)
	assert.True(t, int(row[2].(float64)) == 30, "expected avg=30 for price %v", row)

	sqlText = `
		select 
	        avg(len(email))
	    FROM users
	    GROUP BY "-"
	`
	ctx = td.TestContext(sqlText)
	job, err = exec.BuildSqlJob(ctx)
	assert.True(t, err == nil, "no error %v", err)

	msgs = make([]schema.Message, 0)
	resultWriter = exec.NewResultBuffer(ctx, &msgs)
	job.RootTask.Add(resultWriter)

	err = job.Setup()
	assert.True(t, err == nil)
	err = job.Run()
	time.Sleep(time.Millisecond * 10)
	assert.True(t, err == nil, "no error %v", err)
	assert.True(t, len(msgs) == 1, "should have grouped orders into 1 record %v", len(msgs))
	u.Debugf("msg: %#v", msgs[0])