def setUp(self):
"""Sets up the test environment."""
# The TileService to use for testing:
self.service = TileService()
# It's an error to register the same service twice:
if apiproxy_stub_map.apiproxy.GetStub('datastore_v3') is not None:
return;
# Let's us run unit tests without running the dev server:
os.environ['TZ'] = 'UTC'
time.tzset()
os.environ['SERVER_SOFTWARE'] = 'Development via nose'
os.environ['SERVER_NAME'] = 'Foo'
os.environ['SERVER_PORT'] = '8080'
os.environ['APPLICATION_ID'] = APP_ID
os.environ['USER_EMAIL'] = '*****@*****.**'
os.environ['CURRENT_VERSION_ID'] = 'testing-version'
ds_stub = DatastoreFileStub(APP_ID, None, None)
apiproxy_stub_map.apiproxy.RegisterStub('datastore_v3', ds_stub)
class TileServiceTest(unittest.TestCase):
"""Unit tests for TileService class."""
def setUp(self):
"""Sets up the test environment."""
# The TileService to use for testing:
self.service = TileService()
# It's an error to register the same service twice:
if apiproxy_stub_map.apiproxy.GetStub('datastore_v3') is not None:
return;
# Let's us run unit tests without running the dev server:
os.environ['TZ'] = 'UTC'
time.tzset()
os.environ['SERVER_SOFTWARE'] = 'Development via nose'
os.environ['SERVER_NAME'] = 'Foo'
os.environ['SERVER_PORT'] = '8080'
os.environ['APPLICATION_ID'] = APP_ID
os.environ['USER_EMAIL'] = '*****@*****.**'
os.environ['CURRENT_VERSION_ID'] = 'testing-version'
ds_stub = DatastoreFileStub(APP_ID, None, None)
apiproxy_stub_map.apiproxy.RegisterStub('datastore_v3', ds_stub)
def test_put_tile(self):
self.assertEqual(None, self.service.put_tile(None))
self.assertEqual(None, self.service.put_tile(TileUpdate()))
key_name = '00/021/pa'
key = db.Key.from_path(TileService.TILE_KIND, key_name)
tile = Tile(key=key)
self.assertEqual(key, self.service.put_tile(tile))
def test_put_tile_update(self):
self.assertEqual(None, self.service.put_tile_update(None))
self.assertEqual(None, self.service.put_tile_update(Tile()))
# Test invalid zoom
key_name = '00/xxx/pa'
key = db.Key.from_path(TileService.TILE_KIND, key_name)
tile = Tile(key=key)
db.put(tile)
tile_update_key = self.service.put_tile_update(tile)
self.assertEqual(tile_update_key, None)
# Test valid put
key_name = '00/021/pa'
key = db.Key.from_path(TileService.TILE_KIND, key_name)
tile = Tile(key=key)
db.put(tile)
tile_update_key = self.service.put_tile_update(tile)
self.assertNotEqual(tile_update_key, None)
self.assertEqual(tile_update_key.name(), key_name)
tile_update = db.get(tile_update_key)
self.assertNotEqual(tile_update, None)
self.assertEqual(tile_update.zoom, 3)
def test_zoom_from_key(self):
key_name = '00/021/pa'
self.assertEqual(None, self.service._zoom_from_key_name(None))
self.assertEqual(None, self.service._zoom_from_key_name('/'))
self.assertEqual(None, self.service._zoom_from_key_name('//'))
self.assertEqual(None, self.service._zoom_from_key_name('/x/y'))
self.assertEqual('021', self.service._zoom_from_key_name('00/021/pa'))
def test_is_tile(self):
self.assertFalse(self.service._is_tile(None))
self.assertFalse(self.service._is_tile(8))
self.assertFalse(self.service._is_tile('Tile'))
self.assertFalse(self.service._is_tile(TileUpdate()))
self.assertTrue(self.service._is_tile(Tile()))
def test_tile_from_url(self):
# Tests valid urls:
key_name = '00/021/pa'
for url in [u'/api/entity/00/021/pa.png', u'/00/021/pa.png', u'00/021/pa']:
entity = Tile(key=db.Key.from_path('Tile', key_name))
data = '0101-%s' % url
entity.band = db.Blob(str(data));
entity.put()
tile = self.service.tile_from_url(url)
self.assertTrue(tile is not None)
self.assertEqual(tile.key(), entity.key())
self.assertEqual(tile.band, data)
# Tests None:
url = None
tile = self.service.tile_from_url(url)
self.assertEqual(tile, None)
# Tests invalid urls:
for url in ['', 'foo', '/api/tile/foo/bar/baz.png']:
tile = self.service.tile_from_url(url)
self.assertEqual(tile, None)