def test_missing_info(self):
settings = {'is_missing_info': False, 'max_dist': 500}
stop_filter = Filters.StopFilter('missing_info_filter', settings)
stop_event = Events.StopEvent(generate_stop({}))
self.assertFalse(stop_filter.check_event(stop_event))
for i in [0, 500]:
stop_event.distance = i
self.assertTrue(stop_filter.check_event(stop_event))
stop_event.distance = 'Unknown'
self.assertFalse(stop_filter.check_event(stop_event))
def gen_event(self, values):
""" Generate a generic stop, overriding with an specific values. """
settings = {
"pokestop_id": 0,
"enabled": "True",
"latitude": 37.7876146,
"longitude": -122.390624,
"last_modified_time": 1572241600,
"lure_expiration": 1572241600,
"active_fort_modifier": 0
}
settings.update(values)
return Events.StopEvent(settings)
def test_distance(self):
stop_event = Events.StopEvent(generate_stop({}))
settings = {'min_dist': 5, 'max_dist': 2000}
stop_filter = Filters.StopFilter('distance_filter', settings)
for i in [5, 2000, 1000]:
stop_event.distance = i
self.assertTrue(stop_filter.check_event(stop_event))
settings2 = {'min_dist': 100, 'max_dist': 5000}
stop_filter2 = Filters.StopFilter('distance_filter_2', settings2)
for i in [99, 5001, 9999]:
stop_event.distance = i
self.assertFalse(stop_filter2.check_event(stop_event))
def create_event(items_to_change):
return Events.StopEvent(generate_stop(items_to_change))
