def save(self, hash_name, key, value):
_key = '%s%s' % (hash_name, key)
try:
Redis(connection_pool=self._redis_pool).set(_key, utils.encode_obj(value))
return value
except Exception:
return None
def post_user(self):
self.user_id = utils.fake_name()
path = '/users'
body = utils.encode_obj({'id': self.user_id})
headers = {'Content-Type': 'application/json'}
response = self.simulate_post(path, body=body, headers=headers)
response_user_id = utils.decode_obj(response[0])['id']
self.assertEquals(self.user_id, response_user_id)
self.assertEqual(self.srmock.status, falcon.HTTP_201)
def return_body(self, resp, body, status=None):
resp.body = encode_obj(body)
resp.set_header('Content-Type', 'application/json')
resp.content_type = "application/json"
if status:
resp.status = status
def on_get(self, req, resp, *args, **kwargs):
resp.body = encode_obj({'status': 'ok'})
resp.status = falcon.HTTP_200
def post_url(self):
url = utils.fake_url()
path = '/users/{0}/urls'.format(self.user_id)
body = utils.encode_obj({'url': url})
headers = {'Content-Type': 'application/json'}
return self.simulate_post(path, body=body, headers=headers)
def post_user_registered(self):
path = '/users'
body = utils.encode_obj({'id': self.user_id})
headers = {'Content-Type': 'application/json'}
response = self.simulate_post(path, body=body, headers=headers)
self.assertEqual(self.srmock.status, falcon.HTTP_409)