commands: Added test for Respone error marshalling

ipfs · Oct 10, 2014 · 2071940 · 2071940
1 parent 77afb23
commit 2071940
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/commands/response_test.go b/commands/response_test.go
@@ -1,6 +1,9 @@
 package commands
 
-import "testing"
+import (
+  "testing"
+  "fmt"
+)
 
 type TestOutput struct {
   Foo, Bar string
@@ -29,4 +32,14 @@ func TestMarshalling(t *testing.T) {
   if output != "{\"Foo\":\"beep\",\"Bar\":\"boop\",\"Baz\":1337}" {
     t.Error("Incorrect JSON output")
   }
+
+  res.SetError(fmt.Errorf("You broke something!"), Client)
+  bytes, err = res.Marshal()
+  if err != nil {
+    t.Error("Should have passed")
+  }
+  output = string(bytes)
+  if output != "{\"Message\":\"You broke something!\",\"Code\":1}" {
+    t.Error("Incorrect JSON output")
+  }
 }