def testValidDoubleSubset(self):
selector_file_contents = """{
"file_format_version": 1,
"duration": "30d",
"metric":"average_rtt",
"ip_translation":{
"strategy":"maxmind",
"params":{
"db_snapshots":["2014-08-04"]
}
},
"subsets":[
{
"site":"lga02",
"client_provider":"comcast",
"start_time":"2014-02-01T00:00:00Z"
},
{
"site":"lga01",
"client_provider":"comcast",
"start_time":"2014-02-01T00:00:00Z"
}
]
}"""
selector1_expected = selector.Selector()
selector1_expected.start_time = utils.make_datetime_utc_aware(
datetime.datetime(2014, 2, 1))
selector1_expected.duration = 30 * 24 * 60 * 60
selector1_expected.metric = 'average_rtt'
selector1_expected.site_name = 'lga02'
selector1_expected.ip_translation_spec = (
iptranslation.IPTranslationStrategySpec(
'maxmind', {'db_snapshots': ['2014-08-04']}))
selector1_expected.client_provider = 'comcast'
selector2_expected = selector.Selector()
selector2_expected.start_time = utils.make_datetime_utc_aware(
datetime.datetime(2014, 2, 1))
selector2_expected.duration = 30 * 24 * 60 * 60
selector2_expected.metric = 'average_rtt'
selector2_expected.ip_translation_spec = (
iptranslation.IPTranslationStrategySpec(
'maxmind', {'db_snapshots': ['2014-08-04']}))
selector2_expected.site_name = 'lga01'
selector2_expected.client_provider = 'comcast'
selectors_expected = [selector1_expected, selector2_expected]
self.assertParsedSelectorsMatch(selectors_expected,
selector_file_contents)
def testEncodeSimpleSelectorA(self):
s = selector.Selector()
s.start_time = datetime.datetime(2015, 4, 2, 10, 27, 34)
s.duration = 45
s.site_name = 'mia01'
s.client_provider = 'twc'
s.metric = 'upload_throughput'
s.ip_translation_spec = (iptranslation.IPTranslationStrategySpec(
'maxmind', {'db_snapshots': ['2015-02-05']}))
encoded_expected = """
{
"file_format_version": 1.0,
"duration": "45d",
"metric":"upload_throughput",
"ip_translation":{
"strategy":"maxmind",
"params":{
"db_snapshots":["2015-02-05"]
}
},
"subsets":[
{
"site":"mia01",
"client_provider":"twc",
"start_time":"2015-04-02T10:27:34Z"
}
]
}"""
encoded_actual = selector.SelectorJsonEncoder().encode(s)
self.assertJsonEqual(encoded_expected, encoded_actual)
def testEncodeSimpleSelectorB(self):
s = selector.Selector()
s.start_time = datetime.datetime(2014, 2, 1)
s.duration = 22
s.site_name = 'lga02'
s.client_provider = 'comcast'
s.metric = 'download_throughput'
s.ip_translation_spec = (iptranslation.IPTranslationStrategySpec(
'maxmind', {'db_snapshots': ['2014-08-04']}))
encoded_expected = """
{
"file_format_version": 1.0,
"duration": "22d",
"metric":"download_throughput",
"ip_translation":{
"strategy":"maxmind",
"params":{
"db_snapshots":["2014-08-04"]
}
},
"subsets":[
{
"site":"lga02",
"client_provider":"comcast",
"start_time":"2014-02-01T00:00:00Z"
}
]
}"""
encoded_actual = selector.SelectorJsonEncoder().encode(s)
self.assertJsonEqual(encoded_expected, encoded_actual)
def _createDummyMaxmindStrategySpec(self):
strategy_params = {
'maxmind_dir': '/fake/dir',
'db_snapshots': ['2012-01-01']
}
return iptranslation.IPTranslationStrategySpec('maxmind',
strategy_params)
def testEncodeMultiSelectorOneElement(self):
s = selector.MultiSelector()
s.start_times = [datetime.datetime(2015, 4, 2, 10, 27, 34)]
s.duration = 45
s.sites = ['mia01']
s.client_providers = ['twc']
s.client_countries = ['us']
s.metrics = ['upload_throughput']
s.ip_translation_spec = (iptranslation.IPTranslationStrategySpec(
'maxmind', {'db_snapshots': ['2015-02-05']}))
encoded_expected = """
{
"file_format_version": 1.1,
"duration": "45d",
"metrics": ["upload_throughput"],
"ip_translation": {
"strategy": "maxmind",
"params": {
"db_snapshots": ["2015-02-05"]
}
},
"sites": ["mia01"],
"client_providers": ["twc"],
"client_countries": ["us"],
"start_times": ["2015-04-02T10:27:34Z"]
}"""
encoded_actual = selector.MultiSelectorJsonEncoder().encode(s)
self.assertJsonEqual(encoded_expected, encoded_actual)
def testEncodeMultiSelectorMultiElement(self):
s = selector.MultiSelector()
s.start_times = [
datetime.datetime(2015, 4, 1, 0, 0, 0),
datetime.datetime(2015, 4, 8, 0, 0, 0),
datetime.datetime(2015, 4, 15, 0, 0, 0),
]
s.duration = 7
s.sites = ['iad01', 'lga06', 'mia01', 'nuq03']
s.client_providers = ['comcast', 'twc', 'verizon']
s.metrics = ['download_throughput', 'upload_throughput', 'minimum_rtt']
s.ip_translation_spec = (iptranslation.IPTranslationStrategySpec(
'maxmind', {'db_snapshots': ['2015-02-05']}))
encoded_expected = """
{
"file_format_version": 1.1,
"duration": "7d",
"metrics": ["download_throughput", "upload_throughput", "minimum_rtt"],
"ip_translation": {
"strategy": "maxmind",
"params": {
"db_snapshots": ["2015-02-05"]
}
},
"sites": ["iad01", "lga06", "mia01", "nuq03"],
"client_providers": ["comcast", "twc", "verizon"],
"start_times": ["2015-04-01T00:00:00Z",
"2015-04-08T00:00:00Z",
"2015-04-15T00:00:00Z"]
}"""
encoded_actual = selector.MultiSelectorJsonEncoder().encode(s)
self.assertJsonEqual(encoded_expected, encoded_actual)
def testValidInput_v1dot1_EmptyListValue_OptionalParameter(self):
"""Empty list on optional parameter (client_countries) is handled as None"""
selector_file_contents = """{
"file_format_version": 1.1,
"duration": "30d",
"metrics": ["average_rtt"],
"ip_translation":{
"strategy":"maxmind",
"params":{
"db_snapshots":["2014-08-04"]
}
},
"client_countries": [],
"start_times": ["2014-02-01T00:00:00Z"]
}"""
selector_expected = selector.Selector()
selector_expected.start_time = utils.make_datetime_utc_aware(
datetime.datetime(2014, 2, 1))
selector_expected.duration = 30 * 24 * 60 * 60
selector_expected.metric = 'average_rtt'
selector_expected.ip_translation_spec = (
iptranslation.IPTranslationStrategySpec(
'maxmind', {'db_snapshots': ['2014-08-04']}))
selector_expected.client_country = None
self.assertParsedSingleSelectorMatches(selector_expected,
selector_file_contents)
def testValidInput_v1dot1_Simple_SingleLocationValues_onlyClientCountries(
self):
selector_file_contents = """{
"file_format_version": 1.1,
"duration": "30d",
"metrics": ["average_rtt"],
"ip_translation":{
"strategy":"maxmind",
"params":{
"db_snapshots":["2014-08-04"]
}
},
"client_countries": ["us"],
"start_times": ["2014-02-01T00:00:00Z"]
}"""
selector_expected = selector.Selector()
selector_expected.start_time = utils.make_datetime_utc_aware(
datetime.datetime(2014, 2, 1))
selector_expected.duration = 30 * 24 * 60 * 60
selector_expected.metric = 'average_rtt'
selector_expected.ip_translation_spec = (
iptranslation.IPTranslationStrategySpec(
'maxmind', {'db_snapshots': ['2014-08-04']}))
selector_expected.client_country = 'us'
self.assertParsedSingleSelectorMatches(selector_expected,
selector_file_contents)
def testValidInput_v1dot1_Simple_CountriesCaseInsensitivity(self):
selector_file_contents = """{
"file_format_version": 1.1,
"duration": "30d",
"metrics": ["average_rtt"],
"ip_translation":{
"strategy":"maxmind",
"params":{
"db_snapshots":["2014-08-04"]
}
},
"start_times": ["2014-02-01T00:00:00Z"],
"client_countries": ["us", "Ca", "uK", "AU"]
}"""
selectors_expected = []
selector_base = selector.Selector()
selector_base.start_time = utils.make_datetime_utc_aware(
datetime.datetime(2014, 2, 1))
selector_base.duration = 30 * 24 * 60 * 60
selector_base.metric = "average_rtt"
selector_base.ip_translation_spec = (
iptranslation.IPTranslationStrategySpec(
'maxmind', {'db_snapshots': ['2014-08-04']}))
for client_country in ('us', 'ca', 'uk', 'au'):
selector_copy = copy.copy(selector_base)
selector_copy.client_country = client_country
selectors_expected.append(selector_copy)
self.assertParsedSelectorsMatch(selectors_expected,
selector_file_contents)
def testSuccessfulParseOfValidv1_1FileWithAllOptionalFieldsDefined(self):
selector_file_contents = """{
"file_format_version": 1.1,
"duration": "30d",
"metrics": ["average_rtt"],
"ip_translation":{
"strategy":"maxmind",
"params":{
"db_snapshots":["2014-08-04"]
}
},
"sites": ["lga02"],
"client_providers": ["comcast"],
"client_countries": ["us"],
"start_times": ["2014-02-01T00:00:00Z"]
}"""
selector_expected = selector.Selector()
selector_expected.start_time = utils.make_datetime_utc_aware(
datetime.datetime(2014, 2, 1))
selector_expected.duration = 30 * 24 * 60 * 60
selector_expected.metric = 'average_rtt'
selector_expected.ip_translation_spec = (
iptranslation.IPTranslationStrategySpec(
'maxmind', {'db_snapshots': ['2014-08-04']}))
selector_expected.site = 'lga02'
selector_expected.client_provider = 'comcast'
selector_expected.client_country = 'us'
self.assertParsedSingleSelectorMatches(selector_expected,
selector_file_contents)
def testValidSingleSubsetAllMetric(self):
selector_file_contents = """{
"file_format_version": 1,
"duration": "30d",
"metric":"all",
"ip_translation":{
"strategy":"maxmind",
"params":{
"db_snapshots":["2014-08-04"]
}
},
"subsets":[
{
"site":"lga02",
"client_provider":"comcast",
"start_time":"2014-02-01T00:00:00Z"
}
]
}"""
selector_base = selector.Selector()
selector_base.start_time = utils.make_datetime_utc_aware(
datetime.datetime(2014, 2, 1))
selector_base.duration = 30 * 24 * 60 * 60
selector_base.metric = 'average_rtt'
selector_base.ip_translation_spec = (
iptranslation.IPTranslationStrategySpec(
'maxmind', {'db_snapshots': ['2014-08-04']}))
selector_base.site_name = 'lga02'
selector_base.client_provider = 'comcast'
selectors_expected = []
# TODO(mtlynch): Need to fix this test so that changing the order of the
# expected_metrics list doesn't cause the test to fail.
expected_metrics = ('minimum_rtt', 'upload_throughput',
'packet_retransmit_rate', 'download_throughput',
'average_rtt')
for metric in expected_metrics:
selector_copy = copy.copy(selector_base)
selector_copy.metric = metric
selectors_expected.append(selector_copy)
# The 'all' metric should expand to selectors for every supported metric.
self.assertParsedSelectorsMatch(selectors_expected,
selector_file_contents)
def testValidInput_v1dot1_Complex(self):
selector_file_contents = """{
"file_format_version": 1.1,
"duration": "30d",
"metrics": ["minimum_rtt", "download_throughput", "average_rtt"],
"ip_translation":{
"strategy":"maxmind",
"params":{
"db_snapshots":["2014-08-04"]
}
},
"sites": ["lga01", "lga02"],
"client_providers": ["comcast", "verizon"],
"start_times": ["2014-02-01T00:00:00Z"]
}"""
selectors_expected = []
selector_base = selector.Selector()
selector_base.start_time = utils.make_datetime_utc_aware(
datetime.datetime(2014, 2, 1))
selector_base.duration = 30 * 24 * 60 * 60
selector_base.ip_translation_spec = (
iptranslation.IPTranslationStrategySpec(
'maxmind', {'db_snapshots': ['2014-08-04']}))
sites = ['lga01', 'lga02']
client_providers = ['comcast', 'verizon']
metrics = ['minimum_rtt', 'download_throughput', 'average_rtt']
for client_provider, site, metric in itertools.product(
client_providers, sites, metrics):
selector_copy = copy.copy(selector_base)
selector_copy.metric = metric
selector_copy.client_provider = client_provider
selector_copy.site = site
selectors_expected.append(selector_copy)
self.assertParsedSelectorsMatch(selectors_expected,
selector_file_contents)
