def test_with_rules_and_enrichments(self): add_data_files(self.rules_source, _RULE_FILE_LOCATIONS) assert len(self.rules_source.get_all()) == len(_RULE_FILE_LOCATIONS) add_data_files(self.enrichment_loader_source, _ENRICHMENT_LOADER_LOCATIONS) assert len(self.enrichment_loader_source.get_all()) == len(_ENRICHMENT_LOADER_LOCATIONS) cookie_ids = _generate_cookie_ids(TestIntegration._NUMBER_OF_COOKIES) cookie_ids.append(NAME_ENRICHMENT_LOADER_MATCH_COOKIE) cookie_ids.append(NAME_RULE_MATCH_COOKIE) expected_number_of_times_processed = len(cookie_ids) * 2 + 1 block_until_processed(self.cookie_jar, cookie_ids, expected_number_of_times_processed) self.assertEqual(self.cookie_jar.mark_as_complete.call_count, expected_number_of_times_processed) self.cookie_jar.mark_as_failed.assert_not_called() rule_checker = RuleChecker(self, self.rules_source.get_all()) rule_checker.assert_call_counts( ALL_MATCH_RULE_ID, expected_number_of_times_processed, expected_number_of_times_processed) rule_checker.assert_call_counts( NO_MATCH_RULE_ID, expected_number_of_times_processed, 0) rule_checker.assert_call_counts( NAME_MATCH_RULE_ID, expected_number_of_times_processed, len(_ENRICHMENT_LOADER_LOCATIONS) - 1) rule_checker.assert_call_counts( HASH_ENRICHED_MATCH_RULE_ID, expected_number_of_times_processed, len(cookie_ids)) enrichment_loader_checker = EnrichmentLoaderChecker(self, self.enrichment_loader_source.get_all()) enrichment_loader_checker.assert_call_counts( NO_LOADER_ENRICHMENT_LOADER_ID, expected_number_of_times_processed, 0) enrichment_loader_checker.assert_call_counts( HASH_ENRICHMENT_LOADER_ID, expected_number_of_times_processed - 1, len(cookie_ids)) enrichment_loader_checker.assert_call_counts( NAME_MATCH_LOADER_ENRICHMENT_LOADER_ID, expected_number_of_times_processed, 1)
def test_with_no_rules_or_enrichments(self): cookie_ids = _generate_cookie_ids(TestIntegration._NUMBER_OF_COOKIES) block_until_processed(self.cookie_jar, cookie_ids, TestIntegration._NUMBER_OF_COOKIES) self.assertEqual(self.cookie_jar.mark_as_complete.call_count, len(cookie_ids)) self.cookie_jar.mark_as_failed.assert_not_called()