Add pembody, mwc tests.
This commit is contained in:
parent
3fd30fef69
commit
57a33bcbcc
|
@ -0,0 +1,28 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/pem"
|
||||||
|
"flag"
|
||||||
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
|
|
||||||
|
"github.com/kisom/goutils/lib"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
flag.Parse()
|
||||||
|
if flag.NArg() != 1 {
|
||||||
|
lib.Errx(lib.ExitFailure, "a single filename is required")
|
||||||
|
}
|
||||||
|
|
||||||
|
in, err := ioutil.ReadFile(flag.Arg(0))
|
||||||
|
if err != nil {
|
||||||
|
lib.Err(lib.ExitFailure, err, "couldn't read file")
|
||||||
|
}
|
||||||
|
|
||||||
|
p, _ := pem.Decode(in)
|
||||||
|
if p == nil {
|
||||||
|
lib.Errx(lib.ExitFailure, "%s isn't a PEM-encoded file", flag.Arg(0))
|
||||||
|
}
|
||||||
|
fmt.Printf("%s", p.Bytes)
|
||||||
|
}
|
|
@ -0,0 +1,27 @@
|
||||||
|
package mwc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/kisom/testio"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestMWC(t *testing.T) {
|
||||||
|
buf1 := testio.NewBufCloser(nil)
|
||||||
|
buf2 := testio.NewBufCloser(nil)
|
||||||
|
|
||||||
|
mwc := MultiWriteCloser(buf1, buf2)
|
||||||
|
|
||||||
|
if _, err := mwc.Write([]byte("hello, world")); err != nil {
|
||||||
|
t.Fatalf("%v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !bytes.Equal(buf1.Bytes(), buf2.Bytes()) {
|
||||||
|
t.Fatal("write failed")
|
||||||
|
}
|
||||||
|
|
||||||
|
if !bytes.Equal(buf1.Bytes(), []byte("hello, world")) {
|
||||||
|
t.Fatal("writing failed")
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue