def get(self): parser = reqparse.RequestParser() parser.add_argument('department', type=int) args = parser.parse_args() links = Static.consultation_links() teachers = list() department = args['department'] if department is not None: department = args['department'] if 0 <= department <= 6: teachers = parse_teachers(links[department]) else: return None, 400 else: for department, link in enumerate(links): teachers += parse_teachers(links[department]) if teachers: return teachers, 200 return None, 204
import random import unittest from unittest import TestCase from app import app import routes.consultations from static import Static links = Static.consultation_links() class FlaskTests(TestCase): def setUp(self): self.client = app.test_client(self) def test_root(self): response = self.client.get('/') self.assertEqual(response.status_code, 302) def test_changes(self): response = self.client.get('/changes') self.assertEqual(response.status_code, 200) def test_whole_consultations(self): response = self.client.get('/consultations') self.assertEqual(response.status_code, 200) def test_random_consultations(self): random_consultations = random.randint(0, 6) print(f'Randomly selected consultations: {random_consultations}') response = self.client.get('/consultations',