package mlp_test import ( "testing" "git.fromouter.space/mcg/draft" "git.fromouter.space/mcg/draft/mlp" ) // TestAlternates tries to get alternates (SR/UR/RR) // This might require a while since it needs to generate MANY packs! // This test *requires* an internet connection! func TestAlternates(t *testing.T) { // Load Premiere/CN as they have their own UR ratios prSet, err := mlp.LoadSetHTTP(mlp.SetPremiere) if err != nil { t.Fatalf("Could not fetch set data: %s", err.Error()) } cnSet, err := mlp.LoadSetHTTP(mlp.SetCanterlotNights) if err != nil { t.Fatalf("Could not fetch set data: %s", err.Error()) } // Load set with Royal rares eoSet, err := mlp.LoadSetHTTP(mlp.SetEquestrialOdysseys) if err != nil { t.Fatalf("Could not fetch set data: %s", err.Error()) } // Find all Premiere URs prURs := []string{} for _, card := range prSet.CardData { if card.Rarity == mlp.RarityUltraRare { prURs = append(prURs, card.ID) } } // Find all CN URs cnURs := []string{} for _, card := range cnSet.CardData { if card.Rarity == mlp.RarityUltraRare { cnURs = append(cnURs, card.ID) } } // Get some PR packs and search for URs prurfound := false for i := 0; i < 1000; i++ { pack := draft.MakePack(prSet) // Check for ultra rares for _, id := range prURs { if pack[7].ID == id { prurfound = true break } } if prurfound { t.Logf("PR Ultra Rare found after %d packs\n", i) break } } if !prurfound { t.Fatalf("No PR UR found after 1000 packs") } // Get some CN packs and search for URs cnurfound := false for i := 0; i < 1000; i++ { pack := draft.MakePack(cnSet) // Check for ultra rares for _, id := range cnURs { if pack[7].ID == id { cnurfound = true break } } if cnurfound { t.Logf("CN Ultra Rare found after %d packs\n", i) break } } if !cnurfound { t.Fatalf("No CN UR found after 1000 packs") } eorrfound := false for i := 0; i < 100000; i++ { pack := draft.MakePack(eoSet) // Check for royal rares if pack[6].ID == "eo207/rr1" || pack[6].ID == "eo208/rr2" { t.Logf("EO Royal Rare found after %d packs\n", i) eorrfound = true break } } if !eorrfound { t.Fatalf("No EO RR found after 100k packs") } } // TestPackFixedSets tries to get packs a set that isn't a true set // This should result in empty packs func TestPackFixedSets(t *testing.T) { set, err := mlp.LoadSetBytes(mlp.SetRockNRave, []byte("{}")) if err != nil { t.Fatalf("Could not load set: %s", err.Error()) } pack := draft.MakePack(set) if len(pack) != 0 { t.Fatalf("Expected an empty pack but got %d cards", len(pack)) } }