mirror of https://github.com/dirtbags/moth.git
64 lines
914 B
Go
64 lines
914 B
Go
package main
|
|
|
|
import (
|
|
"archive/zip"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestMothball(t *testing.T) {
|
|
tf, err := ioutil.TempFile("", "mothball")
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
defer os.Remove(tf.Name())
|
|
|
|
w := zip.NewWriter(tf)
|
|
f, err := w.Create("moo.txt")
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
// no Close method
|
|
|
|
_, err = fmt.Fprintln(f, "The cow goes moo")
|
|
//.Write([]byte("The cow goes moo"))
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
w.Close()
|
|
tf.Close()
|
|
|
|
// Now read it in
|
|
mb, err := OpenMothball(tf.Name())
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
|
|
cow, err := mb.Open("moo.txt")
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
|
|
line := make([]byte, 200)
|
|
n, err := cow.Read(line)
|
|
if (err != nil) && (err != io.EOF) {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
|
|
if string(line[:n]) != "The cow goes moo\n" {
|
|
t.Log(line)
|
|
t.Error("Contents didn't match")
|
|
return
|
|
}
|
|
|
|
}
|