1 files changed, 8 insertions, 7 deletions
diff --git a/test/zendex/search_test.exs b/test/zendex/search_test.exs
index a5b6405..bb1f358 100644
--- a/test/zendex/search_test.exs
+++ b/test/zendex/search_test.exs
@@ -12,24 +12,25 @@ defmodule Zendex.SearchTest do
   end
 
   test "doing a search", %{conn: conn} do
-    stub = fn("http://test.zendesk.com/api/v2/search.json?query=requester%3AJimbob+type%3Aticket", _) ->
-      %HTTPoison.Response{body: Poison.encode!(["Jimbob Ticket 1", "Jimbob Ticket 2"])}
+    expected = ["Jimbob Ticket 1", "Jimbob Ticket 2"]
+
+    stub = fn("http://test.zendesk.com/api/v2/search.json?query=requester%3AJimbob+type%3Aticket", _headers) ->
+      %HTTPoison.Response{body: Poison.encode!(expected)}
     end
     :meck.expect(HTTPoison, :get!, stub)
 
-    expected = ["Jimbob Ticket 1", "Jimbob Ticket 2"]
-
     actual = Zendex.Search.query(conn, %{type: "ticket", requester: "Jimbob"})
 
     assert actual == expected
   end
 
   test "doing a search with sorting", %{conn: conn} do
-    stub = fn("http://test.zendesk.com/api/v2/search.json?query=requester%3AReginald+type%3Aticket&sort_by=created_at&sort_order=desc", _) ->
-      %HTTPoison.Response{body: Poison.encode!(["Reginald Ticket 1", "Reginald Ticket 2"])}
+    expected = ["Reginald Ticket 1", "Reginald Ticket 2"]
+
+    stub = fn("http://test.zendesk.com/api/v2/search.json?query=requester%3AReginald+type%3Aticket&sort_by=created_at&sort_order=desc", _headers) ->
+      %HTTPoison.Response{body: Poison.encode!(expected)}
     end
     :meck.expect(HTTPoison, :get!, stub)
-    expected = ["Reginald Ticket 1", "Reginald Ticket 2"]
 
     actual = Zendex.Search.query(conn,
                                  %{type: "ticket", requester: "Reginald"},