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_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_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_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)
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)
