51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
import unittest
|
|
import json
|
|
import urllib.request
|
|
|
|
import util.location
|
|
|
|
|
|
class location(unittest.TestCase):
|
|
def setUp(self):
|
|
patcher = unittest.mock.patch("util.location.urllib.request")
|
|
self.addCleanup(patcher.stop)
|
|
self.request = patcher.start()
|
|
util.location.reset()
|
|
|
|
self.primary = {
|
|
"country": "Middle Earth",
|
|
"longitude": "10.0",
|
|
"latitude": "20.5",
|
|
"ip": "127.0.0.1",
|
|
}
|
|
self.secondary = {
|
|
"country_name": "Rivia",
|
|
"longitude": "-10.0",
|
|
"latitude": "-23",
|
|
"ip": "127.0.0.6",
|
|
}
|
|
|
|
def test_primary_provider(self):
|
|
self.request.urlopen.return_value.read.return_value = json.dumps(self.primary)
|
|
util.location.country()
|
|
self.assertEqual(self.primary["country"], util.location.country())
|
|
self.assertEqual(
|
|
(self.primary["latitude"], self.primary["longitude"]),
|
|
util.location.coordinates(),
|
|
)
|
|
self.assertEqual(self.primary["ip"], util.location.public_ip())
|
|
|
|
def test_secondary_provider(self):
|
|
urlopen = unittest.mock.MagicMock()
|
|
urlopen.read.return_value = json.dumps(self.secondary)
|
|
self.request.urlopen.side_effect = [RuntimeError(), urlopen]
|
|
|
|
self.assertEqual(self.secondary["country_name"], util.location.country())
|
|
self.assertEqual(
|
|
(self.secondary["latitude"], self.secondary["longitude"]),
|
|
util.location.coordinates(),
|
|
)
|
|
self.assertEqual(self.secondary["ip"], util.location.public_ip())
|
|
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|