def test_expected_headers_all(self, mocked_report_type_fields): mocked_report_type_fields.return_value = samples.SAMPLE_REPORT_TYPE_FIELDS fields = samples.SAMPLE_REPORT_TYPE_FIELDS reports = Reports(credentials={}) self.assertEqual(set(fields), set(reports.expected_headers('CLICK_PERFORMANCE_REPORT')))
def test_expected_headers_subset(self, mocked_report_type_fields): mocked_report_type_fields.return_value = samples.SAMPLE_REPORT_TYPE_FIELDS fields = samples.SAMPLE_REPORT_TYPE_FIELDS[:5] names = [field.name for field in fields] reports = Reports(credentials={}) self.assertEqual(set(fields), set(reports.expected_headers('CLICK_PERFORMANCE_REPORT', names)))