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