class TestStrangerSenderService(unittest.TestCase):
@patch('randtalkbot.stranger_sender_service.StrangerSender')
def setUp(self, stranger_sender_cls_mock):
self.bot = Mock()
self.stranger_sender_service = StrangerSenderService(self.bot)
# Cached instance should be cleared for each test.
StrangerSenderService._instance = None
def test_init(self):
self.assertEqual(self.stranger_sender_service._bot, self.bot)
def test_get_instance__cached(self):
StrangerSenderService._instance = self.stranger_sender_service
self.assertEqual(StrangerSenderService.get_instance(), self.stranger_sender_service)
def test_get_instance__not_cached_with_bot(self):
self.assertIsInstance(StrangerSenderService.get_instance(self.bot), StrangerSenderService)
def test_get_instance__not_cached_without_bot(self):
with self.assertRaises(StrangerSenderServiceError):
StrangerSenderService.get_instance()
def test_get_cache_size(self):
self.assertEqual(self.stranger_sender_service.get_cache_size(), 0)
self.stranger_sender_service._stranger_senders = {
0: 0,
1: 1,
}
self.assertEqual(self.stranger_sender_service.get_cache_size(), 2)
@patch('randtalkbot.stranger_sender_service.StrangerSender', create_autospec(StrangerSender))
def test_get_or_create_stranger_sender__cached(self):
from randtalkbot.stranger_sender_service import StrangerSender
stranger_sender = Mock()
stranger = Mock()
stranger.telegram_id = 31416
self.stranger_sender_service._stranger_senders[31416] = stranger_sender
self.assertEqual(
self.stranger_sender_service.get_or_create_stranger_sender(stranger),
stranger_sender,
)
self.assertFalse(StrangerSender.called)
@patch('randtalkbot.stranger_sender_service.StrangerSender', create_autospec(StrangerSender))
def test_get_or_create_stranger_sender__not_cached(self):
from randtalkbot.stranger_sender_service import StrangerSender
stranger_sender = StrangerSender.return_value
stranger = Mock()
stranger.telegram_id = 31416
self.assertEqual(
self.stranger_sender_service.get_or_create_stranger_sender(stranger),
stranger_sender,
)
StrangerSender.assert_called_once_with(self.bot, stranger)
self.assertEqual(self.stranger_sender_service._stranger_senders[31416], stranger_sender)
def test_get_instance__not_cached_without_bot(self):
with self.assertRaises(StrangerSenderServiceError):
StrangerSenderService.get_instance()
def test_get_instance__not_cached_with_bot(self):
self.assertIsInstance(StrangerSenderService.get_instance(self.bot),
StrangerSenderService)
def test_get_instance__cached(self):
StrangerSenderService._instance = self.stranger_sender_service
self.assertEqual(StrangerSenderService.get_instance(),
self.stranger_sender_service)
def setUp(self, stranger_sender_cls_mock):
self.bot = Mock()
self.stranger_sender_service = StrangerSenderService(self.bot)
# Cached instance should be cleared for each test.
StrangerSenderService._instance = None
class TestStrangerSenderService(unittest.TestCase):
@patch('randtalkbot.stranger_sender_service.StrangerSender')
def setUp(self, stranger_sender_cls_mock):
self.bot = Mock()
self.stranger_sender_service = StrangerSenderService(self.bot)
# Cached instance should be cleared for each test.
StrangerSenderService._instance = None
def test_init(self):
self.assertEqual(self.stranger_sender_service._bot, self.bot)
def test_get_instance__cached(self):
StrangerSenderService._instance = self.stranger_sender_service
self.assertEqual(StrangerSenderService.get_instance(),
self.stranger_sender_service)
def test_get_instance__not_cached_with_bot(self):
self.assertIsInstance(StrangerSenderService.get_instance(self.bot),
StrangerSenderService)
def test_get_instance__not_cached_without_bot(self):
with self.assertRaises(StrangerSenderServiceError):
StrangerSenderService.get_instance()
def test_get_cache_size(self):
self.assertEqual(self.stranger_sender_service.get_cache_size(), 0)
self.stranger_sender_service._stranger_senders = {
0: 0,
1: 1,
}
self.assertEqual(self.stranger_sender_service.get_cache_size(), 2)
@patch('randtalkbot.stranger_sender_service.StrangerSender',
create_autospec(StrangerSender))
def test_get_or_create_stranger_sender__cached(self):
from randtalkbot.stranger_sender_service import StrangerSender
stranger_sender = Mock()
stranger = Mock()
stranger.telegram_id = 31416
self.stranger_sender_service._stranger_senders[31416] = stranger_sender
self.assertEqual(
self.stranger_sender_service.get_or_create_stranger_sender(
stranger),
stranger_sender,
)
self.assertFalse(StrangerSender.called)
@patch('randtalkbot.stranger_sender_service.StrangerSender',
create_autospec(StrangerSender))
def test_get_or_create_stranger_sender__not_cached(self):
from randtalkbot.stranger_sender_service import StrangerSender
stranger_sender = StrangerSender.return_value
stranger = Mock()
stranger.telegram_id = 31416
self.assertEqual(
self.stranger_sender_service.get_or_create_stranger_sender(
stranger),
stranger_sender,
)
StrangerSender.assert_called_once_with(self.bot, stranger)
self.assertEqual(self.stranger_sender_service._stranger_senders[31416],
stranger_sender)
def test_get_instance__not_cached_without_bot(self):
with self.assertRaises(StrangerSenderServiceError):
StrangerSenderService.get_instance()
def test_get_instance__not_cached_with_bot(self):
self.assertIsInstance(StrangerSenderService.get_instance(self.bot), StrangerSenderService)
def test_get_instance__cached(self):
StrangerSenderService._instance = self.stranger_sender_service
self.assertEqual(StrangerSenderService.get_instance(), self.stranger_sender_service)
def setUp(self, stranger_sender_cls_mock):
self.bot = Mock()
self.stranger_sender_service = StrangerSenderService(self.bot)
# Cached instance should be cleared for each test.
StrangerSenderService._instance = None