From 52aca30342c8362570eb10c4889873774e35a50d Mon Sep 17 00:00:00 2001
From: Florian Bruhin <git@the-compiler.org>
Date: Mon, 23 Nov 2015 13:31:17 +0100
Subject: [PATCH] tests: Add quteproc.temp_setting.

---
 tests/integration/quteprocess.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/tests/integration/quteprocess.py b/tests/integration/quteprocess.py
index 9499f4abc..3645d7073 100644
--- a/tests/integration/quteprocess.py
+++ b/tests/integration/quteprocess.py
@@ -29,6 +29,7 @@ import os.path
 import datetime
 import logging
 import tempfile
+import contextlib
 
 import yaml
 import pytest
@@ -215,6 +216,14 @@ class QuteProc(testprocess.Process):
         self.send_cmd(':set "{}" "{}" "{}"'.format(sect, opt, value))
         self.wait_for(category='config', message='Config option changed: *')
 
+    @contextlib.contextmanager
+    def temp_setting(self, sect, opt, value):
+        """Context manager to set a setting and reset it on exit."""
+        old_value = self.get_setting(sect, opt)
+        self.set_setting(sect, opt, value)
+        yield
+        self.set_setting(sect, opt, old_value)
+
     def open_path(self, path, new_tab=False):
         url = self._path_to_url(path)
         if new_tab: