Add a version argument to traits DB API

We need to save the Trait object version to the DB when creating traits.

Change-Id: I2c43c27455de6e7017477b0f12b18873c66455ad
Partial-Bug: #1722194

1 files changed, 14 insertions, 1 deletions

diff --git a/ironic/tests/unit/db/utils.py b/ironic/tests/unit/db/utils.py
index 946a4e3d6..71aa3d098 100644
--- a/ironic/tests/unit/db/utils.py
+++ b/ironic/tests/unit/db/utils.py
@@ -513,4 +513,17 @@ def create_test_node_trait(**kw):
     """
     trait = get_test_node_trait(**kw)
     dbapi = db_api.get_instance()
-    return dbapi.add_node_trait(trait['node_id'], trait['trait'])
+    return dbapi.add_node_trait(trait['node_id'], trait['trait'],
+                                trait['version'])
+
+
+def create_test_node_traits(traits, **kw):
+    """Create test node trait entries in DB and return NodeTrait DB objects.
+
+    Function to be used to create test NodeTrait objects in the database.
+
+    :param traits: a list of Strings; traits to create.
+    :param kw: kwargs with overriding values for trait's attributes.
+    :returns: a list of test NodeTrait DB objects.
+    """
+    return [create_test_node_trait(trait=trait, **kw) for trait in traits]