def test_close(self):
left = Mock()
right = Mock()
ct = ComposableTransport(left, right)
ct.close()
left.close.assert_called_once_with()
right.close.assert_called_once_with()
def test_close(self):
left = Mock()
right = Mock()
ct = ComposableTransport(left, right)
ct.close()
left.close.assert_called_once_with()
right.close.assert_called_once_with()
def test_close_with_callbacks(self):
left = Mock()
right = Mock()
ct = ComposableTransport(left, right)
cb = Mock()
ct.add_on_close_callback(cb)
self.assertEquals(ct._close_callbacks, [cb])
ct.close()
left.close.assert_called_once_with()
right.close.assert_called_once_with()
cb.assert_called_once_with(ct, 200, "Closed OK")
def test_close_with_callbacks(self):
left = Mock()
right = Mock()
ct = ComposableTransport(left, right)
cb = Mock()
ct.add_on_close_callback(cb)
self.assertEquals(ct._close_callbacks, [cb])
ct.close()
left.close.assert_called_once_with()
right.close.assert_called_once_with()
cb.assert_called_once_with(ct, 200, "Closed OK")
