commit fdae070f8bcbeb4902114460f8b37b62ea82df53
author Paul Kehrer <paul.l.kehrer@gmail.com> Thu Nov 27 07:50:46 2014 -1000
committer Paul Kehrer <paul.l.kehrer@gmail.com> Thu Nov 27 08:03:22 2014 -1000
tree f5b0df8da3badeb9fe1c2d2845574485f6d4e8b4
parent a28874863a2cc76045e5526ae2b701b9e7733d02

alter vector file loader to support passing read mode

tests/test_fernet.py

diff --git a/tests/test_fernet.py b/tests/test_fernet.py
index 455d985..5b7a9f9 100644
--- a/tests/test_fernet.py
+++ b/tests/test_fernet.py
@@ -7,6 +7,7 @@
 import base64
 import calendar
 import json
+import os
 import time
 
 import iso8601
@@ -24,7 +25,9 @@
 
 
 def json_parametrize(keys, filename):
-    vector_file = cryptography_vectors.open_vector_file('fernet', filename)
+    vector_file = cryptography_vectors.open_vector_file(
+        os.path.join('fernet', filename), "r"
+    )
     with vector_file:
         data = json.load(vector_file)
         return pytest.mark.parametrize(keys, [

tests/utils.py

diff --git a/tests/utils.py b/tests/utils.py
index ad38000..01ab4e6 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -87,8 +87,8 @@
     return encoder.encode(sig)
 
 
-def load_vectors_from_file(filename, loader):
-    with cryptography_vectors.open_vector_file(filename) as vector_file:
+def load_vectors_from_file(filename, loader, mode="r"):
+    with cryptography_vectors.open_vector_file(filename, mode) as vector_file:
         return loader(vector_file)