Fix missing test in get_attachment and get_measurement

Fix bug introduced when backporting get_attachment and get_measurements. Fixes traceback when the requested attachment or measurement is missing.
google · Dec 28, 2017 · 1d96d52 · 1d96d52
1 parent ac2bcae
commit 1d96d52
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/openhtf/core/test_state.py b/openhtf/core/test_state.py
@@ -173,7 +173,7 @@ def get_attachment(self, attachment_name):
         return copy.deepcopy(attachment)
 
 
-    test.logger.warning('Could not find attachment: %s', attachment_name)
+    self.logger.warning('Could not find attachment: %s', attachment_name)
     return None
 
   def get_measurement(self, measurement_name):
@@ -204,7 +204,7 @@ def get_measurement(self, measurement_name):
         measurement = phase_record.measurements[measurement_name]
         return ImmutableMeasurement.FromMeasurement(measurement)
 
-    test.logger.warning('Could not find measurement: %s', measurement_name)
+    self.logger.warning('Could not find measurement: %s', measurement_name)
     return None
 
   @contextlib.contextmanager