def test_execute_ok_refresh_with_primkey(self): conn = None o = None fetch_result = None try: # create test db and insert dummy data os.makedirs(self._db_dir) db_file = os.path.join(self._db_dir, "spam.db") conn = sqlite3.connect(db_file) conn.execute("create table spam_table (id, name, age);") # create dummy input data o = StorageIO() o.save({"id": 1, "name": 1, "age": 1}) instance = SqliteCreation() setattr(instance, "dbname", db_file) setattr(instance, "tblname", "spam_table") setattr(instance, "io", "output") setattr(instance, "refresh", True) instance.execute() cur = conn.execute("select id, name, age from spam_table;") fetch_result = cur.fetchall() finally: shutil.rmtree(self._db_dir) conn.close() assert fetch_result == [("1", "1", "1")]
def __init__(self): self._s = StorageIO() self._step = None self._symbol = None self._parallel = None self._io = None self._logger = None
def __init__(self): self._logger = LisboaLog.get_logger(__name__) self._s = StorageIO() self._step = None self._symbol = None self._parallel = None self._io = None
def test_save_ng(self): s = StorageIO() s.save("spam") assert os.path.exists(self.__tmp_invalid_cache_file) is False
def test_save_ok(self): s = StorageIO() s.save(["spam"]) assert os.path.exists(self.__tmp_valid_cache_file) is True