Test that Host: header is present

jlusiardi · Apr 14, 2019 · 1ac741a · 1ac741a
1 parent 5be2878
commit 1ac741a
Showing 1 changed file with 13 additions and 12 deletions.
diff --git a/tests/secure_http_test.py b/tests/secure_http_test.py
@@ -21,8 +21,7 @@
 
 from homekit.http_impl.secure_http import SecureHttp
 from homekit.exceptions import AccessoryDisconnectedError, EncryptionError
-from homekit.crypto.chacha20poly1305 import chacha20_aead_encrypt
-
+from homekit.crypto.chacha20poly1305 import chacha20_aead_encrypt, chacha20_aead_decrypt
 
 class ResponseProvider(threading.Thread):
     """
@@ -41,16 +40,18 @@ def __init__(self, sock, c2a_key, a2c_key, encryption_fail=False):
         self.encryption_fail = encryption_fail
 
     def run(self):
-        # tmp = self.sock.recv(1024)
-        # length = int.from_bytes(tmp[0:2], 'little')
-        # tmp = tmp[2:]
-        # block = tmp[0:length]
-        # tmp = tmp[length:]
-        # tag = tmp[0:16]
-        # request = chacha20_aead_decrypt(length.to_bytes(2, byteorder='little'),
-        #                                 self.c2a_key,
-        #                                 self.c2a_counter.to_bytes(8, byteorder='little'),
-        #                                 bytes([0, 0, 0, 0]), block + tag)
+        tmp = self.sock.recv(1024)
+        length = int.from_bytes(tmp[0:2], 'little')
+        tmp = tmp[2:]
+        block = tmp[0:length]
+        tmp = tmp[length:]
+        tag = tmp[0:16]
+        request = chacha20_aead_decrypt(length.to_bytes(2, byteorder='little'),
+                                        self.c2a_key,
+                                        self.c2a_counter.to_bytes(8, byteorder='little'),
+                                        bytes([0, 0, 0, 0]), block + tag)
+
+        assert b'Host:' in request
 
         self.c2a_counter += 1