class TestQuery(unittest.TestCase): def setUp(self): self.begin_at = datetime(2016, 1, 1) self.end_at = datetime(2016, 1, 2) self.aq = ApdexQuery(self.begin_at, self.end_at, 0.1) def test_call(self): self.aq.get_satisfied_request_count = MagicMock(return_value=1) self.aq.get_tolerating_request_count = MagicMock(return_value=2) self.aq.get_request_count = MagicMock(return_value=3) self.assertAlmostEqual(self.aq.query(), (1 + 1) / 3.0)
def get_apdex(self): aq = ApdexQuery(self.begin_at, self.end_at, self.APDEX_THRESHOLD) return aq.query()