package transpile import ( "bytes" "io" "testing" "github.com/spf13/afero" ) func TestPuzzle(t *testing.T) { puzzleFs := newTestFs() catFs := NewRecursiveBasePathFs(puzzleFs, "cat0") { pd := NewFsPuzzle(catFs, 1) p, err := pd.Puzzle() if err != nil { t.Error(err) } if (len(p.Answers) == 0) || (p.Answers[0] != "YAML answer") { t.Error("Answers are wrong", p.Answers) } if (len(p.Pre.Authors) != 3) || (p.Pre.Authors[1] != "Buster") { t.Error("Authors are wrong", p.Pre.Authors) } if p.Pre.Body != "

YAML body

\n" { t.Errorf("Body parsed wrong: %#v", p.Pre.Body) } } { p, err := NewFsPuzzle(catFs, 2).Puzzle() if err != nil { t.Error(err) } if (len(p.Answers) == 0) || (p.Answers[0] != "RFC822 answer") { t.Error("Answers are wrong", p.Answers) } if (len(p.Pre.Authors) != 3) || (p.Pre.Authors[1] != "Arthur") { t.Error("Authors are wrong", p.Pre.Authors) } if p.Pre.Body != "

RFC822 body

\n" { t.Errorf("Body parsed wrong: %#v", p.Pre.Body) } } if _, err := NewFsPuzzle(catFs, 3).Puzzle(); err != nil { t.Error("Legacy `puzzle.moth` file:", err) } if _, err := NewFsPuzzle(catFs, 99).Puzzle(); err == nil { t.Error("Non-existent puzzle", err) } if _, err := NewFsPuzzle(catFs, 10).Puzzle(); err == nil { t.Error("Broken YAML") } if _, err := NewFsPuzzle(catFs, 20).Puzzle(); err == nil { t.Error("Bad RFC822 header") } if _, err := NewFsPuzzle(catFs, 21).Puzzle(); err == nil { t.Error("Boken RFC822 header") } { fs := afero.NewMemMapFs() if err := afero.WriteFile(fs, "1/mkpuzzle", []byte("bleat"), 0755); err != nil { t.Error(err) } p := NewFsPuzzle(fs, 1) if _, ok := p.(FsCommandPuzzle); !ok { t.Error("We didn't get an FsCommandPuzzle") } if _, err := p.Puzzle(); err == nil { t.Error("We didn't get an error trying to run a command from a MemMapFs") } } } func TestFsPuzzle(t *testing.T) { catFs := NewRecursiveBasePathFs(NewRecursiveBasePathFs(afero.NewOsFs(), "testdata"), "static") if _, err := NewFsPuzzle(catFs, 1).Puzzle(); err != nil { t.Error(err) } if _, err := NewFsPuzzle(catFs, 2).Puzzle(); err != nil { t.Error(err) } mkpuzzleDir := NewFsPuzzle(catFs, 3) if _, err := mkpuzzleDir.Puzzle(); err != nil { t.Error(err) } if r, err := mkpuzzleDir.Open("moo.txt"); err != nil { t.Error(err) } else { defer r.Close() buf := new(bytes.Buffer) if _, err := io.Copy(buf, r); err != nil { t.Error(err) } if buf.String() != "Moo.\n" { t.Errorf("Wrong body: %#v", buf.String()) } } if r, err := mkpuzzleDir.Open("error"); err == nil { r.Close() t.Error("Error open didn't return error") } if !mkpuzzleDir.Answer("moo") { t.Error("Right answer marked wrong") } if mkpuzzleDir.Answer("wrong") { t.Error("Wrong answer marked correct") } if mkpuzzleDir.Answer("error") { t.Error("Error answer marked correct") } }