Function TestExecUpdateAndUpsert

exec/exec_test.go:335–441

(t *testing.T)

}

func TestExecUpdateAndUpsert(t *testing.T) {

	// By "Loading" table we force it to exist in this non DDL mock store
	mockcsv.LoadTable(mockcsv.SchemaName, "user_event3", "id,user_id,event,date\n1,abcabcabc,signup,\"2012-12-24T17:29:39.738Z\"")

	dbPre, err := schema.OpenConn("mockcsv", "user_event3")
	assert.True(t, err == nil, "%v", err)
	dbTablePre, ok := dbPre.(*mockcsv.Table)
	assert.True(t, ok, "Should be type MockCsvTable but was T:%T", dbTablePre)
	//u.Infof("db:  %#v", dbTable)
	assert.True(t, dbTablePre.Length() == 1, "Should have inserted and have 1 but was %v", dbTablePre.Length())

	sqlText := `
		UPSERT into user_event3 (id, user_id, event, date)
		VALUES
			("1234abcd", "9Ip1aKbeZe2njCDM", "logon", todate("2012/07/07"))
	`
	ctx := td.TestContext(sqlText)
	job, err := exec.BuildSqlJob(ctx)
	assert.True(t, err == nil, "%v", err)

	err = job.Setup()
	assert.True(t, err == nil)
	err = job.Run()
	assert.True(t, err == nil)

	db, err := schema.OpenConn("mockcsv", "user_event3")
	assert.True(t, err == nil, "%v", err)
	dbTable, ok := db.(*mockcsv.Table)
	assert.True(t, ok, "Should be type MockCsvTable ", dbTable)
	//u.Infof("db:  %#v", dbTable)
	assert.True(t, dbTable.Length() == 2, "Should have inserted and have 2 but was %v", dbTable.Length())

	// Now we are going to upsert the same row with changes
	sqlText = `
		UPSERT into user_event3 (id, user_id, event, date)
		VALUES
			("1234abcd", "9Ip1aKbeZe2njCDM", "logon", todate("2013/07/07"))
	`
	ctx = td.TestContext(sqlText)
	job, err = exec.BuildSqlJob(ctx)
	assert.True(t, err == nil, "%v", err)
	job.Setup()
	err = job.Run()
	assert.True(t, err == nil)

	// Should not have inserted, due to id being same
	assert.True(t, dbTable.Length() == 2, "Should have inserted and have 2 but was %v", dbTable.Length())

	// Now lets query it, we are going to use QLBridge Driver
	sqlSelect1 := `
		select id, user_id, event, date
	    FROM user_event3
	    WHERE user_id = "9Ip1aKbeZe2njCDM"
    `
	sqlDb, err := sql.Open("qlbridge", "mockcsv")
	assert.True(t, err == nil, "no error: %v", err)
	assert.True(t, sqlDb != nil, "has conn: %v", sqlDb)