def test_is_open(self): url = 'test-url' sender = Sender(url) # closed self.assertFalse(sender.is_open()) # open sender.session = Mock() self.assertTrue(sender.is_open())
def test_open_already(self): url = 'test-url' # test sender = Sender(url) sender.is_open = Mock(return_value=True) sender.open() # validation self.assertFalse(sender.connection.open.called)
def test_open(self, connection): url = 'test-url' # test sender = Sender(url) sender.is_open = Mock(return_value=False) sender.open() # validation connection.return_value.open.assert_called_once_with() connection.return_value.session.assert_called_once_with() self.assertEqual(sender.session, connection.return_value.session.return_value)
def test_close(self): connection = Mock() session = Mock(connection=connection) # test sender = Sender(None) sender.connection = connection sender.session = session sender.is_open = Mock(return_value=True) sender.close() # validation session.close.assert_called_once_with() self.assertFalse(connection.close.called)