Python 3.x WSGI Testing Framework - PullRequest
Новая
       5

Python 3.x WSGI Testing Framework

1 голос
/ 01 сентября 2011

Учитывая, что у webtest нет версии 3.x (или есть планы по ее разработке), существуют ли какие-либо решения для автоматического тестирования системы приложения WSGI?Я знаю unittest для модульного тестирования - меня больше интересует момент тестирования целых систем.

Я не ищу инструментов, которые помогут разработать приложение - просто тест это.

1 Ответ

2 голосов
/ 02 декабря 2011

В случае, если кто-то еще столкнется с этим, я сам в конечном итоге написал решение.
Вот очень простой класс, который я использую - я просто наследую от WSGIBaseTest вместо TestCase и получаю метод self.request(), в который я могу передавать запросы.
Он сохраняет cookies и автоматически отправляет их в приложение при последующих запросах (до вызова self.new_session()). 

import unittest
from wsgiref import util
import io

class WSGIBaseTest(unittest.TestCase):
    '''Base class for unit-tests. Provides up a simple interface to make requests
    as though they came through a wsgi interface from a user.'''
    def setUp(self):
        '''Set up a fresh testing environment before each test.'''
        self.cookies = []
    def request(self, application, url, post_data = None):
        '''Hand a request to the application as if sent by a client.
        @param application: The callable wsgi application to test.
        @param url: The URL to make the request against.
        @param post_data: A string.'''
        self.response_started = False
        temp = io.StringIO(post)
        environ = {
            'PATH_INFO': url,
            'REQUEST_METHOD': 'POST' if post_data else 'GET',
            'CONTENT_LENGTH': len(post),
            'wsgi.input': temp,
            }
        util.setup_testing_defaults(environ)
        if self.cookies:
            environ['HTTP_COOKIE'] = ';'.join(self.cookies)
        self.response = ''
        for ret in application(environ, self._start_response):
            assert self.response_started
            self.response += str(ret)
        temp.close()
        return response
    def _start_response(self, status, headers):
        '''A callback passed into the application, to simulate a wsgi
        environment.

        @param status: The response status of the application ("200", "404", etc)
        @param headers: Any headers to begin the response with.
        '''
        assert not self.response_started
        self.response_started = True
        self.status = status
        self.headers = headers
        for header in headers:
            # Parse out any cookies and save them to send with later requests.
            if header[0] == 'Set-Cookie':
                var = header[1].split(';', 1)
                if len(var) > 1 and var[1][0:9] == ' Max-Age=':
                    if int(var[1][9:]) > 0:
                        # An approximation, since our cookies never expire unless
                        # explicitly deleted (by setting Max-Age=0).
                        self.cookies.append(var[0])
                    else:
                        index = self.cookies.index(var[0])
                        self.cookies.pop(index)
    def new_session(self):
        '''Start a new session (or pretend to be a different user) by deleting
        all current cookies.'''
        self.cookies = []
...