Merge "Add get fake aggregate test"

1 files changed, 18 insertions, 0 deletions

diff --git a/vio/vio/tests/test_fake_nova_view.py b/vio/vio/tests/test_fake_nova_view.py
index e01e26e..cb63e56 100644
--- a/vio/vio/tests/test_fake_nova_view.py
+++ b/vio/vio/tests/test_fake_nova_view.py
@@ -68,3 +68,21 @@ class TestFakeNovaHypervisors(unittest.TestCase):
         }
         resp = self.view.get(req, "abcd")
         self.assertEqual(200, resp.status_code)
+
+
+class TestFakeNovaAggregate(unittest.TestCase):
+
+    def setUp(self):
+        self.view = views.FakeNovaAggregate()
+
+    @mock.patch.object(fakeResponse, "get_osaggregates")
+    def test_get_server(self, mock_get_osaggregates):
+        req = mock.Mock()
+        req.META = {
+            "HTTP_X_AUTH_TOKEN": Token
+        }
+        mock_get_osaggregates.return_value = {
+            "osaggregates": "1234abcd"
+        }
+        resp = self.view.get(req, "abcd")
+        self.assertEqual(200, resp.status_code)