From: Angus Salkeld <asalkeld@redhat.com>
Date: Fri, 8 Mar 2013 12:19:09 +0000 (+1100)
Subject: Add Quantum floating ip assoc tests to improve coverage.
X-Git-Tag: 2014.1~787^2
X-Git-Url: https://review.fuel-infra.org/gitweb?a=commitdiff_plain;h=e657f054949e8867ac91475d5a6882484f275d7a;p=openstack-build%2Fheat-build.git

Add Quantum floating ip assoc tests to improve coverage.

bug 1152445

Change-Id: I7540b233f665f59bdecb171aa353798d2ce310b5
---

diff --git a/heat/tests/test_quantum.py b/heat/tests/test_quantum.py
index 15f5ad0b..9b8d6514 100644
--- a/heat/tests/test_quantum.py
+++ b/heat/tests/test_quantum.py
@@ -201,6 +201,7 @@ class QuantumFloatingIPTest(unittest.TestCase):
     def setUp(self):
         self.m = mox.Mox()
         self.m.StubOutWithMock(floatingip.FloatingIP, 'quantum')
+        self.m.StubOutWithMock(floatingip.FloatingIPAssociation, 'quantum')
         self.m.StubOutWithMock(port.Port, 'quantum')
 
     def tearDown(self):
@@ -293,3 +294,43 @@ class QuantumFloatingIPTest(unittest.TestCase):
                          p.handle_update({}))
 
         self.m.VerifyAll()
+
+    def test_floatip_port(self):
+        fq = FakeQuantum()
+        floatingip.FloatingIP.quantum().MultipleTimes().AndReturn(fq)
+        floatingip.FloatingIPAssociation.quantum().\
+            MultipleTimes().AndReturn(fq)
+        port.Port.quantum().MultipleTimes().AndReturn(fq)
+
+        self.m.ReplayAll()
+
+        t = self.load_template('Quantum_floating')
+        stack = self.parse_stack(t)
+
+        fip = stack['floating_ip']
+        self.assertEqual(None, fip.create())
+        self.assertEqual(floatingip.FloatingIP.CREATE_COMPLETE, fip.state)
+
+        p = stack['port_floating']
+        self.assertEqual(None, p.create())
+        self.assertEqual(port.Port.CREATE_COMPLETE, p.state)
+
+        fipa = stack['floating_ip_assoc']
+        self.assertEqual(None, fipa.create())
+        self.assertEqual(floatingip.FloatingIPAssociation.CREATE_COMPLETE,
+                         fipa.state)
+
+        fipa.validate()
+
+        fipa_id = fipa.FnGetRefId()
+        fip_id = fip.FnGetRefId()
+        port_id = p.FnGetRefId()
+        self.assertEqual('%s:%s' % (fip_id, port_id), fipa_id)
+        self.assertEqual(floatingip.FloatingIP.UPDATE_REPLACE,
+                         fipa.handle_update({}))
+
+        self.assertEqual(fipa.delete(), None)
+        self.assertEqual(p.delete(), None)
+        self.assertEqual(fip.delete(), None)
+
+        self.m.VerifyAll()