| # Copyright (c) 2013 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 ShillTemporaryProfile(object): |
| """Context enclosing the use of a temporary shill profile. It takes |
| a shill manager dbus object and profile name, and makes sure that |
| this profile is pushed atop the topmost default profile for the duration |
| of this object lifetime.""" |
| def __init__(self, manager, profile_name='test'): |
| self._manager = manager |
| self._profile_name = profile_name |
| |
| |
| def __enter__(self): |
| self._manager.PopAllUserProfiles() |
| try: |
| self._manager.RemoveProfile(self._profile_name) |
| except: |
| pass |
| self._manager.CreateProfile(self._profile_name) |
| self._manager.PushProfile(self._profile_name) |
| return self |
| |
| |
| def __exit__(self, exception, value, traceback): |
| try: |
| self._manager.PopProfile(self._profile_name) |
| self._manager.RemoveProfile(self._profile_name) |
| except: |
| pass |
