Windows compatibility for test suite

1 files changed, 13 insertions, 2 deletions

diff --git a/tests/util.py b/tests/util.py
index 3f39891..1a44e83 100644
--- a/tests/util.py
+++ b/tests/util.py
@@ -181,6 +181,18 @@ def only_ipv6(fn):
 
     return decorated
 
+def only_unix(fn):
+    import nose.plugins.skip
+
+    @functools.wraps(fn)
+    def decorated(*args, **kwargs):
+        if sys.platform == 'win32':
+            raise nose.plugins.skip.SkipTest('Unix only')
+
+        return fn(*args, **kwargs)
+
+    return decorated
+
 def guard_unknown_libcurl_option(fn):
     '''Converts curl error 48, CURLE_UNKNOWN_OPTION, into a SkipTest
     exception. This is meant to be used with tests exercising libcurl
@@ -274,6 +286,5 @@ def with_real_write_file(fn):
     @functools.wraps(fn)
     def wrapper(*args):
         with tempfile.NamedTemporaryFile() as f:
-            with open(f.name, 'w+b') as real_f:
-                return fn(*(list(args) + [real_f]))
+            return fn(*(list(args) + [f.file]))
     return wrapper