[Kimchi-devel] [PATCHv4 3/6] Add mockmodel for vm disk attach

[lvroyce at linux.vnet.ibm.com]

From: Royce Lv <lvroyce at linux.vnet.ibm.com>

Add mockmodel for disk attach.

Signed-off-by: Royce Lv <lvroyce at linux.vnet.ibm.com>
---
 src/kimchi/mockmodel.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/kimchi/mockmodel.py b/src/kimchi/mockmodel.py
index 05720f4..aecc2a1 100644
--- a/src/kimchi/mockmodel.py
+++ b/src/kimchi/mockmodel.py
@@ -663,9 +663,13 @@ class MockModel(object):
 
     def vmstorages_create(self, vm_name, params):
         path = params.get('path')
-        if path.startswith('/') and not os.path.exists(path):
+        if path and path.startswith('/') and not os.path.exists(path):
             raise InvalidParameter("KCHVMSTOR0003E", {'value': path})
-
+        elif params.get('pool'):
+            try:
+                self.storagevolume_lookup(params['pool'], params['vol'])
+            except Exception as e:
+                raise InvalidParameter("KCHVMSTOR0015E", {'error': e})
         dom = self._get_vm(vm_name)
         dev = params.get('dev', None)
         if dev and dev in self.vmstorages_get_list(vm_name):
@@ -970,6 +974,8 @@ class MockVMStorageDevice(object):
     def __init__(self, params):
         self.info = {'dev': params.get('dev'),
                      'type': params.get('type'),
+                     'pool': params.get('pool'),
+                     'vol': params.get('vol'),
                      'path': params.get('path')}
 
 
-- 
1.8.3.2