//go:build mysql package mysql import ( "database/sql" "strings" "testing" ) type recordingGroupSchemaExecutor struct { statements []string } func (e *recordingGroupSchemaExecutor) Exec(query string, _ ...any) (sql.Result, error) { e.statements = append(e.statements, query) return nil, nil } func TestEnsureIMGroupBusinessTablesUsesSelectedTenantDatabase(t *testing.T) { exec := &recordingGroupSchemaExecutor{} if err := ensureIMGroupBusinessTables(exec); err != nil { t.Fatal(err) } if len(exec.statements) != 13 { t.Fatalf("created %d group tables, want 13", len(exec.statements)) } for _, statement := range exec.statements { if strings.Contains(strings.ToLower(statement), "tinode.") || strings.Contains(strings.ToLower(statement), "create database") { t.Fatalf("group schema must use the selected tenant database: %s", statement) } } joined := strings.ToLower(strings.Join(exec.statements, "\n")) for _, expected := range []string{ "primary key(id), unique index uk_im_group_settings_topic(topic)", "primary key(id), unique index uk_im_group_announcement_reads_announcement_user(announcement_id, user_id)", "primary key(id), unique index uk_im_group_mutes_topic_user(topic, user_id)", } { if !strings.Contains(joined, expected) { t.Fatalf("group schema is missing snowflake primary-key definition %q", expected) } } }