New Case.ModifyCase method

v2/case.go

@@ -42,6 +42,12 @@ func (c *Case) AddQuery(key, value string) *Case {
  return c
 }
 
+// ModifyCase allows user do whatever to the case (even
+// rewrite a new one) without interrupting the chaining.
+func (c *Case) ModifyCase(fn func(c *Case) *Case) *Case {
+ return fn(c)
+}
+
 // Expect appends an expectation to the Case
 func (c *Case) Expect(exp Expectation) *Case {
  c.Expectations = append(c.Expectations, exp)

v2/case_test.go

@@ -345,6 +345,30 @@ func TestCase_Describe(t *testing.T) {
  }
 }
 
+func TestCase_ModifyCase(t *testing.T) {
+ var urlResult string
+
+ testVal := RandString(20)
+ r, err := http.NewRequest("GET", "http://example.com/foo/bar", nil)
+ if err != nil {
+ t.Fatalf("unexpected error: %s", err)
+ }
+
+ c := &restit.Case{
+ Request: r,
+ }
+ c.
+ AddQuery("hello", testVal).
+ ModifyCase(func(c *restit.Case) *restit.Case {
+ urlResult = c.Request.URL.String()
+ return c
+ })
+
+ if want, have := "http://example.com/foo/bar?hello="+testVal, urlResult; want != have {
+ t.Errorf("expected %#v, got %#v", want, have)
+ }
+}
+
 func TestCase_Expect(t *testing.T) {
  c := &restit.Case{}
  str := RandString(20)