2 files changed, 161 insertions, 0 deletions

diff --git a/xdelta3/go/src/regtest.go b/xdelta3/go/src/regtest.go
new file mode 100644
index 0000000..aee419d
--- /dev/null
+++ b/xdelta3/go/src/regtest.go
@@ -0,0 +1,83 @@
+package main
+
+import (
+        "io"
+        "io/ioutil"
+        "xdelta"
+)
+
+const (
+        prog = "/Users/jmacd/src/xdelta/xdelta3/xdelta3"
+)
+
+func drain(f io.ReadCloser) <-chan []byte {
+        c := make(chan []byte)
+        go func() {
+                if b, err := ioutil.ReadAll(f); err != nil {
+                        panic(err)
+                } else {
+                        c <- b
+                }
+        }()
+        return c
+}
+
+func empty(f io.ReadCloser) {
+        go func() {
+                if _, err := ioutil.ReadAll(f); err != nil {
+                        panic(err)
+                }
+        }()
+}
+
+func write(f io.WriteCloser, b []byte) {
+        if _, err := f.Write(b); err != nil {
+                panic(err)
+        }
+        if err := f.Close(); err != nil {
+                panic(err)
+        }
+}
+
+func main() {
+        x := xdelta.Program{prog}
+        smokeTest(x)
+}
+
+func smokeTest(p xdelta.Program) {
+        target := "Hello world!"
+        source := "Hello world, nice to meet you!"
+        
+        run, err := p.Exec(true, []string{"-e"})
+        if err != nil {
+                panic(err)
+        }
+        encodeout := drain(run.Stdout)
+        empty(run.Stderr)
+
+        write(run.Stdin, []byte(target))
+        write(run.Srcin, []byte(source))
+
+        if err := run.Cmd.Wait(); err != nil {
+                panic(err)
+        }
+
+        run, err = p.Exec(true, []string{"-d"})
+        if err != nil {
+                panic(err)
+        }
+
+        decodeout := drain(run.Stdout)
+        empty(run.Stderr)
+
+        write(run.Stdin, <-encodeout)
+        write(run.Srcin, []byte(source))
+
+        if err := run.Cmd.Wait(); err != nil {
+                panic(err)
+        }
+
+        if string(<-decodeout) != target {
+                panic("It's not working!!!")
+        }
+}
diff --git a/xdelta3/go/src/xdelta/test.go b/xdelta3/go/src/xdelta/test.go
new file mode 100644
index 0000000..d586538
--- /dev/null
+++ b/xdelta3/go/src/xdelta/test.go
@@ -0,0 +1,78 @@
+package xdelta
+
+import (
+        "io"
+        "io/ioutil"
+        "os"
+        "os/exec"
+        "path"
+        "golang.org/x/sys/unix"
+)
+
+var (
+        tmpDir = "/tmp"
+)
+
+type Program struct {
+        Path string
+}
+
+type Run struct {
+        Cmd exec.Cmd
+        Testdir string
+        Srcfile string
+        Stdin io.WriteCloser
+        Srcin io.WriteCloser
+        Stdout io.ReadCloser
+        Stderr io.ReadCloser
+}       
+
+func (b *Program) Exec(srcfifo bool, flags []string) (*Run, error) {
+        var err error
+        run := &Run{}
+        if run.Testdir, err = ioutil.TempDir(tmpDir, "xrt"); err != nil {
+                return nil, err
+        }
+        args := []string{b.Path}
+        if srcfifo {
+                run.Srcfile = path.Join(run.Testdir, "source")
+                if err = unix.Mkfifo(run.Srcfile, 0600); err != nil {
+                        return nil, err
+                }
+                // Because OpenFile blocks on the Fifo until the reader
+                // arrives, a pipe to defer open
+                read, write := io.Pipe()
+                run.Srcin = write
+
+                go func() {
+                        fifo, err := os.OpenFile(run.Srcfile, os.O_WRONLY, 0600)
+                        if err != nil {
+                                panic(err)
+                        }
+                        if _, err := io.Copy(fifo, read); err != nil {
+                                panic(err)
+                        }
+                        if err := fifo.Close(); err != nil {
+                                panic(err)
+                        }
+                }()
+                args = append(args, "-s")
+                args = append(args, run.Srcfile)
+        }
+        if run.Stdin, err = run.Cmd.StdinPipe(); err != nil {
+                return nil, err
+        }
+        if run.Stdout, err = run.Cmd.StdoutPipe(); err != nil {
+                return nil, err
+        }
+        if run.Stderr, err = run.Cmd.StderrPipe(); err != nil {
+                return nil, err
+        }
+
+        run.Cmd.Path = b.Path
+        run.Cmd.Args = append(args, flags...)
+        run.Cmd.Dir = run.Testdir
+        run.Cmd.Start()
+
+        return run, nil
+}