Exemplo n.º 1
0
class TestCountResults(TestTSVResults):

    PATH = "/testservice/countservice"
    FORMAT = "count"
    EXPECTED_RESULTS = ['25']
    EXPECTED_COUNT = 25

    def testCount(self):
        """Should be able to get count as an integer"""
        attempts = 0
        def do_tests(error=None):
            if attempts < 5:
                try:
                    self.assertEqual(self.query.count(), self.EXPECTED_COUNT)
                    self.assertEqual(self.template.count(), self.EXPECTED_COUNT)
                except IOError, e:
                    do_tests(e)
            else:
                raise RuntimeError("Error connecting to " + self.query.service.root, error)

        do_tests()


if __name__ == '__main__':
    server = TestServer()
    server.start()
    time.sleep(0.1) # Avoid race conditions with the server
    unittest.main()
    server.shutdown()
Exemplo n.º 2
0
                    self.assertRaises(ModelError,
                                      lambda: departments[idx].company.foo)

                assertEqual(len(departments), 8)

        self.do_unpredictable_test(logic)


class TestCountResults(TestTSVResults):  # pragma: no cover

    PATH = "/testservice/countservice"
    FORMAT = "count"
    EXPECTED_RESULTS = ['25']
    EXPECTED_COUNT = 25

    def testCount(self):
        """Should be able to get count as an integer"""
        def logic():
            self.assertEqual(self.query.count(), self.EXPECTED_COUNT)
            self.assertEqual(self.template.count(), self.EXPECTED_COUNT)

        self.do_unpredictable_test(logic)


if __name__ == '__main__':  # pragma: no cover
    server = TestServer()
    server.start()
    time.sleep(0.1)  # Avoid race conditions with the server
    unittest.main()
    server.shutdown()