package envelope import ( "bytes" "encoding/xml" "errors" "fmt" "io" "reflect" "testing" "github.com/google/go-cmp/cmp" ) type testArgs struct { Foo string Bar string } // TestWriteRead tests the round-trip of writing an envelope and reading it back. func TestWriteRead(t *testing.T) { argsIn := &testArgs{ Foo: "foo-1", Bar: "bar-2", } actionIn := NewSendAction("urn:schemas-upnp-org:service:FakeService:1", "MyAction", argsIn) buf := &bytes.Buffer{} err := Write(buf, actionIn) if err != nil { t.Fatalf("Write want success, got err=%v", err) } t.Logf("Encoded envelope:\n%v", buf) argsOut := &testArgs{} actionOut := NewRecvAction(argsOut) err = Read(buf, actionOut) if err != nil { t.Errorf("Read want success, got err=%v", err) } if diff := cmp.Diff(actionIn, actionOut); diff != "" { t.Errorf("\nwant actionOut=%+v\ngot %+v\ndiff:\n%s", actionIn, actionOut, diff) } if diff := cmp.Diff(argsIn, argsOut); diff != "" { t.Errorf("\nwant argsOut=%+v\ngot %+v\ndiff:\n%s", argsIn, argsOut, diff) } } // TestRead tests read against a semi-real encoded envelope. func TestRead(t *testing.T) { env := []byte(` foo-1 bar-2 `) argsOut := &testArgs{} actionOut := NewRecvAction(argsOut) if err := Read(bytes.NewBuffer(env), actionOut); err != nil { t.Fatalf("Read want success, got err=%v", err) } wantArgsOut := &testArgs{ Foo: "foo-1", Bar: "bar-2", } if diff := cmp.Diff(wantArgsOut, argsOut); diff != "" { t.Errorf("want argsOut=%+v, got %+v\ndiff:\n%s", wantArgsOut, argsOut, diff) } } func TestReadFault(t *testing.T) { env := []byte(xml.Header + ` dummy code dummy string dummy actor dummy detail `) err := Read(bytes.NewBuffer(env), NewRecvAction(&testArgs{})) if err == nil { t.Fatal("want err != nil, got nil") } gotFault, ok := err.(*Fault) if !ok { t.Fatalf("want *Fault, got %T", err) } wantFault := &Fault{ Code: "dummy code", String: "dummy string", Actor: "dummy actor", Detail: FaultDetail{Raw: []byte("dummy detail")}, } if !reflect.DeepEqual(wantFault, gotFault) { t.Errorf("want %+v, got %+v", wantFault, gotFault) } } func TestFault(t *testing.T) { tests := []struct { name string err error wantIs bool }{ {"plain fault", &Fault{Code: "code"}, true}, {"wrapped fault", fmt.Errorf("wrapper: %w", &Fault{Code: "code"}), true}, {"other error", io.EOF, false}, } for _, test := range tests { test := test // copy for closure t.Run(test.name, func(t *testing.T) { if got, want := errors.Is(test.err, ErrFault), test.wantIs; got != want { t.Errorf("got errors.Is(%v, ErrFault)=>%t, want %t", test.err, got, want) } if !test.wantIs { return } var fault *Fault if got, want := errors.As(test.err, &fault), true; got != want { t.Fatalf("got errors.As(%v, ...)=>%t, want %t", test.err, got, want) } if got, want := fault.Code, "code"; got != want { t.Errorf("got fault.Code=%q, want %q", got, want) } }) } }