Add missing tests for nova.db.api.agent_build_* methods

blueprint db-api-tests

Change-Id: If8f8a3f02bc0ea73b9174caad60107c7d87f62c0

1 files changed, 49 insertions, 0 deletions

diff --git a/nova/tests/test_db_api.py b/nova/tests/test_db_api.py
index 11165e168..0ec1d2c06 100644
--- a/nova/tests/test_db_api.py
+++ b/nova/tests/test_db_api.py
@@ -3724,6 +3724,55 @@ class BlockDeviceMappingTestCase(test.TestCase):
         self.assertEqual(bdms[0]['device_name'], 'fake2')
 
 
+class AgentBuildTestCase(test.TestCase, ModelsObjectComparatorMixin):
+
+    """Tests for db.api.agent_build_* methods."""
+
+    def setUp(self):
+        super(AgentBuildTestCase, self).setUp()
+        self.ctxt = context.get_admin_context()
+
+    def test_agent_build_create_and_get_all(self):
+        self.assertEqual(0, len(db.agent_build_get_all(self.ctxt)))
+        agent_build = db.agent_build_create(self.ctxt, {'os': 'GNU/HURD'})
+        all_agent_builds = db.agent_build_get_all(self.ctxt)
+        self.assertEqual(1, len(all_agent_builds))
+        self._assertEqualObjects(agent_build, all_agent_builds[0])
+
+    def test_agent_build_get_by_triple(self):
+        agent_build = db.agent_build_create(self.ctxt, {'hypervisor': 'kvm',
+                                'os': 'FreeBSD', 'architecture': 'x86_64'})
+        self.assertIsNone(db.agent_build_get_by_triple(self.ctxt, 'kvm',
+                                                        'FreeBSD', 'i386'))
+        self._assertEqualObjects(agent_build, db.agent_build_get_by_triple(
+                                    self.ctxt, 'kvm', 'FreeBSD', 'x86_64'))
+
+    def test_agent_build_destroy(self):
+        agent_build = db.agent_build_create(self.ctxt, {})
+        self.assertEqual(1, len(db.agent_build_get_all(self.ctxt)))
+        db.agent_build_destroy(self.ctxt, agent_build.id)
+        self.assertEqual(0, len(db.agent_build_get_all(self.ctxt)))
+
+    def test_agent_build_update(self):
+        agent_build = db.agent_build_create(self.ctxt, {'os': 'HaikuOS'})
+        db.agent_build_update(self.ctxt, agent_build.id, {'os': 'ReactOS'})
+        self.assertEqual('ReactOS', db.agent_build_get_all(self.ctxt)[0].os)
+
+    @test.testtools.skip("bug 1181967")
+    def test_agent_build_destroy_destroyed(self):
+        agent_build = db.agent_build_create(self.ctxt, {})
+        db.agent_build_destroy(self.ctxt, agent_build.id)
+        self.assertRaises(exception.AgentBuildNotFound,
+            db.agent_build_destroy, self.ctxt, agent_build.id)
+
+    @test.testtools.skip("bug 1181967")
+    def test_agent_build_update_destroyed(self):
+        agent_build = db.agent_build_create(self.ctxt, {'os': 'HaikuOS'})
+        db.agent_build_destroy(self.ctxt, agent_build.id)
+        self.assertRaises(exception.AgentBuildNotFound,
+            db.agent_build_update, self.ctxt, agent_build.id, {'os': 'OS/2'})
+
+
 class VirtualInterfaceTestCase(test.TestCase, ModelsObjectComparatorMixin):
     def setUp(self):
         super(VirtualInterfaceTestCase, self).setUp()