1
0
mirror of https://github.com/cheat/cheat.git synced 2024-12-21 20:25:06 +01:00
cheat/internal/sheet/copy_test.go
Chris Lane 233a9de1aa feat: implement --all flag
Implement an `--all` flag that can be used to view cheatsheets on all
chaetpaths. (Resolves )
2020-11-27 16:39:34 -05:00

109 lines
2.4 KiB
Go

package sheet
import (
"io/ioutil"
"os"
"path"
"testing"
)
// TestCopyFlat asserts that Copy correctly copies files at a single level of
// depth
func TestCopyFlat(t *testing.T) {
// mock a cheatsheet file
text := "this is the cheatsheet text"
src, err := ioutil.TempFile("", "foo-src")
if err != nil {
t.Errorf("failed to mock cheatsheet: %v", err)
}
defer src.Close()
defer os.Remove(src.Name())
if _, err := src.WriteString(text); err != nil {
t.Errorf("failed to write to mock cheatsheet: %v", err)
}
// mock a cheatsheet struct
sheet, err := New("foo", "community", src.Name(), []string{}, false)
if err != nil {
t.Errorf("failed to init cheatsheet: %v", err)
}
// compute the outfile's path
outpath := path.Join(os.TempDir(), sheet.Title)
defer os.Remove(outpath)
// attempt to copy the cheatsheet
err = sheet.Copy(outpath)
if err != nil {
t.Errorf("failed to copy cheatsheet: %v", err)
}
// assert that the destination file contains the correct text
got, err := ioutil.ReadFile(outpath)
if err != nil {
t.Errorf("failed to read destination file: %v", err)
}
if string(got) != text {
t.Errorf(
"destination file contained wrong text: want: '%s', got: '%s'",
text,
got,
)
}
}
// TestCopyDeep asserts that Copy correctly copies files at several levels of
// depth
func TestCopyDeep(t *testing.T) {
// mock a cheatsheet file
text := "this is the cheatsheet text"
src, err := ioutil.TempFile("", "foo-src")
if err != nil {
t.Errorf("failed to mock cheatsheet: %v", err)
}
defer src.Close()
defer os.Remove(src.Name())
if _, err := src.WriteString(text); err != nil {
t.Errorf("failed to write to mock cheatsheet: %v", err)
}
// mock a cheatsheet struct
sheet, err := New(
"/cheat-tests/alpha/bravo/foo",
"community",
src.Name(),
[]string{},
false,
)
if err != nil {
t.Errorf("failed to init cheatsheet: %v", err)
}
// compute the outfile's path
outpath := path.Join(os.TempDir(), sheet.Title)
defer os.RemoveAll(path.Join(os.TempDir(), "cheat-tests"))
// attempt to copy the cheatsheet
err = sheet.Copy(outpath)
if err != nil {
t.Errorf("failed to copy cheatsheet: %v", err)
}
// assert that the destination file contains the correct text
got, err := ioutil.ReadFile(outpath)
if err != nil {
t.Errorf("failed to read destination file: %v", err)
}
if string(got) != text {
t.Errorf(
"destination file contained wrong text: want: '%s', got: '%s'",
text,
got,
)
}
}