commit c1d8463aac0b2a3a8b684e702ddd29cf335aa5fb
author Hirokazu Honda <hiroh@chromium.org> Sat Mar 03 08:34:37 2018 +0000
committer android-build-merger <android-build-merger@google.com> Sat Mar 03 08:34:37 2018 +0000
tree 01fcfebac0bcdb84e3fc02b7a423a80d26c31c4f
parent 0c868e200cc0798ef9168637e8deff8aafca32e3
parent 528820bdfd64097463dd3f738719fdf96ff9a440

device_capability: Add func that returns whether a specified capability is available on a device am: 110939e8da am: cdbb39bc9e am: 8b0a223f57
am: 528820bdfd

Change-Id: I3291149f8b3897b02162ec82272a9572d34818bf

client/cros/video/device_capability.py

diff --git a/client/cros/video/device_capability.py b/client/cros/video/device_capability.py
index ad9d090..de97026 100644
--- a/client/cros/video/device_capability.py
+++ b/client/cros/video/device_capability.py
@@ -119,7 +119,7 @@
         try:
             return self.capabilities[cap]
         except KeyError:
-            raise error.TestFail("Unexpected capability: %s" % capability)
+            raise error.TestFail("Unexpected capability: %s" % cap)
 
 
     def ensure_capability(self, cap):
@@ -128,3 +128,10 @@
         """
         if self.get_capability(cap) != 'yes':
             raise error.TestNAError("Missing Capability: %s" % cap)
+
+
+    def have_capability(self, cap):
+        """
+        Return whether cap is available.
+        """
+        return self.get_capability(cap) == 'yes'