| # Copyright 2015 The Chromium OS Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| class WiFiInterfaceClaimContext(object): |
| """Context that encapsulates claiming of a wifi interface. |
| |
| This context ensures that if the test fails while the interface is claimed |
| we will attempt to release it before our test exits. |
| |
| """ |
| |
| def __init__(self, client): |
| self._client = client |
| |
| |
| def __enter__(self): |
| self._client.claim_wifi_if() |
| |
| |
| def __exit__(self, exception, value, traceback): |
| self._client.release_wifi_if() |
