55 lines
2.0 KiB
Python
55 lines
2.0 KiB
Python
import threading
|
|
import unittest
|
|
|
|
from services.config import config
|
|
from services.protocol.conversation import _remove_image_conversation_later
|
|
|
|
|
|
class FakeBackend:
|
|
def __init__(self) -> None:
|
|
self.called = threading.Event()
|
|
|
|
def delete_conversation(self, conversation_id: str) -> dict:
|
|
self.called.set()
|
|
return {}
|
|
|
|
|
|
class RemoveImageConversationGateTests(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
self._saved = dict(config.data)
|
|
|
|
def tearDown(self) -> None:
|
|
config.data = self._saved
|
|
|
|
def _removed(self, *, after_result: bool, always: bool, success: bool) -> bool:
|
|
config.data = dict(
|
|
self._saved,
|
|
image_remove_conversation_after_result=after_result,
|
|
image_remove_conversation_always=always,
|
|
)
|
|
backend = FakeBackend()
|
|
_remove_image_conversation_later(backend, "conv-1", success=success)
|
|
return backend.called.wait(2.0)
|
|
|
|
def test_both_off_never_removes(self) -> None:
|
|
self.assertFalse(self._removed(after_result=False, always=False, success=True))
|
|
self.assertFalse(self._removed(after_result=False, always=False, success=False))
|
|
|
|
def test_after_result_only_removes_on_success(self) -> None:
|
|
self.assertTrue(self._removed(after_result=True, always=False, success=True))
|
|
self.assertFalse(self._removed(after_result=True, always=False, success=False))
|
|
|
|
def test_always_removes_regardless_of_success(self) -> None:
|
|
self.assertTrue(self._removed(after_result=False, always=True, success=True))
|
|
self.assertTrue(self._removed(after_result=False, always=True, success=False))
|
|
|
|
def test_empty_conversation_id_is_noop(self) -> None:
|
|
config.data = dict(self._saved, image_remove_conversation_always=True)
|
|
backend = FakeBackend()
|
|
_remove_image_conversation_later(backend, "", success=False)
|
|
self.assertFalse(backend.called.wait(0.2))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|