| """Tests for the self-modification module.""" |
|
|
| import os |
| import tempfile |
| from pathlib import Path |
|
|
| import pytest |
|
|
| from infj_bot.core.self_modify import ( |
| SelfModification, |
| IMPROVEMENT_AREAS, |
| MAX_PENDING_PROPOSALS, |
| ) |
|
|
|
|
| @pytest.fixture |
| def tmp_db(): |
| fd, path = tempfile.mkstemp(suffix=".db") |
| os.close(fd) |
| yield Path(path) |
| os.unlink(path) |
|
|
|
|
| class TestSelfModification: |
| def test_init_creates_db(self, tmp_db): |
| SelfModification(db_path=tmp_db) |
| assert tmp_db.exists() |
|
|
| def test_propose_improvement(self, tmp_db): |
| sm = SelfModification(db_path=tmp_db) |
| p = sm.propose_improvement() |
| assert p is not None |
| assert "id" in p |
| assert p["status"] == "pending" |
| assert p["area"] in IMPROVEMENT_AREAS |
| assert len(p["description"]) > 10 |
| assert "observed_need" in p |
| assert len(sm.proposals) == 1 |
|
|
| def test_propose_improvement_specific_area(self, tmp_db): |
| sm = SelfModification(db_path=tmp_db) |
| p = sm.propose_improvement(area="creativity") |
| assert p["area"] == "creativity" |
|
|
| def test_approve_reject_apply(self, tmp_db): |
| sm = SelfModification(db_path=tmp_db) |
| p = sm.propose_improvement() |
| pid = p["id"] |
|
|
| sm.approve_proposal(pid) |
| sm2 = SelfModification(db_path=tmp_db) |
| loaded = [x for x in sm2.proposals if x["id"] == pid] |
| assert loaded[0]["status"] == "approved" |
|
|
| sm.reject_proposal(pid) |
| sm3 = SelfModification(db_path=tmp_db) |
| loaded = [x for x in sm3.proposals if x["id"] == pid] |
| assert loaded[0]["status"] == "rejected" |
|
|
| sm.apply_proposal(pid) |
| sm4 = SelfModification(db_path=tmp_db) |
| loaded = [x for x in sm4.proposals if x["id"] == pid] |
| assert loaded[0]["status"] == "applied" |
|
|
| def test_get_pending_proposals(self, tmp_db): |
| sm = SelfModification(db_path=tmp_db) |
| assert sm.get_pending_proposals() == [] |
| p = sm.propose_improvement() |
| assert len(sm.get_pending_proposals()) == 1 |
| sm.approve_proposal(p["id"]) |
| |
| sm2 = SelfModification(db_path=tmp_db) |
| assert len(sm2.get_pending_proposals()) == 0 |
|
|
| def test_max_pending_limit(self, tmp_db): |
| sm = SelfModification(db_path=tmp_db) |
| for _ in range(MAX_PENDING_PROPOSALS + 2): |
| sm.propose_improvement() |
| assert sm._count_pending() <= MAX_PENDING_PROPOSALS |
|
|
| def test_format_self_modify_prompt(self, tmp_db): |
| sm = SelfModification(db_path=tmp_db) |
| assert sm.format_self_modify_prompt() == "" |
| sm.propose_improvement() |
| prompt = sm.format_self_modify_prompt() |
| assert "IMPROVEMENTS" in prompt |
| assert "pending" not in prompt |
|
|
| def test_format_self_modify_prompt_empty(self, tmp_db): |
| sm = SelfModification(db_path=tmp_db) |
| assert sm.format_self_modify_prompt() == "" |
|
|
