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)
Пример #2
0
 def test_get_instance__not_cached_without_bot(self):
     with self.assertRaises(StrangerSenderServiceError):
         StrangerSenderService.get_instance()
Пример #3
0
 def test_get_instance__not_cached_with_bot(self):
     self.assertIsInstance(StrangerSenderService.get_instance(self.bot),
                           StrangerSenderService)
Пример #4
0
 def test_get_instance__cached(self):
     StrangerSenderService._instance = self.stranger_sender_service
     self.assertEqual(StrangerSenderService.get_instance(),
                      self.stranger_sender_service)
Пример #5
0
 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
Пример #6
0
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