mirror of https://github.com/VinGarcia/ksql.git
Update TestInsert so its decoupled from the adapters
Vinícius Garcia
parent
eb030b1d76
commit
0b0e3a2901
598
ksql_test.go
598
ksql_test.go
|
|
@ -684,318 +684,332 @@ func QueryOneTest(
|
||||||
|
|
||||||
func TestInsert(t *testing.T) {
|
func TestInsert(t *testing.T) {
|
||||||
for _, config := range supportedConfigs {
|
for _, config := range supportedConfigs {
|
||||||
t.Run(config.driver, func(t *testing.T) {
|
InsertTest(t,
|
||||||
t.Run("success cases", func(t *testing.T) {
|
config,
|
||||||
t.Run("single primary key tables", func(t *testing.T) {
|
func(t *testing.T) (DBAdapter, io.Closer) {
|
||||||
err := createTables(config.driver)
|
db, close := connectDB(t, config)
|
||||||
if err != nil {
|
return db, close
|
||||||
t.Fatal("could not create test table!, reason:", err.Error())
|
},
|
||||||
}
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
t.Run("should insert one user correctly", func(t *testing.T) {
|
func InsertTest(
|
||||||
db, closer := connectDB(t, config)
|
t *testing.T,
|
||||||
defer closer.Close()
|
config testConfig,
|
||||||
|
newDBAdapter func(t *testing.T) (DBAdapter, io.Closer),
|
||||||
ctx := context.Background()
|
) {
|
||||||
c := newTestDB(db, config.driver)
|
t.Run(config.driver, func(t *testing.T) {
|
||||||
|
t.Run("success cases", func(t *testing.T) {
|
||||||
u := User{
|
t.Run("single primary key tables", func(t *testing.T) {
|
||||||
Name: "Fernanda",
|
|
||||||
Address: Address{
|
|
||||||
Country: "Brazil",
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
err := c.Insert(ctx, UsersTable, &u)
|
|
||||||
assert.Equal(t, nil, err)
|
|
||||||
assert.NotEqual(t, 0, u.ID)
|
|
||||||
|
|
||||||
result := User{}
|
|
||||||
err = getUserByID(c.db, c.dialect, &result, u.ID)
|
|
||||||
assert.Equal(t, nil, err)
|
|
||||||
|
|
||||||
assert.Equal(t, u.Name, result.Name)
|
|
||||||
assert.Equal(t, u.Address, result.Address)
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("should insert ignoring the ID for sqlite and multiple ids", func(t *testing.T) {
|
|
||||||
if supportedDialects[config.driver].InsertMethod() != insertWithLastInsertID {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// Using columns "id" and "name" as IDs:
|
|
||||||
table := NewTable("users", "id", "name")
|
|
||||||
|
|
||||||
db, closer := connectDB(t, config)
|
|
||||||
defer closer.Close()
|
|
||||||
|
|
||||||
ctx := context.Background()
|
|
||||||
c := newTestDB(db, config.driver)
|
|
||||||
|
|
||||||
u := User{
|
|
||||||
Name: "No ID returned",
|
|
||||||
Age: 3434, // Random number to avoid false positives on this test
|
|
||||||
|
|
||||||
Address: Address{
|
|
||||||
Country: "Brazil 3434",
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
err = c.Insert(ctx, table, &u)
|
|
||||||
assert.Equal(t, nil, err)
|
|
||||||
assert.Equal(t, uint(0), u.ID)
|
|
||||||
|
|
||||||
result := User{}
|
|
||||||
err = getUserByName(c.db, config.driver, &result, "No ID returned")
|
|
||||||
assert.Equal(t, nil, err)
|
|
||||||
|
|
||||||
assert.Equal(t, u.Age, result.Age)
|
|
||||||
assert.Equal(t, u.Address, result.Address)
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("should work with anonymous structs", func(t *testing.T) {
|
|
||||||
db, closer := connectDB(t, config)
|
|
||||||
defer closer.Close()
|
|
||||||
|
|
||||||
ctx := context.Background()
|
|
||||||
c := newTestDB(db, config.driver)
|
|
||||||
err = c.Insert(ctx, UsersTable, &struct {
|
|
||||||
ID int `ksql:"id"`
|
|
||||||
Name string `ksql:"name"`
|
|
||||||
Address map[string]interface{} `ksql:"address,json"`
|
|
||||||
}{Name: "fake-name", Address: map[string]interface{}{"city": "bar"}})
|
|
||||||
assert.Equal(t, nil, err)
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("should work with preset IDs", func(t *testing.T) {
|
|
||||||
db, closer := connectDB(t, config)
|
|
||||||
defer closer.Close()
|
|
||||||
|
|
||||||
ctx := context.Background()
|
|
||||||
c := newTestDB(db, config.driver)
|
|
||||||
|
|
||||||
usersByName := NewTable("users", "name")
|
|
||||||
|
|
||||||
err = c.Insert(ctx, usersByName, &struct {
|
|
||||||
Name string `ksql:"name"`
|
|
||||||
Age int `ksql:"age"`
|
|
||||||
}{Name: "Preset Name", Age: 5455})
|
|
||||||
assert.Equal(t, nil, err)
|
|
||||||
|
|
||||||
var inserted User
|
|
||||||
err := getUserByName(db, config.driver, &inserted, "Preset Name")
|
|
||||||
assert.Equal(t, nil, err)
|
|
||||||
assert.Equal(t, 5455, inserted.Age)
|
|
||||||
})
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("composite key tables", func(t *testing.T) {
|
|
||||||
err := createTables(config.driver)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal("could not create test table!, reason:", err.Error())
|
|
||||||
}
|
|
||||||
|
|
||||||
t.Run("should insert in composite key tables correctly", func(t *testing.T) {
|
|
||||||
db, closer := connectDB(t, config)
|
|
||||||
defer closer.Close()
|
|
||||||
|
|
||||||
ctx := context.Background()
|
|
||||||
c := newTestDB(db, config.driver)
|
|
||||||
|
|
||||||
table := NewTable("user_permissions", "id", "user_id", "perm_id")
|
|
||||||
err = c.Insert(ctx, table, &UserPermission{
|
|
||||||
UserID: 1,
|
|
||||||
PermID: 42,
|
|
||||||
})
|
|
||||||
tt.AssertNoErr(t, err)
|
|
||||||
|
|
||||||
userPerms, err := getUserPermissionsByUser(db, config.driver, 1)
|
|
||||||
tt.AssertNoErr(t, err)
|
|
||||||
tt.AssertEqual(t, len(userPerms), 1)
|
|
||||||
tt.AssertEqual(t, userPerms[0].UserID, 1)
|
|
||||||
tt.AssertEqual(t, userPerms[0].PermID, 42)
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("should accept partially provided values for composite key tables", func(t *testing.T) {
|
|
||||||
db, closer := connectDB(t, config)
|
|
||||||
defer closer.Close()
|
|
||||||
|
|
||||||
ctx := context.Background()
|
|
||||||
c := newTestDB(db, config.driver)
|
|
||||||
|
|
||||||
// Table defined with 3 values, but we'll provide only 2,
|
|
||||||
// the third will be generated for the purposes of this test:
|
|
||||||
table := NewTable("user_permissions", "id", "user_id", "perm_id")
|
|
||||||
permission := UserPermission{
|
|
||||||
UserID: 2,
|
|
||||||
PermID: 42,
|
|
||||||
}
|
|
||||||
err = c.Insert(ctx, table, &permission)
|
|
||||||
tt.AssertNoErr(t, err)
|
|
||||||
|
|
||||||
userPerms, err := getUserPermissionsByUser(db, config.driver, 2)
|
|
||||||
tt.AssertNoErr(t, err)
|
|
||||||
|
|
||||||
// Should retrieve the generated ID from the database,
|
|
||||||
// only if the database supports returning multiple values:
|
|
||||||
switch c.dialect.InsertMethod() {
|
|
||||||
case insertWithNoIDRetrieval, insertWithLastInsertID:
|
|
||||||
tt.AssertEqual(t, permission.ID, 0)
|
|
||||||
tt.AssertEqual(t, len(userPerms), 1)
|
|
||||||
tt.AssertEqual(t, userPerms[0].UserID, 2)
|
|
||||||
tt.AssertEqual(t, userPerms[0].PermID, 42)
|
|
||||||
case insertWithReturning, insertWithOutput:
|
|
||||||
tt.AssertNotEqual(t, permission.ID, 0)
|
|
||||||
tt.AssertEqual(t, len(userPerms), 1)
|
|
||||||
tt.AssertEqual(t, userPerms[0].ID, permission.ID)
|
|
||||||
tt.AssertEqual(t, userPerms[0].UserID, 2)
|
|
||||||
tt.AssertEqual(t, userPerms[0].PermID, 42)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
})
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("testing error cases", func(t *testing.T) {
|
|
||||||
err := createTables(config.driver)
|
err := createTables(config.driver)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal("could not create test table!, reason:", err.Error())
|
t.Fatal("could not create test table!, reason:", err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
t.Run("should report error for invalid input types", func(t *testing.T) {
|
t.Run("should insert one user correctly", func(t *testing.T) {
|
||||||
db, closer := connectDB(t, config)
|
db, closer := newDBAdapter(t)
|
||||||
defer closer.Close()
|
defer closer.Close()
|
||||||
|
|
||||||
ctx := context.Background()
|
ctx := context.Background()
|
||||||
c := newTestDB(db, config.driver)
|
c := newTestDB(db, config.driver)
|
||||||
|
|
||||||
err = c.Insert(ctx, UsersTable, "foo")
|
u := User{
|
||||||
assert.NotEqual(t, nil, err)
|
Name: "Fernanda",
|
||||||
|
Address: Address{
|
||||||
err = c.Insert(ctx, UsersTable, nullable.String("foo"))
|
Country: "Brazil",
|
||||||
assert.NotEqual(t, nil, err)
|
},
|
||||||
|
|
||||||
err = c.Insert(ctx, UsersTable, map[string]interface{}{
|
|
||||||
"name": "foo",
|
|
||||||
"age": 12,
|
|
||||||
})
|
|
||||||
assert.NotEqual(t, nil, err)
|
|
||||||
|
|
||||||
cantInsertSlice := []interface{}{
|
|
||||||
&User{Name: "foo", Age: 22},
|
|
||||||
&User{Name: "bar", Age: 32},
|
|
||||||
}
|
}
|
||||||
err = c.Insert(ctx, UsersTable, cantInsertSlice)
|
|
||||||
assert.NotEqual(t, nil, err)
|
|
||||||
|
|
||||||
// We might want to support this in the future, but not for now:
|
err := c.Insert(ctx, UsersTable, &u)
|
||||||
err = c.Insert(ctx, UsersTable, User{Name: "not a ptr to user", Age: 42})
|
assert.Equal(t, nil, err)
|
||||||
assert.NotEqual(t, nil, err)
|
assert.NotEqual(t, 0, u.ID)
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("should report error if for some reason the insertMethod is invalid", func(t *testing.T) {
|
result := User{}
|
||||||
db, closer := connectDB(t, config)
|
err = getUserByID(c.db, c.dialect, &result, u.ID)
|
||||||
defer closer.Close()
|
|
||||||
|
|
||||||
ctx := context.Background()
|
|
||||||
c := newTestDB(db, config.driver)
|
|
||||||
|
|
||||||
// This is an invalid value:
|
|
||||||
c.dialect = brokenDialect{}
|
|
||||||
|
|
||||||
err = c.Insert(ctx, UsersTable, &User{Name: "foo"})
|
|
||||||
assert.NotEqual(t, nil, err)
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("should report error if it receives a nil pointer to a struct", func(t *testing.T) {
|
|
||||||
db, closer := connectDB(t, config)
|
|
||||||
defer closer.Close()
|
|
||||||
|
|
||||||
ctx := context.Background()
|
|
||||||
c := newTestDB(db, config.driver)
|
|
||||||
|
|
||||||
var user *User
|
|
||||||
err := c.Insert(ctx, UsersTable, user)
|
|
||||||
assert.NotEqual(t, nil, err)
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("should report error if table contains an empty ID name", func(t *testing.T) {
|
|
||||||
db, closer := connectDB(t, config)
|
|
||||||
defer closer.Close()
|
|
||||||
|
|
||||||
ctx := context.Background()
|
|
||||||
c := newTestDB(db, config.driver)
|
|
||||||
|
|
||||||
err := c.Insert(ctx, NewTable("users", ""), &User{Name: "fake-name"})
|
|
||||||
tt.AssertErrContains(t, err, "ksql.Table", "ID", "empty string")
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("should report error if ksql.Table.name is empty", func(t *testing.T) {
|
|
||||||
db, closer := connectDB(t, config)
|
|
||||||
defer closer.Close()
|
|
||||||
|
|
||||||
ctx := context.Background()
|
|
||||||
c := newTestDB(db, config.driver)
|
|
||||||
|
|
||||||
err := c.Insert(ctx, NewTable("", "id"), &User{Name: "fake-name"})
|
|
||||||
tt.AssertErrContains(t, err, "ksql.Table", "table name", "empty string")
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("should not panic if a column doesn't exist in the database", func(t *testing.T) {
|
|
||||||
db, closer := connectDB(t, config)
|
|
||||||
defer closer.Close()
|
|
||||||
|
|
||||||
ctx := context.Background()
|
|
||||||
c := newTestDB(db, config.driver)
|
|
||||||
|
|
||||||
err = c.Insert(ctx, UsersTable, &struct {
|
|
||||||
ID string `ksql:"id"`
|
|
||||||
NonExistingColumn int `ksql:"non_existing"`
|
|
||||||
Name string `ksql:"name"`
|
|
||||||
}{NonExistingColumn: 42, Name: "fake-name"})
|
|
||||||
assert.NotEqual(t, nil, err)
|
|
||||||
msg := err.Error()
|
|
||||||
assert.Equal(t, true, strings.Contains(msg, "column"))
|
|
||||||
assert.Equal(t, true, strings.Contains(msg, "non_existing"))
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("should not panic if the ID column doesn't exist in the database", func(t *testing.T) {
|
|
||||||
db, closer := connectDB(t, config)
|
|
||||||
defer closer.Close()
|
|
||||||
|
|
||||||
ctx := context.Background()
|
|
||||||
c := newTestDB(db, config.driver)
|
|
||||||
|
|
||||||
brokenTable := NewTable("users", "non_existing_id")
|
|
||||||
|
|
||||||
_ = c.Insert(ctx, brokenTable, &struct {
|
|
||||||
ID string `ksql:"non_existing_id"`
|
|
||||||
Age int `ksql:"age"`
|
|
||||||
Name string `ksql:"name"`
|
|
||||||
}{Age: 42, Name: "fake-name"})
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("should not panic if the ID column is missing in the struct", func(t *testing.T) {
|
|
||||||
db, closer := connectDB(t, config)
|
|
||||||
defer closer.Close()
|
|
||||||
|
|
||||||
ctx := context.Background()
|
|
||||||
c := newTestDB(db, config.driver)
|
|
||||||
|
|
||||||
err = c.Insert(ctx, UsersTable, &struct {
|
|
||||||
Age int `ksql:"age"`
|
|
||||||
Name string `ksql:"name"`
|
|
||||||
}{Age: 42, Name: "Inserted With no ID"})
|
|
||||||
assert.Equal(t, nil, err)
|
assert.Equal(t, nil, err)
|
||||||
|
|
||||||
var u User
|
assert.Equal(t, u.Name, result.Name)
|
||||||
err = getUserByName(db, config.driver, &u, "Inserted With no ID")
|
assert.Equal(t, u.Address, result.Address)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("should insert ignoring the ID for sqlite and multiple ids", func(t *testing.T) {
|
||||||
|
if supportedDialects[config.driver].InsertMethod() != insertWithLastInsertID {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Using columns "id" and "name" as IDs:
|
||||||
|
table := NewTable("users", "id", "name")
|
||||||
|
|
||||||
|
db, closer := newDBAdapter(t)
|
||||||
|
defer closer.Close()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
c := newTestDB(db, config.driver)
|
||||||
|
|
||||||
|
u := User{
|
||||||
|
Name: "No ID returned",
|
||||||
|
Age: 3434, // Random number to avoid false positives on this test
|
||||||
|
|
||||||
|
Address: Address{
|
||||||
|
Country: "Brazil 3434",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
err = c.Insert(ctx, table, &u)
|
||||||
assert.Equal(t, nil, err)
|
assert.Equal(t, nil, err)
|
||||||
assert.NotEqual(t, uint(0), u.ID)
|
assert.Equal(t, uint(0), u.ID)
|
||||||
assert.Equal(t, 42, u.Age)
|
|
||||||
|
result := User{}
|
||||||
|
err = getUserByName(c.db, config.driver, &result, "No ID returned")
|
||||||
|
assert.Equal(t, nil, err)
|
||||||
|
|
||||||
|
assert.Equal(t, u.Age, result.Age)
|
||||||
|
assert.Equal(t, u.Address, result.Address)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("should work with anonymous structs", func(t *testing.T) {
|
||||||
|
db, closer := newDBAdapter(t)
|
||||||
|
defer closer.Close()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
c := newTestDB(db, config.driver)
|
||||||
|
err = c.Insert(ctx, UsersTable, &struct {
|
||||||
|
ID int `ksql:"id"`
|
||||||
|
Name string `ksql:"name"`
|
||||||
|
Address map[string]interface{} `ksql:"address,json"`
|
||||||
|
}{Name: "fake-name", Address: map[string]interface{}{"city": "bar"}})
|
||||||
|
assert.Equal(t, nil, err)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("should work with preset IDs", func(t *testing.T) {
|
||||||
|
db, closer := newDBAdapter(t)
|
||||||
|
defer closer.Close()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
c := newTestDB(db, config.driver)
|
||||||
|
|
||||||
|
usersByName := NewTable("users", "name")
|
||||||
|
|
||||||
|
err = c.Insert(ctx, usersByName, &struct {
|
||||||
|
Name string `ksql:"name"`
|
||||||
|
Age int `ksql:"age"`
|
||||||
|
}{Name: "Preset Name", Age: 5455})
|
||||||
|
assert.Equal(t, nil, err)
|
||||||
|
|
||||||
|
var inserted User
|
||||||
|
err := getUserByName(db, config.driver, &inserted, "Preset Name")
|
||||||
|
assert.Equal(t, nil, err)
|
||||||
|
assert.Equal(t, 5455, inserted.Age)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("composite key tables", func(t *testing.T) {
|
||||||
|
err := createTables(config.driver)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal("could not create test table!, reason:", err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
t.Run("should insert in composite key tables correctly", func(t *testing.T) {
|
||||||
|
db, closer := newDBAdapter(t)
|
||||||
|
defer closer.Close()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
c := newTestDB(db, config.driver)
|
||||||
|
|
||||||
|
table := NewTable("user_permissions", "id", "user_id", "perm_id")
|
||||||
|
err = c.Insert(ctx, table, &UserPermission{
|
||||||
|
UserID: 1,
|
||||||
|
PermID: 42,
|
||||||
|
})
|
||||||
|
tt.AssertNoErr(t, err)
|
||||||
|
|
||||||
|
userPerms, err := getUserPermissionsByUser(db, config.driver, 1)
|
||||||
|
tt.AssertNoErr(t, err)
|
||||||
|
tt.AssertEqual(t, len(userPerms), 1)
|
||||||
|
tt.AssertEqual(t, userPerms[0].UserID, 1)
|
||||||
|
tt.AssertEqual(t, userPerms[0].PermID, 42)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("should accept partially provided values for composite key tables", func(t *testing.T) {
|
||||||
|
db, closer := newDBAdapter(t)
|
||||||
|
defer closer.Close()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
c := newTestDB(db, config.driver)
|
||||||
|
|
||||||
|
// Table defined with 3 values, but we'll provide only 2,
|
||||||
|
// the third will be generated for the purposes of this test:
|
||||||
|
table := NewTable("user_permissions", "id", "user_id", "perm_id")
|
||||||
|
permission := UserPermission{
|
||||||
|
UserID: 2,
|
||||||
|
PermID: 42,
|
||||||
|
}
|
||||||
|
err = c.Insert(ctx, table, &permission)
|
||||||
|
tt.AssertNoErr(t, err)
|
||||||
|
|
||||||
|
userPerms, err := getUserPermissionsByUser(db, config.driver, 2)
|
||||||
|
tt.AssertNoErr(t, err)
|
||||||
|
|
||||||
|
// Should retrieve the generated ID from the database,
|
||||||
|
// only if the database supports returning multiple values:
|
||||||
|
switch c.dialect.InsertMethod() {
|
||||||
|
case insertWithNoIDRetrieval, insertWithLastInsertID:
|
||||||
|
tt.AssertEqual(t, permission.ID, 0)
|
||||||
|
tt.AssertEqual(t, len(userPerms), 1)
|
||||||
|
tt.AssertEqual(t, userPerms[0].UserID, 2)
|
||||||
|
tt.AssertEqual(t, userPerms[0].PermID, 42)
|
||||||
|
case insertWithReturning, insertWithOutput:
|
||||||
|
tt.AssertNotEqual(t, permission.ID, 0)
|
||||||
|
tt.AssertEqual(t, len(userPerms), 1)
|
||||||
|
tt.AssertEqual(t, userPerms[0].ID, permission.ID)
|
||||||
|
tt.AssertEqual(t, userPerms[0].UserID, 2)
|
||||||
|
tt.AssertEqual(t, userPerms[0].PermID, 42)
|
||||||
|
}
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
}
|
|
||||||
|
t.Run("testing error cases", func(t *testing.T) {
|
||||||
|
err := createTables(config.driver)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal("could not create test table!, reason:", err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
t.Run("should report error for invalid input types", func(t *testing.T) {
|
||||||
|
db, closer := newDBAdapter(t)
|
||||||
|
defer closer.Close()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
c := newTestDB(db, config.driver)
|
||||||
|
|
||||||
|
err = c.Insert(ctx, UsersTable, "foo")
|
||||||
|
assert.NotEqual(t, nil, err)
|
||||||
|
|
||||||
|
err = c.Insert(ctx, UsersTable, nullable.String("foo"))
|
||||||
|
assert.NotEqual(t, nil, err)
|
||||||
|
|
||||||
|
err = c.Insert(ctx, UsersTable, map[string]interface{}{
|
||||||
|
"name": "foo",
|
||||||
|
"age": 12,
|
||||||
|
})
|
||||||
|
assert.NotEqual(t, nil, err)
|
||||||
|
|
||||||
|
cantInsertSlice := []interface{}{
|
||||||
|
&User{Name: "foo", Age: 22},
|
||||||
|
&User{Name: "bar", Age: 32},
|
||||||
|
}
|
||||||
|
err = c.Insert(ctx, UsersTable, cantInsertSlice)
|
||||||
|
assert.NotEqual(t, nil, err)
|
||||||
|
|
||||||
|
// We might want to support this in the future, but not for now:
|
||||||
|
err = c.Insert(ctx, UsersTable, User{Name: "not a ptr to user", Age: 42})
|
||||||
|
assert.NotEqual(t, nil, err)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("should report error if for some reason the insertMethod is invalid", func(t *testing.T) {
|
||||||
|
db, closer := newDBAdapter(t)
|
||||||
|
defer closer.Close()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
c := newTestDB(db, config.driver)
|
||||||
|
|
||||||
|
// This is an invalid value:
|
||||||
|
c.dialect = brokenDialect{}
|
||||||
|
|
||||||
|
err = c.Insert(ctx, UsersTable, &User{Name: "foo"})
|
||||||
|
assert.NotEqual(t, nil, err)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("should report error if it receives a nil pointer to a struct", func(t *testing.T) {
|
||||||
|
db, closer := newDBAdapter(t)
|
||||||
|
defer closer.Close()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
c := newTestDB(db, config.driver)
|
||||||
|
|
||||||
|
var user *User
|
||||||
|
err := c.Insert(ctx, UsersTable, user)
|
||||||
|
assert.NotEqual(t, nil, err)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("should report error if table contains an empty ID name", func(t *testing.T) {
|
||||||
|
db, closer := newDBAdapter(t)
|
||||||
|
defer closer.Close()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
c := newTestDB(db, config.driver)
|
||||||
|
|
||||||
|
err := c.Insert(ctx, NewTable("users", ""), &User{Name: "fake-name"})
|
||||||
|
tt.AssertErrContains(t, err, "ksql.Table", "ID", "empty string")
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("should report error if ksql.Table.name is empty", func(t *testing.T) {
|
||||||
|
db, closer := newDBAdapter(t)
|
||||||
|
defer closer.Close()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
c := newTestDB(db, config.driver)
|
||||||
|
|
||||||
|
err := c.Insert(ctx, NewTable("", "id"), &User{Name: "fake-name"})
|
||||||
|
tt.AssertErrContains(t, err, "ksql.Table", "table name", "empty string")
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("should not panic if a column doesn't exist in the database", func(t *testing.T) {
|
||||||
|
db, closer := newDBAdapter(t)
|
||||||
|
defer closer.Close()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
c := newTestDB(db, config.driver)
|
||||||
|
|
||||||
|
err = c.Insert(ctx, UsersTable, &struct {
|
||||||
|
ID string `ksql:"id"`
|
||||||
|
NonExistingColumn int `ksql:"non_existing"`
|
||||||
|
Name string `ksql:"name"`
|
||||||
|
}{NonExistingColumn: 42, Name: "fake-name"})
|
||||||
|
assert.NotEqual(t, nil, err)
|
||||||
|
msg := err.Error()
|
||||||
|
assert.Equal(t, true, strings.Contains(msg, "column"))
|
||||||
|
assert.Equal(t, true, strings.Contains(msg, "non_existing"))
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("should not panic if the ID column doesn't exist in the database", func(t *testing.T) {
|
||||||
|
db, closer := newDBAdapter(t)
|
||||||
|
defer closer.Close()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
c := newTestDB(db, config.driver)
|
||||||
|
|
||||||
|
brokenTable := NewTable("users", "non_existing_id")
|
||||||
|
|
||||||
|
_ = c.Insert(ctx, brokenTable, &struct {
|
||||||
|
ID string `ksql:"non_existing_id"`
|
||||||
|
Age int `ksql:"age"`
|
||||||
|
Name string `ksql:"name"`
|
||||||
|
}{Age: 42, Name: "fake-name"})
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("should not panic if the ID column is missing in the struct", func(t *testing.T) {
|
||||||
|
db, closer := newDBAdapter(t)
|
||||||
|
defer closer.Close()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
c := newTestDB(db, config.driver)
|
||||||
|
|
||||||
|
err = c.Insert(ctx, UsersTable, &struct {
|
||||||
|
Age int `ksql:"age"`
|
||||||
|
Name string `ksql:"name"`
|
||||||
|
}{Age: 42, Name: "Inserted With no ID"})
|
||||||
|
assert.Equal(t, nil, err)
|
||||||
|
|
||||||
|
var u User
|
||||||
|
err = getUserByName(db, config.driver, &u, "Inserted With no ID")
|
||||||
|
assert.Equal(t, nil, err)
|
||||||
|
assert.NotEqual(t, uint(0), u.ID)
|
||||||
|
assert.Equal(t, 42, u.Age)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
type brokenDialect struct{}
|
type brokenDialect struct{}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue