class TestGeoServer(unittest.TestCase):
def setUp(self):
self.server = GeoServer(None, AsObj(**{
'GEOIP_PATH': 'data/',
'PORT': 0,
}))
def test_valid(self):
fake = FakeSocket('get 8.8.8.8\n')
self.server.handle(fake, None)
value = fake.getvalue().split('\n')[1]
resp = json.loads(value)
self.failUnless('success' in resp)
self.failUnlessEqual(resp['success']['country_code'], 'US')
def test_bad_command(self):
fake = FakeSocket('banana\n')
self.server.handle(fake, None)
value = fake.getvalue().split('\n')[1]
self.failUnless('error' in value)
def test_bad_request(self):
fake = FakeSocket('\n')
self.server.handle(fake, None)
value = fake.getvalue().split('\n')[1]
self.failUnless('error' in value)
def test_empty_info(self):
fake = FakeSocket('get 127.0.0.1\n')
self.server.handle(fake, None)
value = fake.getvalue().split('\n')[1]
self.failUnless('error' in value)
def setUp(self):
self.server = GeoServer(None, AsObj(**{
'GEOIP_PATH': 'data/',
'PORT': 0,
}))