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)
