Example #1
0
func TestInvalidParameter(t *testing.T) {
	db := NewDB()
	query, err := db.Select("name").From("users").Where("id = ?", "1 or where 1=1; select * from user;").Query()
	if query != nil {
		t.Fatal(err)
	}
	assert.Equal(t, err, errors.New("invalid parameter"))
}
Example #2
0
func TestSelect(t *testing.T) {
	db := NewDB()
	query, err := db.Select("name", "age", "email").From("users").Query()
	if err != nil {
		t.Fatal(err)
	}
	var actual string = query.String()
	var expected string = "SELECT name, age, email FROM users"
	assert.Equal(t, actual, expected)
}
Example #3
0
func TestJoin(t *testing.T) {
	db := NewDB()
	query, err := db.Select("name").From("users").Where("id = 1").Join("depts").On("users.dept_id = depts.id").OrderDesc().Query()
	if err != nil {
		t.Fatal(err)
	}
	var actual string = query.String()
	var expected string = "SELECT name FROM users WHERE id = 1 JOIN depts ON users.dept_id = depts.id ORDER BY DESC"
	assert.Equal(t, actual, expected)
}
Example #4
0
func TestWhere(t *testing.T) {
	db := NewDB()
	query, err := db.Select("name").From("users").Where("id = ? and age > ?", "1", "20").Query()
	if err != nil {
		t.Fatal(err)
	}
	var actual string = query.String()
	var expected string = "SELECT name FROM users WHERE id = 1 and age > 20"
	assert.Equal(t, actual, expected)
}