104 lines
2.8 KiB
Go
104 lines
2.8 KiB
Go
// Copyright (C) 2017 ScyllaDB
|
|
// Use of this source code is governed by a ALv2-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
//go:build all || integration
|
|
// +build all integration
|
|
|
|
package gocqlx_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/gocql/gocql"
|
|
"github.com/google/go-cmp/cmp"
|
|
|
|
"github.com/scylladb/gocqlx/v2"
|
|
"github.com/scylladb/gocqlx/v2/gocqlxtest"
|
|
"github.com/scylladb/gocqlx/v2/qb"
|
|
)
|
|
|
|
func TestBatch(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
cluster := gocqlxtest.CreateCluster()
|
|
if err := gocqlxtest.CreateKeyspace(cluster, "batch_test"); err != nil {
|
|
t.Fatal("create keyspace:", err)
|
|
}
|
|
|
|
session, err := gocqlx.WrapSession(cluster.CreateSession())
|
|
if err != nil {
|
|
t.Fatal("create session:", err)
|
|
}
|
|
t.Cleanup(func() {
|
|
session.Close()
|
|
})
|
|
|
|
basicCreateAndPopulateKeyspace(t, session, "batch_test")
|
|
|
|
song := Song{
|
|
ID: mustParseUUID("60fc234a-8481-4343-93bb-72ecab404863"),
|
|
Title: "La Petite Tonkinoise",
|
|
Album: "Bye Bye Blackbird",
|
|
Artist: "Joséphine Baker",
|
|
Tags: []string{"jazz"},
|
|
Data: []byte("music"),
|
|
}
|
|
playlist := PlaylistItem{
|
|
ID: mustParseUUID("6a6255d9-680f-4cb5-b9a2-27cf4a810344"),
|
|
Title: "La Petite Tonkinoise",
|
|
Album: "Bye Bye Blackbird",
|
|
Artist: "Joséphine Baker",
|
|
SongID: mustParseUUID("60fc234a-8481-4343-93bb-72ecab404863"),
|
|
}
|
|
|
|
insertSong := qb.Insert("batch_test.songs").
|
|
Columns("id", "title", "album", "artist", "tags", "data").Query(session)
|
|
insertPlaylist := qb.Insert("batch_test.playlists").
|
|
Columns("id", "title", "album", "artist", "song_id").Query(session)
|
|
selectSong := qb.Select("batch_test.songs").Where(qb.Eq("id")).Query(session)
|
|
selectPlaylist := qb.Select("batch_test.playlists").Where(qb.Eq("id")).Query(session)
|
|
|
|
t.Run("batch inserts", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
type batchQry struct {
|
|
qry *gocqlx.Queryx
|
|
arg interface{}
|
|
}
|
|
|
|
qrys := []batchQry{
|
|
{qry: insertSong, arg: song},
|
|
{qry: insertPlaylist, arg: playlist},
|
|
}
|
|
|
|
b := session.NewBatch(gocql.LoggedBatch)
|
|
for _, qry := range qrys {
|
|
if err := b.BindStruct(qry.qry, qry.arg); err != nil {
|
|
t.Fatal("BindStruct failed:", err)
|
|
}
|
|
}
|
|
if err := session.ExecuteBatch(b); err != nil {
|
|
t.Fatal("batch execution:", err)
|
|
}
|
|
|
|
// verify song was inserted
|
|
var gotSong Song
|
|
if err := selectSong.BindStruct(song).Get(&gotSong); err != nil {
|
|
t.Fatal("select song:", err)
|
|
}
|
|
if diff := cmp.Diff(gotSong, song); diff != "" {
|
|
t.Errorf("expected %v song, got %v, diff: %q", song, gotSong, diff)
|
|
}
|
|
|
|
// verify playlist item was inserted
|
|
var gotPlayList PlaylistItem
|
|
if err := selectPlaylist.BindStruct(playlist).Get(&gotPlayList); err != nil {
|
|
t.Fatal("select song:", err)
|
|
}
|
|
if diff := cmp.Diff(gotPlayList, playlist); diff != "" {
|
|
t.Errorf("expected %v playList, got %v, diff: %q", playlist, gotPlayList, diff)
|
|
}
|
|
})
|
|
}
|