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)))
