def setUp(self):
self.client = app.test_client()
self.ong = OngDocument(name='ong', description='The coolest ong').save()
self.user = UserDocument(name="iury",
email="*****@*****.**",
_password="******",
ong_id=self.ong.id).save()
def setUp(self):
UserDocument.drop_collection()
self.ong = OngDocument(name='ong', description='The coolest ong').save()
self.user = UserDocument(name="iury",
email="*****@*****.**",
_password="******",
ong_id=self.ong.id).save()
def get_search_query(phrase):
query = Q(
'function_score',
query=MultiMatch(
fields=['email'],
query=phrase
),
)
return UserDocument.search().query(query)
def test_get_many_users(self):
UserDocument(name='willian', email='*****@*****.**',
_password='******',
ong_id=self.ong.id).save()
limit = 10
response = self.client.get("/user/?limit={limit}".format(limit=limit))
decoded_response = json.loads(response.data.decode())
self.assertEqual(len(decoded_response), 2)
self.assertEqual(response.status_code, 200)
def tearDown(self):
UserDocument.drop_collection()
OngDocument.drop_collection()
def setUp(self):
self.user = UserDocument(name="iury", email="*****@*****.**", _password="******").save()
class UserTests(unittest.TestCase):
def setUp(self):
self.client = app.test_client()
self.ong = OngDocument(name='ong', description='The coolest ong').save()
self.user = UserDocument(name="iury",
email="*****@*****.**",
_password="******",
ong_id=self.ong.id).save()
def tearDown(self):
UserDocument.drop_collection()
OngDocument.drop_collection()
def test_get_one_user(self):
response = self.client.get("/user/{id}".format(id=self.user.id))
decoded_response = json.loads(response.data.decode())
self.assertEquals(decoded_response, self.user.to_dict())
def test_get_nonexistent_user(self):
id_ = '123abc'
response = self.client.get("/user/{id}".format(id=id_))
self.assertEquals(response.status_code, 404)
def test_get_many_users(self):
UserDocument(name='willian', email='*****@*****.**',
_password='******',
ong_id=self.ong.id).save()
limit = 10
response = self.client.get("/user/?limit={limit}".format(limit=limit))
decoded_response = json.loads(response.data.decode())
self.assertEqual(len(decoded_response), 2)
self.assertEqual(response.status_code, 200)
def test_malformed_user_request(self):
response = self.client.get("/user/")
decoded_response = json.loads(response.data.decode())
self.assertEqual(response.status_code, 400)
self.assertDictEqual(decoded_response, {'message': 'You must provide limit or id'})
def test_add_user(self):
data = {
'name': 'travis',
'email': '*****@*****.**',
'password': '******',
'ong_id': self.ong.id
}
response = self.client.post("/user/", data=data)
saved_user = UserDocument.objects.get(email='*****@*****.**')
decoded_response = json.loads(response.data.decode())
self.assertEqual(response.status_code, 201)
self.assertDictEqual(decoded_response, saved_user.to_dict())
def test_delete_user(self):
id_ = self.user.id
response = self.client.delete('/user/{id}'.format(id=id_))
self.assertEqual(response.status_code, 204)
def test_delete_user_insufficient_params(self):
response = self.client.delete('/user/')
decoded_response = json.loads(response.data.decode())
self.assertEqual(response.status_code, 400)
self.assertDictEqual(decoded_response, {"message": "You must provide an id"})
def test_put_user_insufficient_params(self):
response = self.client.put('/user/')
decoded_response = json.loads(response.data.decode())
self.assertEqual(response.status_code, 400)
self.assertDictEqual(decoded_response, {"message": "You must provide an id"})
def test_put_user(self):
data = {"name": "newname", "email": "*****@*****.**"}
response = self.client.put('/user/{id}'.format(id=self.user.id), data=data)
user = UserDocument.objects.get(id=self.user.id)
self.assertEqual(response.status_code, 200)
self.assertEqual(user.name, data['name'])
self.assertEqual(user.email, data['email'])
def test_put_not_found_user(self):
response = self.client.put("/user/1012")
self.assertEqual(response.status_code, 404)
def setUp(self):
UserDocument.drop_collection()
self.user = UserDocument(name="iury", email="*****@*****.**", _password="******").save()
