From fdd27860724cd57db6df059a97e98289f88ce6ac Mon Sep 17 00:00:00 2001
From: termie <github@anarkystic.com>
Date: Wed, 25 May 2011 15:42:24 -0700
Subject: add support to rpc for multicall

---
 nova/tests/test_rpc.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'nova/tests')

diff --git a/nova/tests/test_rpc.py b/nova/tests/test_rpc.py
index 44d7c91eb..92ddfcffc 100644
--- a/nova/tests/test_rpc.py
+++ b/nova/tests/test_rpc.py
@@ -49,6 +49,17 @@ class RpcTestCase(test.TestCase):
                                                  "args": {"value": value}})
         self.assertEqual(value, result)
 
+    def test_multicall_succeed_three_times(self):
+        """Get a value through rpc call"""
+        value = 42
+        result = rpc.multicall(self.context,
+                              'test',
+                              {"method": "echo_three_times",
+                               "args": {"value": value}})
+
+        for x in result:
+            self.assertEqual(value, x)
+
     def test_context_passed(self):
         """Makes sure a context is passed through rpc call"""
         value = 42
@@ -126,6 +137,12 @@ class TestReceiver(object):
         LOG.debug(_("Received %s"), context)
         return context.to_dict()
 
+    @staticmethod
+    def echo_three_times(context, value):
+        context.reply(value)
+        context.reply(value)
+        context.reply(value)
+
     @staticmethod
     def fail(context, value):
         """Raises an exception with the value sent in"""
-- 
cgit