package business

import (
	"strings"
	"testing"
)

func TestMySQLBusinessQueryIndexesAreAdditiveAndIdempotent(t *testing.T) {
	db := openRehearsalMySQL(t)
	definitions := map[string]string{
		"tab_promote":                  "id BIGINT NOT NULL,source_db VARCHAR(10) NOT NULL,create_time BIGINT,PRIMARY KEY(id,source_db)",
		"tab_spend":                    "id BIGINT NOT NULL,source_db VARCHAR(10) NOT NULL,promote_id BIGINT,pay_status INT,pay_time BIGINT,pay_amount DECIMAL(20,2),PRIMARY KEY(id,source_db)",
		"tab_sw_report_day":            "dedup_mode INT,stat_date DATE,source_db VARCHAR(10),promote_id BIGINT,game_id BIGINT,total_pay DECIMAL(20,2)",
		"tab_sw_report_day_key":        "dedup_mode INT,stat_date DATE,source_db VARCHAR(10),promote_id BIGINT,game_id BIGINT,metric INT,identity_key VARCHAR(64)",
		"tab_sw_report_server_day":     "game_name VARCHAR(64),dedup_mode INT,stat_date DATE,source_db VARCHAR(10),game_id BIGINT,server_num VARCHAR(32),total_pay DECIMAL(20,2)",
		"tab_sw_report_server_day_key": "game_name VARCHAR(64),dedup_mode INT,stat_date DATE,source_db VARCHAR(10),game_id BIGINT,server_num VARCHAR(32),metric INT,identity_key VARCHAR(64)",
	}
	for table, columns := range definitions {
		if err := db.Exec("CREATE TABLE `" + table + "` (" + columns + ") ENGINE=InnoDB").Error; err != nil {
			t.Fatal(err)
		}
	}
	if err := db.Exec("INSERT INTO tab_spend VALUES(77,'lx',55,1,123456,12.30)").Error; err != nil {
		t.Fatal(err)
	}
	for attempt := 0; attempt < 2; attempt++ {
		if err := migrateBusinessQueryIndexes(db); err != nil {
			t.Fatal(err)
		}
	}
	for _, index := range businessQueryIndexes {
		var columns []string
		if err := db.Raw("SELECT column_name FROM information_schema.statistics WHERE table_schema=DATABASE() AND table_name=? AND index_name=? ORDER BY seq_in_index", index.table, index.name).Scan(&columns).Error; err != nil {
			t.Fatal(err)
		}
		if strings.Join(columns, ",") != strings.Join(index.columns, ",") {
			t.Fatalf("index %s has unexpected columns: %v", index.name, columns)
		}
	}
	var amount string
	if err := db.Raw("SELECT pay_amount FROM tab_spend WHERE id=77 AND source_db='lx'").Scan(&amount).Error; err != nil || amount != "12.30" {
		t.Fatalf("index migration changed business data: %s, %v", amount, err)
	}
	if err := db.Exec("ALTER TABLE tab_spend DROP INDEX idx_sw_spend_time_summary, ADD INDEX idx_sw_spend_time_summary(pay_time)").Error; err != nil {
		t.Fatal(err)
	}
	if err := migrateBusinessQueryIndexes(db); err == nil || !strings.Contains(err.Error(), "idx_sw_spend_time_summary") {
		t.Fatalf("must reject a conflicting existing index: %v", err)
	}
}