Example #1
0
    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")]
Example #2
0
 def __init__(self):
     self._s = StorageIO()
     self._step = None
     self._symbol = None
     self._parallel = None
     self._io = None
     self._logger = None
Example #3
0
 def __init__(self):
     self._logger = LisboaLog.get_logger(__name__)
     self._s = StorageIO()
     self._step = None
     self._symbol = None
     self._parallel = None
     self._io = None
Example #4
0
 def test_save_ng(self):
     s = StorageIO()
     s.save("spam")
     assert os.path.exists(self.__tmp_invalid_cache_file) is False
Example #5
0
 def test_save_ok(self):
     s = StorageIO()
     s.save(["spam"])
     assert os.path.exists(self.__tmp_valid_cache_file) is True