Splitters in tests.

tests/characters/test_alice_can_grant_and_revoke.py

@@ -1,13 +1,13 @@
 import datetime
 
+from apistar.test import TestClient
+
 from nkms.characters import Ursula
 from nkms.crypto.api import keccak_digest
 from nkms.crypto.constants import PUBLIC_KEY_LENGTH
 from nkms.crypto.powers import SigningPower, EncryptingPower
 from nkms.crypto.utils import BytestringSplitter
 from tests.utilities import MockNetworkyStuff
-from apistar.test import TestClient
-
 from umbral.keys import UmbralPublicKey
 
 
@@ -16,7 +16,8 @@ def test_grant(alice, bob, ursulas):
     policy_end_datetime = datetime.datetime.now() + datetime.timedelta(days=5)
     n = 5
     uri = b"this_is_the_path_to_which_access_is_being_granted"
-    policy = alice.grant(bob, uri, networky_stuff, m=3, n=n, expiration=policy_end_datetime)
+    policy = alice.grant(bob, uri, networky_stuff, m=3, n=n,
+                         expiration=policy_end_datetime)
 
     # The number of policies is equal to the number of Ursulas we're using (n)
     assert len(policy._accepted_contracts) == n
@@ -33,7 +34,13 @@ def test_grant(alice, bob, ursulas):
 def test_alice_can_get_ursulas_keys_via_rest(alice, ursulas):
     mock_client = TestClient(ursulas[0].rest_app)
     response = mock_client.get('http://localhost/public_keys')
-    splitter = BytestringSplitter((UmbralPublicKey, PUBLIC_KEY_LENGTH))
-    signing_key_bytes, encrypting_key_bytes = splitter(response.content, return_remainder=True)
-    stranger_ursula_from_public_keys = Ursula.from_public_keys((SigningPower, signing_key_bytes), (EncryptingPower, encrypting_key_bytes))
+    splitter = BytestringSplitter(
+        (UmbralPublicKey, PUBLIC_KEY_LENGTH, {"as_b64": False}),
+        (UmbralPublicKey, PUBLIC_KEY_LENGTH, {"as_b64": False})
+    )
+    signing_key, encrypting_key = splitter(response.content)
+    stranger_ursula_from_public_keys = Ursula.from_public_keys((SigningPower,
+                                                                signing_key),
+                                                               (EncryptingPower,
+                                                                encrypting_key))
     assert stranger_ursula_from_public_keys == ursulas[0]