ki-backend/ki/test/test_profile_endpoint.py

239 lines
8.5 KiB
Python
Raw Normal View History

2021-07-05 19:37:05 +02:00
# SPDX-FileCopyrightText: WTF Kooperative eG <https://wtf-eg.de/>
#
# SPDX-License-Identifier: AGPL-3.0-or-later
2021-06-20 19:25:27 +02:00
import unittest
import json
2021-06-27 14:25:44 +02:00
from app import app
2021-06-26 10:51:39 +02:00
from ki.models import User
2021-06-27 14:25:44 +02:00
from ki.test.ApiTest import ApiTest
2021-06-20 19:25:27 +02:00
2021-06-27 14:25:44 +02:00
class TestProfileEndpoint(ApiTest):
2021-06-26 12:16:14 +02:00
maxDiff = None
2021-06-27 14:25:44 +02:00
def test_update_profile_unauthorised(self):
login_data = {"username": "klaus", "password": "jutta"}
login_response = self.client.post("/users/login", data=json.dumps(login_data), content_type="application/json")
2021-06-20 19:25:27 +02:00
2021-06-27 14:25:44 +02:00
self.assertEqual(login_response.status_code, 200)
self.assertIn("token", login_response.json)
2021-06-20 19:25:27 +02:00
2021-06-27 14:25:44 +02:00
response = self.client.post("/users/1/profile",
data=json.dumps({}),
content_type="application/json",
headers={"Authorization": "Bearer " + login_response.json["token"]})
2021-06-20 19:25:27 +02:00
2021-06-27 14:25:44 +02:00
self.assertEqual(response.status_code, 403)
2021-06-20 20:13:19 +02:00
2021-06-26 12:16:14 +02:00
def test_update_profile(self):
2021-06-27 14:25:44 +02:00
token = self.login("peter", "geheim")["token"]
2021-06-20 20:13:19 +02:00
data = {
2021-06-26 12:16:14 +02:00
"pronouns": "Monsieur",
2021-06-20 20:13:19 +02:00
"volunteerwork": "ja",
"freetext": "Hallo",
2021-07-02 16:26:36 +02:00
"visible": True,
2021-06-27 12:20:36 +02:00
"address": {
"name": "Peeeda",
"street": "Bachstraße",
"house_number": "42x",
"additional": "oben",
"postcode": "23232",
"city": "Travemünde",
"country": "Deutschland"
2021-06-27 13:07:54 +02:00
},
"contacts": [{
"id": 1,
"contacttype": {
2021-07-02 16:49:14 +02:00
"id": 4,
2021-06-27 13:07:54 +02:00
"name": "Matrix"
},
"content": "@peeda:wtf-eg.de"
}, {
"contacttype": {
"name": "Rohrpost"
},
"content": "Ausgang 2"
2021-06-27 13:38:16 +02:00
}],
"skills": [{
"id": 1,
"skill": {
"id": 3,
"name": "Python"
},
"level": 4
}, {
"skill": {
"name": "Tschunkproduktion"
},
"level": 5
2021-06-27 13:51:07 +02:00
}],
"languages": [{
"id": 1,
"language": {
"id": "de",
"name": "Deutsch"
},
"level": 4
}, {
"language": {
"id": "es",
"name": "Spanisch"
},
"level": 2
2021-06-27 13:07:54 +02:00
}]
2021-06-20 20:13:19 +02:00
}
response = self.client.post("/users/1/profile",
data=json.dumps(data),
content_type="application/json",
2021-06-27 14:25:44 +02:00
headers={"Authorization": "Bearer " + token})
2021-06-20 20:13:19 +02:00
self.assertEqual(response.status_code, 200)
2021-06-20 20:21:49 +02:00
with app.app_context():
user = User.query.filter(User.id == 1).first()
profile = user.profile
2021-06-26 12:16:14 +02:00
self.assertEqual("Monsieur", profile.pronouns)
2021-06-20 20:21:49 +02:00
self.assertEqual("ja", profile.volunteerwork)
self.assertEqual("Hallo", profile.freetext)
2021-07-02 16:26:36 +02:00
self.assertTrue(profile.visible)
2021-06-20 20:13:19 +02:00
2021-06-27 12:20:36 +02:00
address = profile.address
self.assertEqual(address.name, "Peeeda")
2021-06-27 13:07:54 +02:00
self.assertEqual(address.street, "Bachstraße")
self.assertEqual(address.house_number, "42x")
self.assertEqual(address.additional, "oben")
self.assertEqual(address.postcode, "23232")
self.assertEqual(address.city, "Travemünde")
self.assertEqual(address.country, "Deutschland")
contacts = profile.contacts
self.assertEqual(len(contacts), 2)
first_contact = contacts[0]
self.assertEqual(first_contact.contacttype.name, "Matrix")
self.assertEqual(first_contact.content, "@peeda:wtf-eg.de")
second_contact = contacts[1]
self.assertEqual(second_contact.contacttype.name, "Rohrpost")
self.assertEqual(second_contact.content, "Ausgang 2")
2021-06-27 12:20:36 +02:00
2021-06-27 13:51:07 +02:00
skills = profile.skills
self.assertEqual(len(skills), 2)
first_skill = skills[0]
self.assertEqual(first_skill.skill.id, 3)
self.assertEqual(first_skill.skill.name, "Python")
self.assertEqual(first_skill.level, 4)
second_skill = skills[1]
self.assertEqual(second_skill.skill.id, 13)
self.assertEqual(second_skill.skill.name, "Tschunkproduktion")
self.assertEqual(second_skill.level, 5)
languages = profile.languages
self.assertEqual(len(languages), 2)
first_language = languages[0]
self.assertEqual(first_language.language_id, "de")
self.assertEqual(first_language.level, 4)
second_language = languages[1]
self.assertEqual(second_language.language_id, "es")
self.assertEqual(second_language.level, 2)
2021-07-02 16:33:48 +02:00
def test_get_profile_unauthorised(self):
response = self.client.get("/users/1/profile")
self.assertEqual(response.status_code, 401)
2021-06-20 19:25:27 +02:00
def test_get_profile(self):
login_data = {"username": "peter", "password": "geheim"}
2021-06-27 13:38:16 +02:00
login_response = self.client.post("/users/login", data=json.dumps(login_data), content_type="application/json")
2021-06-20 19:25:27 +02:00
self.assertEqual(login_response.status_code, 200)
self.assertIn("token", login_response.json)
response = self.client.get("/users/1/profile",
2021-06-27 13:38:16 +02:00
headers={"Authorization": "Bearer " + login_response.json["token"]})
2021-06-20 19:25:27 +02:00
self.assertEqual(response.status_code, 200)
2021-06-26 12:16:14 +02:00
self.assertDictEqual(
2021-06-20 19:25:27 +02:00
response.json, {
"profile": {
2021-06-26 12:16:14 +02:00
"user_id": 1,
2021-06-26 10:51:39 +02:00
"nickname": "peternichtlustig",
"pronouns": "Herr Dr. Dr.",
2021-06-26 12:16:14 +02:00
"freetext": "Ich mag Kaffee",
"volunteerwork": "Gartenverein",
2021-07-02 16:26:36 +02:00
"visible": False,
2021-06-26 12:16:14 +02:00
"address": {
"additional": "Hinterhaus",
"city": "Bielefeld",
"country": "Deutschland",
"house_number": "23i",
"id": 1,
"name": "Peter Nichtlustig",
"postcode": "13337",
"profile_id": 1,
"street": "Waldweg"
},
"contacts": [{
"id": 1,
"profile_id": 1,
"contacttype": {
2021-07-02 16:49:14 +02:00
"id": 4,
2021-06-26 12:16:14 +02:00
"name": "Matrix"
},
"content": "@peter:wtf-eg.de"
2021-06-27 13:07:54 +02:00
}, {
"id": 2,
"profile_id": 1,
"contacttype": {
2021-07-02 16:49:14 +02:00
"id": 1,
2021-06-27 13:07:54 +02:00
"name": "E-Mail"
},
"content": "peter@wtf-eg.de"
2021-06-26 12:16:14 +02:00
}],
"skills": [{
2021-06-27 13:38:16 +02:00
"profile_id": 1,
"skill": {
"id": 1,
"name": "PHP",
"icon_url": "/skills/1/icon"
},
"level": 5
}, {
2021-06-26 12:16:14 +02:00
"profile_id": 1,
"skill": {
"id": 3,
"name": "Python",
"icon_url": "/skills/3/icon"
},
2021-06-27 13:38:16 +02:00
"level": 3
2021-06-26 12:16:14 +02:00
}],
"languages": [{
"profile_id": 1,
"language": {
"id": "de",
"name": "Deutsch",
"icon_url": "/languages/de/icon"
},
"level": 5
}, {
"profile_id": 1,
"language": {
"id": "fr",
"name": "Französisch",
"icon_url": "/languages/fr/icon"
},
"level": 3
}]
2021-06-20 19:25:27 +02:00
}
})
if __name__ == "main":
unittest.main()