def test_open_already(self): url = 'test-url' node = Mock(address='test') # test reader = Reader(node, url) reader.is_open = Mock(return_value=True) reader.open() # validation self.assertFalse(reader.connection.open.called)
def test_open_already(self): url = 'test-url' node = Mock(address='test') # test reader = Reader(node, url) reader.is_open = Mock(return_value=True) reader.open() # validation self.assertFalse(reader.connection.open.called)
def test_open(self, connection): url = 'test-url' node = Mock(address='test') # test reader = Reader(node, url) reader.is_open = Mock(return_value=False) reader.open() # validation connection.return_value.open.assert_called_once_with() connection.return_value.session.assert_called_once_with() connection.return_value.session.return_value.receiver.assert_called_once_with(node.address) self.assertEqual(reader.session, connection.return_value.session.return_value) self.assertEqual(reader.receiver, reader.session.receiver.return_value)
def test_open(self, connection): url = 'test-url' node = Mock(address='test') # test reader = Reader(node, url) reader.is_open = Mock(return_value=False) reader.open() # validation connection.return_value.open.assert_called_once_with() connection.return_value.session.assert_called_once_with() connection.return_value.session.return_value.receiver.assert_called_once_with( node.address) self.assertEqual(reader.session, connection.return_value.session.return_value) self.assertEqual(reader.receiver, reader.session.receiver.return_value)