def test_with_ip(self):
service = LocastService(self.config, Geo())
service._set_attrs_from_geo = set_attrs = MagicMock()
service._find_location()
set_attrs.assert_called_with(IP_URL)
def test_with_zipcode(self):
service = LocastService(self.config, Geo("90210"))
service._set_attrs_from_geo = set_attrs = MagicMock()
service._find_location()
url = f"{DMA_URL}/zip/90210"
set_attrs.assert_called_with(url)
def test_with_coords(self):
service = LocastService(self.config,
Geo(None, {
"longitude": 1.0,
"latitude": 2.0
}))
service._set_attrs_from_geo = set_attrs = MagicMock()
service._find_location()
url = f"{DMA_URL}/2.0/1.0"
set_attrs.assert_called_with(url)