class TestQuery(unittest.TestCase):

    def setUp(self):

        self.begin_at = datetime(2016, 1, 1)
        self.end_at = datetime(2016, 1, 2)
        self.sccq = StatusCodeCountQuery(self.begin_at, self.end_at, 200)

        self.mock_get_es()

    def mock_get_es(self):

        self.sccq.get_es = MagicMock()
        self.sccq.get_es.return_value = self.mock_es = MagicMock()
        self.mock_es.count.return_value = {'count': 1}

    def test_call(self):

        self.assertEqual(self.sccq.query(), 1)

        trc = TimeRangeClause(begin_time=self.begin_at, end_time=self.end_at)
        rc = RangeClause('backend_status_code', min_val=200, max_val=300)
        self.mock_es.count.assert_called_with(
            index='logstash-2016.01.01',
            body={
                'filter': {
                    'bool': {'filter': [trc.get_clause(), rc.get_clause()]}
                }
            }
        )
    def get_http_status_code_count(self, status_code_class):

        sccq = StatusCodeCountQuery(
            self.begin_at,
            self.end_at,
            status_code_class
        )
        return sccq.query()
Example #3
0
class TestQuery(unittest.TestCase):
    def setUp(self):

        self.begin_at = datetime(2016, 1, 1)
        self.end_at = datetime(2016, 1, 2)
        self.sccq = StatusCodeCountQuery(self.begin_at, self.end_at, 200)

        self.mock_get_es()

    def mock_get_es(self):

        self.sccq.get_es = MagicMock()
        self.sccq.get_es.return_value = self.mock_es = MagicMock()
        self.mock_es.count.return_value = {'count': 1}

    def test_call(self):

        field = 'backend_status_code'
        self.assertEqual(self.sccq.query(), 1)
        self.mock_es.count.assert_called_with(
            index='logstash-2016.01.01',
            body={
                'query': {
                    'bool': {
                        'filter': [{
                            'range': {
                                'timestamp': {
                                    'gte': 1451606400000,
                                    'lt': 1451692800000
                                }
                            }
                        }, {
                            'exists': {
                                'field': 'rails.controller#action'
                            }
                        }, {
                            'term': {
                                'domain_name': 'api.thekono.com'
                            }
                        }, {
                            'range': {
                                'backend_processing_time': {
                                    'gte': 0
                                }
                            }
                        }, {
                            'range': {
                                field: {
                                    'gte': 200,
                                    'lt': 300
                                }
                            }
                        }]
                    }
                }
            })
Example #4
0
    def get_http_status_code_count(self, status_code_class):

        sccq = StatusCodeCountQuery(self.begin_at, self.end_at,
                                    status_code_class)
        return sccq.query()