1 files changed, 70 insertions, 0 deletions
diff --git a/cyborg_enhancement/mitaka_version/cyborg/cyborg/tests/unit/fake_deployable.py b/cyborg_enhancement/mitaka_version/cyborg/cyborg/tests/unit/fake_deployable.py
new file mode 100644
index 0000000..0f1c8c8
--- /dev/null
+++ b/cyborg_enhancement/mitaka_version/cyborg/cyborg/tests/unit/fake_deployable.py
@@ -0,0 +1,70 @@
+# Copyright 2018 Huawei Technologies Co.,LTD.
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+
+import datetime
+
+from oslo_serialization import jsonutils
+from oslo_utils import uuidutils
+
+from cyborg import objects
+from cyborg.objects import fields
+
+
+def fake_db_deployable(**updates):
+    root_uuid = uuidutils.generate_uuid()
+    db_deployable = {
+        'id': 1,
+        'deleted': False,
+        'uuid': root_uuid,
+        'name': 'dp_name',
+        'parent_uuid': None,
+        'root_uuid': root_uuid,
+        'pcie_address': '00:7f:0b.2',
+        'host': 'host_name',
+        'board': 'KU115',
+        'vendor': 'Xilinx',
+        'version': '1.0',
+        'type': 'pf',
+        'assignable': True,
+        'instance_uuid': None,
+        'availability': 'Available',
+        'accelerator_id': 1
+        }
+
+    for name, field in objects.Deployable.fields.items():
+        if name in db_deployable:
+            continue
+        if field.nullable:
+            db_deployable[name] = None
+        elif field.default != fields.UnspecifiedDefault:
+            db_deployable[name] = field.default
+        else:
+            raise Exception('fake_db_deployable needs help with %s' % name)
+
+    if updates:
+        db_deployable.update(updates)
+
+    return db_deployable
+
+
+def fake_deployable_obj(context, obj_dpl_class=None, **updates):
+    if obj_dpl_class is None:
+        obj_dpl_class = objects.Deployable
+    expected_attrs = updates.pop('expected_attrs', None)
+    deploy = obj_dpl_class._from_db_object(context,
+                                           obj_dpl_class(),
+                                           fake_db_deployable(**updates),
+                                           expected_attrs=expected_attrs)
+    deploy.obj_reset_changes()
+    return deploy
+\ No newline at end of file