def test_get_cells_no_response(): account = test_account_name() config = create_core_test_config({ "login": { "username": account }, "mapping": { "cell_radius": 500 } }) api_wrapper = create_mock_api_wrapper(config) google_maps = Mock(spec=Client) logger = Mock() logger.log = Mock(return_value=None) mapper = Mapper(config, api_wrapper, google_maps, logger) api_wrapper.set_position(51.5044524, -0.0752479, 10) pgo = api_wrapper.get_api() pgo.set_response("get_map_objects", {}) api_wrapper.call = Mock(return_value=None) # Clean up any old location logs if os.path.isfile('data/last-location-' + account + '.json'): os.unlink('data/last-location-' + account + '.json') cells = mapper.get_cells(51.5044524, -0.0752479) assert len(cells) == 0
def test_get_cells(self): account = test_account_name() config = create_core_test_config({ "debug": True, "login": { "username": account, }, "mapping": { "cell_radius": 500 } }) api_wrapper = create_mock_api_wrapper(config) google_maps = Mock(spec=Client) logger = Mock() logger.log = Mock(return_value=None) mapper = Mapper(config, api_wrapper, google_maps, logger) api_wrapper.set_position(51.5044524, -0.0752479, 10) pgo = api_wrapper.get_api() pgo.set_response( "get_map_objects", { "map_cells": [ self._create_map_cell(1), self._create_map_cell(2), self._create_map_cell(3), self._create_map_cell(4), self._create_map_cell(5) ] }) # Clean up any old location logs if os.path.isfile('data/last-location-' + account + '.json'): os.unlink('data/last-location-' + account + '.json') cells = mapper.get_cells(51.5044524, -0.0752479) assert len(cells) == 5 assert bool( os.path.isfile('data/last-location-' + account + '.json')) is True with open('data/last-location-' + account + '.json') as data_file: data = json.load(data_file) assert data["lat"] == 51.5044524 assert data["lng"] == -0.0752479 os.unlink('data/last-location-' + account + '.json')