def test_is_open(self): url = 'test-url' node = Mock(address='test') reader = Reader(node, url=url) # closed self.assertFalse(reader.is_open()) # open reader.receiver = Mock() self.assertTrue(reader.is_open())
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.receiver.assert_called_once_with(node.address) self.assertEqual(reader.receiver, reader.connection.receiver.return_value)
def test_close(self): connection = Mock() receiver = Mock() receiver.close.side_effect = KeyError node = Mock(address='test') # test reader = Reader(node, '') reader.connection = connection reader.receiver = receiver reader.is_open = Mock(return_value=True) reader.close() # validation receiver.close.assert_called_once_with() self.assertFalse(connection.close.called)