Fix schemagen and test generated code

cmd/schemagen/testdata/models_test.go

@@ -0,0 +1,91 @@
+package schemagentest
+
+import (
+	"flag"
+	"github.com/google/go-cmp/cmp"
+	"strings"
+	"testing"
+	"time"
+
+	"github.com/gocql/gocql"
+	"github.com/scylladb/gocqlx/v2"
+	"github.com/scylladb/gocqlx/v2/qb"
+)
+
+var flagCluster = flag.String("cluster", "127.0.0.1", "a comma-separated list of host:port or host tuples")
+
+func TestModelLoad(t *testing.T) {
+	session, err := gocqlx.WrapSession(gocql.NewCluster(strings.Split(*flagCluster, ",")...).CreateSession())
+	if err != nil {
+		t.Fatal("create session:", err.Error())
+	}
+	defer session.Close()
+
+	// Keyspace, types and table are created at `schemaget_test.go` at `createTestSchema`
+
+	song := SongsStruct{
+		Id:       gocql.TimeUUID(),
+		Title:    "title",
+		Album:    "album",
+		Artist:   "artist",
+		Duration: gocql.Duration{Nanoseconds: int64(5 * time.Minute)},
+		Tags:     []string{"tag1", "tag2"},
+		Data:     []byte("data"),
+	}
+
+	err = qb.Insert("schemagen.songs").
+		Columns("id", "title", "album", "artist", "duration", "tags", "data").
+		Query(session).
+		BindStruct(&song).
+		Exec()
+	if err != nil {
+		t.Fatal("failed to insert song:", err.Error())
+	}
+
+	loadedSong := SongsStruct{}
+	err = qb.Select("schemagen.songs").
+		Columns("id", "title", "album", "artist", "duration", "tags", "data").
+		Where(qb.Eq("id")).
+		Query(session).
+		BindMap(map[string]interface{}{"id": song.Id}).
+		Get(&loadedSong)
+	if err != nil {
+		t.Fatal("failed to select song:", err)
+	}
+	if diff := cmp.Diff(song, loadedSong); diff != "" {
+		t.Error("loaded song is different from inserted song:", diff)
+	}
+
+	pl := PlaylistsStruct{
+		Id:     gocql.TimeUUID(),
+		Title:  "title",
+		Album:  AlbumUserType{Name: "album", Songwriters: []string{"songwriter1", "songwriter2"}},
+		Artist: "artist",
+		SongId: gocql.TimeUUID(),
+	}
+
+	err = qb.Insert("schemagen.playlists").
+		Columns("id", "title", "album", "artist", "song_id").
+		Query(session).
+		BindStruct(&pl).
+		Exec()
+	if err != nil {
+		t.Fatal("failed to insert playlist:", err.Error())
+	}
+
+	loadedPl := PlaylistsStruct{}
+
+	err = qb.Select("schemagen.playlists").
+		Columns("id", "title", "album", "artist", "song_id").
+		Where(qb.Eq("id")).
+		Query(session).
+		BindMap(map[string]interface{}{"id": pl.Id}).
+		Get(&loadedPl)
+	if err != nil {
+		t.Fatal("failed to select playlist:", err.Error())
+	}
+
+	if diff := cmp.Diff(pl, loadedPl); diff != "" {
+		t.Error("loaded playlist is different from inserted song:", diff)
+	}
+}