def test_datehist_agg(self): """_datehist_agg should return an A with proper values.""" # pylint: disable=W0212 result = SpawnsData._datehist_agg(self.interval, self.start, self.end) self.assertDictEqual(result.to_dict(), self.DATEHIST_AGG)
def test_spawn_finish_query(self): """_spawn_finish_query should return a Search with proper values.""" expected_aggs = {'per_interval': self.DATEHIST_AGG} expected_aggs['per_interval']['aggs'] = { 'per_success': { 'terms': { 'size': 0, 'field': 'success', 'shard_min_doc_count': 0, 'min_doc_count': 0 } } } expected_query = deepcopy(self.QUERY_BASE) expected_query['bool']['must'].append({'term': {'event': 'finish'}}) # pylint: disable=W0212 result = SpawnsData._spawn_finish_query(self.start, self.end, self.interval) self.assertDictEqual(result.to_dict()['query'], expected_query) self.assertListEqual(result.to_dict()['sort'], self.TIMESTAMP_SORT) self.assertDictEqual(result.to_dict()['aggs'], expected_aggs)
def test_spawn_finish_query(self): """_spawn_finish_query should return a Search with proper values.""" expected_aggs = {"per_interval": self.DATEHIST_AGG} expected_aggs["per_interval"]["aggs"] = { "per_success": {"terms": {"size": 0, "field": "success", "shard_min_doc_count": 0, "min_doc_count": 0}} } expected_query = deepcopy(self.QUERY_BASE) expected_query["bool"]["must"].append({"term": {"event": "finish"}}) # pylint: disable=W0212 result = SpawnsData._spawn_finish_query(self.start, self.end, self.interval) self.assertDictEqual(result.to_dict()["query"], expected_query) self.assertListEqual(result.to_dict()["sort"], self.TIMESTAMP_SORT) self.assertDictEqual(result.to_dict()["aggs"], expected_aggs)
def test_spawn_finish_query(self): """_spawn_finish_query should return a Search with proper values.""" self.maxDiff = None expected_aggs = {'per_interval': self.DATEHIST_AGG} expected_aggs['per_interval']['aggs'] = { 'per_success': { 'terms': {'size': 0, 'field': 'success.raw', 'shard_min_doc_count': 0, 'min_doc_count': 0}} } expected_query = deepcopy(self.QUERY_BASE) expected_query['bool']['must'].append({'term': {'event': 'finish'}}) # pylint: disable=W0212 result = SpawnsData._spawn_finish_query(self.start, self.end, self.interval) self.assertDictEqual(result.to_dict()['query'], expected_query) self.assertListEqual(result.to_dict()['sort'], self.TIMESTAMP_SORT) self.assertDictEqual(result.to_dict()['aggs'], expected_aggs)