def setUp(self):
self._contact = user.Contact(-1, "Contactium", "*****@*****.**",
[])
contact_header = {
"user": self._contact.serialize().replace("\"", "\\\""),
"password": "******"
}
self._contact = user.deserialize(
get_response(
requests.post(
"https://prbw36cvje.execute-api.us-east-1.amazonaws.com/dev/user/",
headers=contact_header)))
def setUp(self):
self._citizen = user.Citizen(
-1, "Test Testington", "*****@*****.**",
[user.Contact(-1, "Contactium", "*****@*****.**", [])], [],
"Teststrasse 10", "Testerup", "1000")
citizen_header = {
"user": self._citizen.serialize().replace("\"", "\\\""),
"password": "******"
}
self._citizen = user.deserialize(
get_response(
requests.post(
"https://prbw36cvje.execute-api.us-east-1.amazonaws.com/dev/user/",
headers=citizen_header)))
def get_contact_all():
all_contacts = []
conn = psycopg2.connect(connect_str)
cursor = conn.cursor()
cursor.execute(
"SELECT users.id, users.name, users.email FROM users, contact WHERE users.id = contact.userID")
citizens_raw = cursor.fetchall()
for ctc in citizens_raw:
devices = get_user_devices(ctc[0])
all_contacts.append(user.Contact(ctc[0], ctc[1], ctc[2], devices))
conn.commit()
cursor.close()
conn.close()
return all_contacts
def get_contact(id):
if not id:
return None
conn = psycopg2.connect(connect_str)
cursor = conn.cursor()
cursor.execute(
"SELECT users.id, users.name, users.email FROM users, contact WHERE users.id = contact.userID AND users.id = %s",
[id])
ctc = cursor.fetchone()
devices = get_user_devices(ctc[0])
conn.commit()
cursor.close()
conn.close()
return user.Contact(ctc[0], ctc[1], ctc[2], devices)
def get_citizen_contacts(citizen_id):
contacts = []
conn = psycopg2.connect(connect_str)
cursor = conn.cursor()
cursor.execute(
"SELECT users.id, users.name, users.email FROM users, contact, associateswith WHERE users.id = contact.userID AND associateswith.citizenID = %s AND associateswith.contactID = users.id",
[citizen_id])
ctc = cursor.fetchall()
for contact in ctc:
devices = get_user_devices(contact[0])
contacts.append(user.Contact(contact[0], contact[1], contact[2], devices))
conn.commit()
cursor.close()
conn.close()
return contacts
def test_get_and_post_user_contact(self):
_user = user.Contact(-1, "Test Testington", "*****@*****.**", [])
user_post_uri = "https://prbw36cvje.execute-api.us-east-1.amazonaws.com/dev/user/"
user_delete_uri = "https://prbw36cvje.execute-api.us-east-1.amazonaws.com/dev/user/"
user_post_header = {
"user": _user.serialize().replace("\"", "\\\""),
"password": "******"
}
user_get_header = {"email": "*****@*****.**", "password": "******"}
_user = user.deserialize(
get_response(requests.post(user_post_uri,
headers=user_post_header)))
user_get_uri = "https://prbw36cvje.execute-api.us-east-1.amazonaws.com/dev/user/"
_user_response = user.deserialize(
get_response(requests.get(user_get_uri, headers=user_get_header)))
user_delete_header = {
"user":
_user_response.serialize().replace("\"", "\\\""),
"token":
"eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ1c2VyX2lkIjoiMjM0IiwidXNlcl9yb2xlIjoidXNlckFkbWluIn0._zMvLSI7elh4ebv-1iaE8nXYdbzjyvpBNlDNhNSJMc0"
}
_delete_response = user.deserialize(
get_response(
requests.delete(user_delete_uri, headers=user_delete_header)))
_user.token = ""
_user_response.token = ""
self.assertDictEqual(
json.loads(_user.serialize().replace("'", "\"")),
json.loads(_user_response.serialize().replace("'", "\"")))
self.assertEqual(-1, _delete_response.id)
def make_contact(self, data):
_user = user.Contact(data["id"], data["name"], data["email"],
data["devices"])
_user.token = data["token"]
return _user