44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
from unittest.mock import Mock, patch
|
|
|
|
from django.test import TestCase
|
|
|
|
from core.clients import transport
|
|
|
|
|
|
class SignalUnlinkFallbackTests(TestCase):
|
|
@patch("core.clients.transport._wipe_signal_cli_local_state")
|
|
@patch("requests.delete")
|
|
def test_signal_unlink_uses_rest_delete_when_available(
|
|
self,
|
|
mock_delete,
|
|
mock_wipe,
|
|
):
|
|
ok_response = Mock()
|
|
ok_response.ok = True
|
|
mock_delete.return_value = ok_response
|
|
|
|
result = transport.unlink_account("signal", "+447700900000")
|
|
|
|
self.assertTrue(result)
|
|
self.assertTrue(mock_delete.called)
|
|
mock_wipe.assert_not_called()
|
|
|
|
@patch("core.clients.transport._wipe_signal_cli_local_state")
|
|
@patch("requests.delete")
|
|
def test_signal_unlink_falls_back_to_local_wipe(
|
|
self,
|
|
mock_delete,
|
|
mock_wipe,
|
|
):
|
|
bad_response = Mock()
|
|
bad_response.ok = False
|
|
mock_delete.return_value = bad_response
|
|
mock_wipe.return_value = True
|
|
|
|
result = transport.unlink_account("signal", "+447700900000")
|
|
|
|
self.assertTrue(result)
|
|
self.assertEqual(2, mock_delete.call_count)
|
|
mock_wipe.assert_called_once()
|
|
|