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