def test_get_headers_api_names(self): builder = request_builder.SA360ReportRequestBuilder(AGENCY_ID) self.assertEqual(builder.get_headers(ui_names=False), [ 'advertiser', 'account', 'campaign', 'keywordText', 'keywordMatchType', 'keywordMaxCpc', 'clicks', 'keywordId', 'advertiserId' ])
def test_no_advertiser(self): builder = request_builder.SA360ReportRequestBuilder(AGENCY_ID) start_date = datetime.date(YEAR, MONTH, DAY) end_date = datetime.date(YEAR, MONTH + 1, DAY) body = builder.build(start_date, end_date) self.expected_body['timeRange']['startDate'] = DATE_FORMAT % ( YEAR, MONTH, DAY) self.expected_body['timeRange']['endDate'] = DATE_FORMAT % ( YEAR, MONTH + 1, DAY) self.assertEqual(body, self.expected_body)
def test_get_headers(self): builder = request_builder.SA360ReportRequestBuilder(AGENCY_ID) self.assertEqual(builder.get_headers(), [ 'Advertiser', 'Account', 'Campaign', 'Keyword', 'Match type', 'Keyword max CPC', 'Clicks', 'Keyword ID', 'Advertiser ID', ])
def test_custom_columns(self): custom_columns = [{'columnName': 'columnA'}, {'columnName': 'columnB'}] builder = request_builder.SA360ReportRequestBuilder( AGENCY_ID, columns=custom_columns) start_date = datetime.date(YEAR, MONTH, DAY) end_date = datetime.date(YEAR, MONTH + 1, DAY) body = builder.build(start_date, end_date) self.expected_body['timeRange']['startDate'] = DATE_FORMAT % ( YEAR, MONTH, DAY) self.expected_body['timeRange']['endDate'] = DATE_FORMAT % ( YEAR, MONTH + 1, DAY) self.expected_body['columns'] = custom_columns self.assertEqual(body, self.expected_body)
def test_advertiser_list(self): advertiser_id_list = ['adv1', 'adv2', 'adv3'] builder = request_builder.SA360ReportRequestBuilder( AGENCY_ID, advertiser_id_list) start_date = datetime.date(YEAR, MONTH, DAY) end_date = datetime.date(YEAR, MONTH + 1, DAY) body = builder.build(start_date, end_date) self.expected_body['timeRange']['startDate'] = DATE_FORMAT % ( YEAR, MONTH, DAY) self.expected_body['timeRange']['endDate'] = DATE_FORMAT % ( YEAR, MONTH + 1, DAY) self.expected_body['filters'].insert( 1, { 'column': { 'columnName': 'advertiserId' }, 'operator': 'in', 'values': advertiser_id_list }) self.assertEqual(body, self.expected_body)
def test_get_headers_api_names_custom_columns(self): custom_columns = [{'columnName': 'columnA'}, {'columnName': 'columnB'}] builder = request_builder.SA360ReportRequestBuilder( AGENCY_ID, columns=custom_columns) headers = builder.get_headers(ui_names=False) self.assertEqual(headers, ['columnA', 'columnB'])
def test_not_valid_end_date(self): builder = request_builder.SA360ReportRequestBuilder(AGENCY_ID) start_date = datetime.date(YEAR, MONTH, DAY) end_date = 'not_valid' with self.assertRaises(AttributeError): builder.build(start_date, end_date)