def test_cleanup(self):
    rpc.NOTIFIER = mock.Mock()
    rpc.NOTIFICATION_TRANSPORT = mock.Mock()
    rpc.TRANSPORT = mock.Mock()
    trans_cleanup = mock.Mock()
    not_trans_cleanup = mock.Mock()
    rpc.TRANSPORT.cleanup = trans_cleanup
    rpc.NOTIFICATION_TRANSPORT.cleanup = not_trans_cleanup
    rpc.cleanup()
    trans_cleanup.assert_called_once_with()
    not_trans_cleanup.assert_called_once_with()
    self.assertIsNone(rpc.TRANSPORT)
    self.assertIsNone(rpc.NOTIFICATION_TRANSPORT)
    self.assertIsNone(rpc.NOTIFIER)
def test_cleanup(self):
    rpc.NOTIFIER = mock.Mock()
    rpc.NOTIFICATION_TRANSPORT = mock.Mock()
    rpc.TRANSPORT = mock.Mock()
    trans_cleanup = mock.Mock()
    not_trans_cleanup = mock.Mock()
    rpc.TRANSPORT.cleanup = trans_cleanup
    rpc.NOTIFICATION_TRANSPORT.cleanup = not_trans_cleanup
    rpc.cleanup()
    trans_cleanup.assert_called_once_with()
    not_trans_cleanup.assert_called_once_with()
    self.assertIsNone(rpc.TRANSPORT)
    self.assertIsNone(rpc.NOTIFICATION_TRANSPORT)
    self.assertIsNone(rpc.NOTIFIER)