File size: 2,007 Bytes
49eeebd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | 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()
|