From 0cfd53700dd0379e9c7c18686e5df33d286be95e Mon Sep 17 00:00:00 2001
From: David Medina <davidmedina9@gmail.com>
Date: Fri, 10 Feb 2012 21:54:38 +0100
Subject: Wip on services.users

---
 pygithub3/tests/utils/base.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

(limited to 'pygithub3/tests/utils')

diff --git a/pygithub3/tests/utils/base.py b/pygithub3/tests/utils/base.py
index 571d8ac..d9049c9 100644
--- a/pygithub3/tests/utils/base.py
+++ b/pygithub3/tests/utils/base.py
@@ -11,12 +11,20 @@ def mock_json(content):
     return content
 
 
-def mock_response(status_code=200, content={}):
+def mock_response(status_code='get', content={}):
+    CODES = dict(get=200, patch=200, post=201, delete=204)
     response = Mock(name='response')
-    response.status_code = status_code
+    response.status_code = CODES[str(status_code).lower()] or status_code
     response.content = content
     return response
 
+
+def mock_response_result(status_code='get'):
+    response = mock_response(status_code, content=[{}, {}])
+    response.headers = {'link': ''}
+    return response
+
+
 class DummyResource(Resource):
     pass
 
-- 
cgit v1.2.3-59-g8ed1b