Fix dummy driver

[sfa.git] / sfa / dummy / dummyslices.py

diff --git a/sfa/dummy/dummyslices.py b/sfa/dummy/dummyslices.py
index e65dddc..d334aa2 100644 (file)
--- a/sfa/dummy/dummyslices.py
+++ b/sfa/dummy/dummyslices.py
@@ -72,7 +72,7 @@ class DummySlices:
         if 'node_ids' not in slice.keys():
             slice['node_ids']=[] 
         nodes = self.driver.shell.GetNodes({'node_ids': slice['node_ids']})
-        current_slivers = [node['hostname'] for node in nodes]
+        current_slivers = [node['node_id'] for node in nodes]
 
         # remove nodes not in rspec
         deleted_nodes = list(set(current_slivers).difference(requested_slivers))
@@ -103,7 +103,7 @@ class DummySlices:
             slice['user_ids'] = []
         else:
             slice = slices[0]
-            if slice_record.get('expires'):
+            if slice_record and slice_record.get('expires'):
                 requested_expires = int(datetime_to_epoch(utcparse(slice_record['expires'])))
                 if requested_expires and slice['expires'] != requested_expires:
                     self.driver.shell.UpdateSlice( {'slice_id': slice['slice_id'], 'fields':{'expires' : requested_expires}})