GOMODULES/ksql
1
0
Fork 0
mirror of https://github.com/VinGarcia/ksql.git synced 2025-04-28 05:32:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update TestQueryChunks to include the new json feature

Browse Source
This commit is contained in:
Vinícius Garcia 2021-03-02 22:49:20 -03:00
parent 8c425e93d5
commit 665bd35a85
1 changed files with 19 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
kiss_orm_test.go
View File

@ -805,12 +805,15 @@ func TestQueryChunks(t *testing.T) {
ctx := context.Background() ctx := context.Background()
c := newTestDB(db, driver, "users") c := newTestDB(db, driver, "users")
_ = c.Insert(ctx, &User{Name: "User1"}) _ = c.Insert(ctx, &User{
Name: "User1",
Address: Address{Country: "BR"},
})
var length int var length int
var u User var u User
err = c.QueryChunks(ctx, ChunkParser{ err = c.QueryChunks(ctx, ChunkParser{
Query: `select * from users where name = ` + c.dialect.Placeholder(0), Query: `SELECT * FROM users WHERE name = ` + c.dialect.Placeholder(0),
Params: []interface{}{"User1"}, Params: []interface{}{"User1"},
ChunkSize: 100, ChunkSize: 100,
@ -827,6 +830,7 @@ func TestQueryChunks(t *testing.T) {
assert.Equal(t, 1, length) assert.Equal(t, 1, length)
assert.NotEqual(t, uint(0), u.ID) assert.NotEqual(t, uint(0), u.ID)
assert.Equal(t, "User1", u.Name) assert.Equal(t, "User1", u.Name)
assert.Equal(t, "BR", u.Address.Country)
}) })
t.Run("should query one chunk correctly", func(t *testing.T) { t.Run("should query one chunk correctly", func(t *testing.T) {
@ -841,8 +845,8 @@ func TestQueryChunks(t *testing.T) {
ctx := context.Background() ctx := context.Background()
c := newTestDB(db, driver, "users") c := newTestDB(db, driver, "users")
_ = c.Insert(ctx, &User{Name: "User1"}) _ = c.Insert(ctx, &User{Name: "User1", Address: Address{Country: "US"}})
_ = c.Insert(ctx, &User{Name: "User2"}) _ = c.Insert(ctx, &User{Name: "User2", Address: Address{Country: "BR"}})
var lengths []int var lengths []int
var users []User var users []User
@ -861,10 +865,14 @@ func TestQueryChunks(t *testing.T) {
assert.Equal(t, nil, err) assert.Equal(t, nil, err)
assert.Equal(t, 1, len(lengths)) assert.Equal(t, 1, len(lengths))
assert.Equal(t, 2, lengths[0]) assert.Equal(t, 2, lengths[0])
assert.NotEqual(t, uint(0), users[0].ID) assert.NotEqual(t, uint(0), users[0].ID)
assert.Equal(t, "User1", users[0].Name) assert.Equal(t, "User1", users[0].Name)
assert.Equal(t, "US", users[0].Address.Country)
assert.NotEqual(t, uint(0), users[1].ID) assert.NotEqual(t, uint(0), users[1].ID)
assert.Equal(t, "User2", users[1].Name) assert.Equal(t, "User2", users[1].Name)
assert.Equal(t, "BR", users[1].Address.Country)
}) })
t.Run("should query chunks of 1 correctly", func(t *testing.T) { t.Run("should query chunks of 1 correctly", func(t *testing.T) {
@ -879,8 +887,8 @@ func TestQueryChunks(t *testing.T) {
ctx := context.Background() ctx := context.Background()
c := newTestDB(db, driver, "users") c := newTestDB(db, driver, "users")
_ = c.Insert(ctx, &User{Name: "User1"}) _ = c.Insert(ctx, &User{Name: "User1", Address: Address{Country: "US"}})
_ = c.Insert(ctx, &User{Name: "User2"}) _ = c.Insert(ctx, &User{Name: "User2", Address: Address{Country: "BR"}})
var lengths []int var lengths []int
var users []User var users []User
@ -898,11 +906,15 @@ func TestQueryChunks(t *testing.T) {
assert.Equal(t, nil, err) assert.Equal(t, nil, err)
assert.Equal(t, 2, len(users)) assert.Equal(t, 2, len(users))
assert.Equal(t, []int{1, 1}, lengths)
assert.NotEqual(t, uint(0), users[0].ID) assert.NotEqual(t, uint(0), users[0].ID)
assert.Equal(t, "User1", users[0].Name) assert.Equal(t, "User1", users[0].Name)
assert.Equal(t, "US", users[0].Address.Country)
assert.NotEqual(t, uint(0), users[1].ID) assert.NotEqual(t, uint(0), users[1].ID)
assert.Equal(t, "User2", users[1].Name) assert.Equal(t, "User2", users[1].Name)
assert.Equal(t, []int{1, 1}, lengths) assert.Equal(t, "BR", users[1].Address.Country)
}) })
t.Run("should load partially filled chunks correctly", func(t *testing.T) { t.Run("should load partially filled chunks correctly", func(t *testing.T) {