tests.py

from app import app
import unittest
import sys
from flask import request, url_for
from flask_testing import TestCase

class TestPageLoad(unittest.TestCase):

    def setUp(self):
        """
        создание тестового веб-сервиса
        """
        app.config['TESTING'] = True
        app.config['WTF_CSRF_ENABLED'] = False
        app.config['DEBUG'] = False

        sys.argv = ['app.py', 'static/urls.yml']
        self.app = app.test_client()
        self.assertEqual(app.debug, False)

    def test_main_page(self):
        """
        тест, проверяющий загрузку главной страницы
        """
        response = self.app.get('/', follow_redirects=True)
        self.assertEqual(response.status_code, 200)

    def test_wrong_short_url(self):
        """
        тест, проверяющий результат неправильно введенной короткой ссылки
        """
        tester = app.test_client(self)
        response = tester.get('/wwww', content_type='html/text')
        self.assertEqual(response.status_code, 404)

    def test_right_short_url(self):
        """
        тест, проверяющий результат правильно введенной короткой ссылки
        """
        tester = app.test_client(self)
        response = tester.get('/yah44', content_type='html/text')
        self.assertEqual(response.status_code, 302)

    def test_post_wrong_long_url(self):
        """
        тест, проверяющий результат неправильно введенной длинной ссылки (POST-запрос)
        """
        tester = app.test_client(self)
        response = tester.post('/add', data=dict(url='wrong_url'), follow_redirects=True)
        self.assertEqual(response.status_code, 404)

    def test_post_right_long_url(self):
        """
        тест, проверяющий результат правильно введенной длинной ссылки (POST-запрос)
        """
        tester = app.test_client(self)
        response = tester.post('/add', data=dict(url='https://yandex.ru/'), follow_redirects=True)
        self.assertEqual(response.status_code, 200)


    # def test_redirect(self):
    #      tester = app.test_client(self)
    #      with tester:
    #          response = tester.get(url_for('/yah44'), follow_redirects=True)
    #          assert response.request.path == 'https://www.yahoo.com/'


# class TestClientUtils(TestCase):
#     def test_assert_redirects(self):
#         response = self.client.get("'/yah44'")
#         self.assertRedirects(response, "https://www.yahoo.com/")

if __name__ == '__main__':
    unittest.main()