예제 #1
0
    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
예제 #2
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')