Function test_histmanager_memory_fallback_reopens_db

tests/test_history.py:287–321

(hmmax3, tmp_path, caplog)

def test_histmanager_memory_fallback_reopens_db(hmmax3, tmp_path, caplog):
    hist_file = tmp_path / "history.sqlite"
    ip = get_ipython()
    hm1 = None
    hm2 = None
    lock = None
    try:
        hm1 = HistoryManager(shell=ip, hist_file=hist_file)
        hm1.end_session()
        hm1.save_thread.stop()
        hm1.db.close()

        lock = sqlite3.connect(hist_file)
        lock.execute("BEGIN IMMEDIATE")

        hm2 = HistoryManager(shell=ip, hist_file=hist_file)
        assert hm2.hist_file == ":memory:"
        assert hm2.db.execute("PRAGMA database_list").fetchall() == [(0, "main", "")]

        hm2.store_inputs(1, "a = 1")
        hm2.writeout_cache()
        assert list(hm2.get_tail(1, include_latest=True)) == [
            (hm2.session_number, 1, "a = 1")
        ]
    finally:
        if hm2 is not None:
            hm2.end_session()
            hm2.db.close()
        if lock is not None:
            lock.rollback()
            lock.close()
        hm1 = None
        hm2 = None
        caplog.clear()
        gc.collect()


def test_histmanager_thread_start_failure_uses_memory(