Пример #1
0
class TestRes:
    def setUp(self):
        self.res = Response(start_res)

    def test_status(self):
        assert self.res.status_code == 200
        assert self.res.status(500) is self.res
        assert self.res.status_code == 500
        assert self.res.status_text is None
        assert self.res.status(418, "I\"m a teapot") is self.res
        assert self.res.status_text == "I\"m a teapot"

    def test_header(self):
        assert not self.res.headers
        assert self.res.header('X-Test', 'test too') is self.res
        assert self.res.headers['X-Test'] == 'test too'

    def test_push(self):
        assert not self.res.body
        assert self.res.push('push') is self.res
        assert self.res.body[-1] == b'push'

    def test_ok(self):
        result = self.res.header('X-Test', 'test too').ok()
        assert isinstance(result, Result)
        assert self.res.ok(True).ok()

    def test_err(self):
        assert self.res.err(True).err()
        try:
            raise self.res.err(True)
        except Err as err:
            assert err.err()

    def test_redirect(self):
        assert self.res.redirect(True).err()
        assert self.res.status_code == 302
Пример #2
0
class TestRes:
    def setUp(self):
        self.res = Response(start_res)

    def test_status(self):
        assert self.res.status_code == 200
        assert self.res.status(500) is self.res
        assert self.res.status_code == 500
        assert self.res.status_text is None
        assert self.res.status(418, "I\"m a teapot") is self.res
        assert self.res.status_text == "I\"m a teapot"

    def test_header(self):
        assert not self.res.headers
        assert self.res.header('X-Test', 'test too') is self.res
        assert self.res.headers['X-Test'] == 'test too'

    def test_push(self):
        assert not self.res.body
        assert self.res.push('push') is self.res
        assert self.res.body[-1] == b'push'

    def test_ok(self):
        result = self.res.header('X-Test', 'test too').ok()
        assert isinstance(result, Result)
        assert self.res.ok(True).ok()

    def test_err(self):
        assert self.res.err(True).err()
        try:
            raise self.res.err(True)
        except Err as err:
            assert err.err()

    def test_redirect(self):
        assert self.res.redirect(True).err()
        assert self.res.status_code == 302