44 lines
1.3 KiB
Go
44 lines
1.3 KiB
Go
//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) != 12 {
|
|
t.Fatalf("created %d group tables, want 12", 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)
|
|
}
|
|
}
|
|
}
|