def test_combine_worker_dbs():
session_folder = make_session_folder()
# make a random number of tempdbs with dummy data
img = new("RGB", (256, 256), "red")
data = img_to_buf(img, 'jpeg').read()
z = randint(2, 5)
for x in xrange(z):
TempDB(session_folder).insert_image_blob(x, 0, 0, Binary(data))
# confirm that combine_worker_dbs assimilates all tempdb's into gpkg
chdir(session_folder) # necessary to put gpkg in session_folder
gpkg = Geopackage("test.gpkg", 4326)
combine_worker_dbs(gpkg)
result = gpkg.execute("select count(*) from tiles;")
assert (result.fetchone())[0] == z
def test_assimilate_error(self):
session_folder = make_session_folder()
chdir(session_folder)
with Geopackage(session_folder, 3395) as gpkg:
with raises(IOError):
gpkg.assimilate("None")
remove(join(getcwd(), gpkg.file_path))
def test_assimilate_error(self):
session_folder = make_session_folder()
chdir(session_folder)
gpkg = Geopackage(session_folder, 3395)
remove(join(getcwd(), gpkg.file_path))
gpkg.execute("")
OE = None
try:
gpkg.assimilate("None")
except OperationalError as OE:
print "OperationalError should occur."
finally:
assert OE is not None and type(OE) == OperationalError
def make_gpkg():
filename = uuid4().hex + '.gpkg'
tmp_file = join(gettempdir(), filename)
return Geopackage(tmp_file, 4326)
def test_execute_return(self):
session_folder = make_session_folder()
chdir(session_folder)
gpkg = Geopackage(session_folder, 9804)
result = gpkg.execute("select count(*) from tiles;")
assert (result.fetchone())[0] == 0
def test_file_path(self):
filename = uuid4().hex + '.gpkg'
tmp_dir = gettempdir()
tmp_file = join(tmp_dir, filename)
gpkg = Geopackage(tmp_file, 3857)
assert gpkg.file_path == tmp_file
