Compare commits
2 commits
37645e37f7
...
3184ce476a
Author | SHA1 | Date | |
---|---|---|---|
3184ce476a | |||
d2a2e4526b |
3 changed files with 33 additions and 3 deletions
15
mlp/mlp.go
15
mlp/mlp.go
|
@ -41,3 +41,18 @@ const (
|
||||||
BlockOdyssey BlockID = "EO" // Odyssey block - EO/HM/MT
|
BlockOdyssey BlockID = "EO" // Odyssey block - EO/HM/MT
|
||||||
BlockDefenders BlockID = "DE" // Defenders block - DE/SB/FF
|
BlockDefenders BlockID = "DE" // Defenders block - DE/SB/FF
|
||||||
)
|
)
|
||||||
|
|
||||||
|
var allSets = []SetID{
|
||||||
|
SetPremiere,
|
||||||
|
SetCanterlotNights,
|
||||||
|
SetRockNRave,
|
||||||
|
SetCelestialSolstice,
|
||||||
|
SetCrystalGames,
|
||||||
|
SetAbsoluteDiscord,
|
||||||
|
SetEquestrialOdysseys,
|
||||||
|
SetHighMagic,
|
||||||
|
SetMarksInTime,
|
||||||
|
SetDefendersOfEquestria,
|
||||||
|
SetSeaquestriaBeyond,
|
||||||
|
SetFriendsForever,
|
||||||
|
}
|
||||||
|
|
11
mlp/set.go
11
mlp/set.go
|
@ -120,3 +120,14 @@ func LoadSetHTTP(id SetID) (*Set, error) {
|
||||||
|
|
||||||
return LoadSetBytes(id, data)
|
return LoadSetBytes(id, data)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// LoadAllSets just loads all sets from the web
|
||||||
|
func LoadAllSets() error {
|
||||||
|
for _, set := range allSets {
|
||||||
|
_, err := LoadSetHTTP(set)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
|
@ -75,6 +75,12 @@ func TestAllLoads(t *testing.T) {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("[LoadSetMemory] Could not load set: %s", err.Error())
|
t.Fatalf("[LoadSetMemory] Could not load set: %s", err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Load all remaining sets
|
||||||
|
err = mlp.LoadAllSets()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("[LoadAllSets] Could not load remaining sets: %s", err.Error())
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// TestNotLoadedErr tests that LoadSetMemory fails if set is not cached
|
// TestNotLoadedErr tests that LoadSetMemory fails if set is not cached
|
||||||
|
@ -166,10 +172,8 @@ func TestMalformedJSONLoad(t *testing.T) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
var testServer *httptest.Server
|
|
||||||
|
|
||||||
func TestMain(m *testing.M) {
|
func TestMain(m *testing.M) {
|
||||||
testServer = httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
|
testServer := httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
|
||||||
switch req.URL.Path {
|
switch req.URL.Path {
|
||||||
case "/nopenope.json":
|
case "/nopenope.json":
|
||||||
// 404
|
// 404
|
||||||
|
|
Loading…
Reference in a new issue