示例#1
0
 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
示例#2
0
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',