Function TestParseUpdateCommand

parser/parser_test.go:711–752

(t *testing.T)

}

func TestParseUpdateCommand(t *testing.T) {
	tests := []struct {
		input             string
		expectedTableName string
		expectedChanges   map[token.Token]ast.Anonymitifier
	}{
		{
			input: "UPDATE tbl SET colName TO 5;", expectedTableName: "tbl", expectedChanges: map[token.Token]ast.Anonymitifier{
				{Type: token.IDENT, Literal: "colName"}: {Token: token.Token{Type: token.LITERAL, Literal: "5"}},
			},
		},
		{
			input: "UPDATE tbl1 SET colName1 TO 'hi hello', colName2 TO 5;", expectedTableName: "tbl1", expectedChanges: map[token.Token]ast.Anonymitifier{
				{Type: token.IDENT, Literal: "colName1"}: {Token: token.Token{Type: token.IDENT, Literal: "hi hello"}},
				{Type: token.IDENT, Literal: "colName2"}: {Token: token.Token{Type: token.LITERAL, Literal: "5"}},
			},
		},
		{
			input: "UPDATE tbl1 SET colName1 TO NULL, colName2 TO 'NULL';", expectedTableName: "tbl1", expectedChanges: map[token.Token]ast.Anonymitifier{
				{Type: token.IDENT, Literal: "colName1"}: {Token: token.Token{Type: token.NULL, Literal: "NULL"}},
				{Type: token.IDENT, Literal: "colName2"}: {Token: token.Token{Type: token.LITERAL, Literal: "NULL"}},
			},
		},
	}

	for testIndex, tt := range tests {
		lexer := lexer.RunLexer(tt.input)
		parserInstance := New(lexer)
		sequences, err := parserInstance.ParseSequence()
		if err != nil {
			t.Fatalf("[%d] Got error from parser: %s", testIndex, err)
		}

		if len(sequences.Commands) != 1 {
			t.Fatalf("[%d] sequences does not contain 1 statements. got=%d", testIndex, len(sequences.Commands))
		}

		if !testUpdateStatement(t, sequences.Commands[0], tt.expectedTableName, tt.expectedChanges) {
			return
		}
	}
}

func TestParseUpdateCommandWithWhere(t *testing.T) {
	tests := []struct {