| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154 | import unittestimport jsonclass FunctionalTests(unittest.TestCase):    def setUp(self):        from frameworkbenchmarks import main        from webtest import TestApp        app = main({}, **{"sqlalchemy.url": "postgresql:///frameworkbenchmarks"})        self.testapp = TestApp(app)    def _get(self, url, content_type="application/json"):        res = self.testapp.get(url, status=200)        self.assertTrue("Content-Length" in res.headers)        return res    def _test_obj(self, obj):        self.assertTrue("id" in obj)        self.assertTrue("randomNumber" in obj)        self.assertTrue(1 <= obj["randomNumber"] <= 10000)    def test_json(self):        """        /json        """        res = self._get("/json")        self.assertEqual(res.body, b"""{"message":"Hello, World!"}""")    def test_db(self):        """        /db        """        res = self._get("/db")        obj = json.loads(res.body)        self._test_obj(obj)    def test_queries_0(self):        """        /queries?queries=0        """        res = self._get("/queries?queries=0")        self.assertEqual(len(json.loads(res.body)), 1)    def test_queries_999(self):        """        /queries?queries=999        """        res = self._get("/queries?queries=999")        self.assertEqual(len(json.loads(res.body)), 500)    def test_queries_10(self):        """        /queries?queries=10 objects        """        res = self._get("/queries?queries=10")        objset = json.loads(res.body)        for obj in objset:            self._test_obj(obj)    def test_fortunes(self):        """        /fortunes        """        res = self._get("/fortunes")        self.assertEqual(res.body.decode('utf-8').strip(), fortunes.strip())    def test_updates(self):        """        /updates?queries=10        """        res = self._get("/updates?queries=10")        objset = json.loads(res.body)        # don't bother with more...        for obj in objset:            self._test_obj(obj)    def test_plaintext(self):        """        /plaintext        """        res = self._get("/plaintext", content_type="text/plain")        self.assertEqual(res.body, b"Hello, World!")fortunes = """<!DOCTYPE html><html><head>    <title>Fortunes</title></head><body>    <table>    <tr>        <th>id</th>        <th>message</th>    </tr>    <tr>        <td>11</td>        <td><script>alert("This should not be displayed in a browser alert box.");</script></td>    </tr>    <tr>        <td>4</td>        <td>A bad random number generator: 1, 1, 1, 1, 1, 4.33e+67, 1, 1, 1</td>    </tr>    <tr>        <td>5</td>        <td>A computer program does what you tell it to do, not what you want it to do.</td>    </tr>    <tr>        <td>2</td>        <td>A computer scientist is someone who fixes things that aren't broken.</td>    </tr>    <tr>        <td>8</td>        <td>A list is only as strong as its weakest link. — Donald Knuth</td>    </tr>    <tr>        <td>0</td>        <td>Additional fortune added at request time.</td>    </tr>    <tr>        <td>3</td>        <td>After enough decimal places, nobody gives a damn.</td>    </tr>    <tr>        <td>7</td>        <td>Any program that runs right is obsolete.</td>    </tr>    <tr>        <td>10</td>        <td>Computers make very fast, very accurate mistakes.</td>    </tr>    <tr>        <td>6</td>        <td>Emacs is a nice operating system, but I prefer UNIX. — Tom Christaensen</td>    </tr>    <tr>        <td>9</td>        <td>Feature: A bug with seniority.</td>    </tr>    <tr>        <td>1</td>        <td>fortune: No such file or directory</td>    </tr>    <tr>        <td>12</td>        <td>フレームワークのベンチマーク</td>    </tr>    </table></body></html>"""
 |