feat: add tests

Author: rohan-chaturvedi

tests/test_decrypt.py

@@ -0,0 +1,41 @@
+import pytest
+from src.phase import Phase
+import src
+
+APP_ID = "phApp:v1:e0e50cb9a1953c610126b4092093b1beca51d08d91fc3d9f8d90482a32853215"
+APP_SECRET = "pss:v1:d261abecb6708c18bebdb8b2748ee574e2b0bdeaf19b081a5f10006cc83d48d0:d146c8c6d326a7842ff9b2da0da455b3f7f568a70808e2eb0cfc5143d4fe170f:59e413612e06d75d251e3416361d0743345a9c9eda1cbcf2b1ef16e3077c011c"
+APP_SECRET_INVALID = "pss:v1:d251abecb6708c18bebdb8b2748ee574e2b0bdeaf19b081a5f10006cc83d48d0:d146c8c6d326a7842ff9b2da0da455b3f7f568a70808e2eb0cfc5143d4fe170d:59e413612e06d75d251e3416361d0743345a9c9eda1cbcf2b1ef16e3077c012d"
+
+
+@pytest.fixture(scope="module")
+def phase_instance():
+    return Phase(APP_ID, APP_SECRET)
+
+
+def mock_fetch_app_key(appToken, wrapKey, appId, dataSize):
+    return "e35ae9560207c90fa3dd68a8715e13a1ef988bffa284db73f04328df17f37cfe"
+
+
+def test_phase_decrypt_returns_correct_plaintext(phase_instance, monkeypatch):
+    data = "Signal"
+
+    monkeypatch.setattr(src.phase, "fetch_app_key", mock_fetch_app_key)
+
+    ciphertext = phase_instance.encrypt(data)
+
+    plaintext = phase_instance.decrypt(ciphertext)
+
+    assert plaintext is not None
+    assert plaintext == data
+
+
+def test_phase_decrypt_rejects_promise_with_incorrect_app_secret(monkeypatch):
+    phase = Phase(APP_ID, APP_SECRET_INVALID)
+
+    monkeypatch.setattr(src.phase, "fetch_app_key", mock_fetch_app_key)
+
+    data = "Signal"
+    ciphertext = phase.encrypt(data)
+
+    with pytest.raises(ValueError, match="Something went wrong"):
+        phase.decrypt(ciphertext)

tests/test_encrypt.py

@@ -0,0 +1,41 @@
+import pytest
+import re
+from src.phase import Phase
+
+APP_ID = "phApp:v1:cd2d579490fd794f1640590220de86a3676fa7979d419056bc631741b320b701"
+APP_SECRET = "pss:v1:a7a0822aa4a4e4d37919009264200ba6ab978d92c8b4f7db5ae9ce0dfaf604fe:801605dfb89822ff52957abe39949bcfc44b9058ad81de58dd54fb0b110037b4b2bbde5a1143d31bbb3895f72e4ee52f5bd:625d395987f52c37022063eaf9b6260cad9ca03c99609213f899cae7f1bb04e7"
+
+
+@pytest.fixture(scope="module")
+def phase_instance():
+    return Phase(APP_ID, APP_SECRET)
+
+
+def test_phase_encrypt_returns_valid_ph(phase_instance):
+    plaintext = "Signal"
+    tag = "Phase Tag"
+    PH_VERSION = "v1"
+
+    ciphertext = phase_instance.encrypt(plaintext, tag)
+
+    assert ciphertext is not None
+    segments = ciphertext.split(":")
+    assert len(segments) == 5
+    assert segments[0] == "ph"
+    assert segments[1] == PH_VERSION
+    assert segments[4] == tag
+    assert re.match("^[0-9a-f]+$", segments[2]) is not None
+    assert re.match(
+        "^(?:[A-Za-z0-9+/]{4})*(?:[A-Za-z0-9+/]{2}==|[A-Za-z0-9+/]{3}=$)", segments[3]) is not None
+
+
+def test_phase_encrypt_produces_same_length_ciphertexts(phase_instance):
+    data = "hello world"
+    num_of_trials = 10
+    ciphertext_lengths = set()
+
+    for _ in range(num_of_trials):
+        ciphertext = phase_instance.encrypt(data)
+        ciphertext_lengths.add(len(ciphertext))
+
+    assert len(ciphertext_lengths) == 1

tests/test_init.py

@@ -0,0 +1,22 @@
+from src.phase import Phase
+import pytest
+
+APP_ID_INVALID = "phApp:v1:cd2d579490fd794f1640590220de86a3676fa7979d419056bc631741b320b701"
+APP_SECRET_INVALID = "pss:v1:a7a0822aa4a4e4d37919009264200ba6ab978d92c8b4f7db5ae9ce0dfaf604fe:801605dfb89822ff52957abe39949bcfc44b9058ad81de58dd54fb0b110037b4b2bbde5a1143d31bbb3895f72e4ee52f5bd:625d395987f52c37022063eaf9b6260cad9ca03c99609213f899cae7f1bb04e7"
+
+
+@pytest.fixture(scope="module")
+def phase_instance():
+    return Phase(APP_ID_INVALID, APP_SECRET_INVALID)
+
+
+def test_invalid_app_id(phase_instance):
+    invalid_app_id = "phApp:version:cd2d579490fd794f1640590220de86a3676fa7979d419056bc631741b320b701"
+    with pytest.raises(ValueError, match="Invalid Phase APP_ID"):
+        Phase(invalid_app_id, APP_SECRET_INVALID)
+
+
+def test_invalid_app_secret(phase_instance):
+    invalid_app_secret = "pss:v1:00000000000000000000000000000000:00000000000000000000000000000000:00000000000000000000000000000000"
+    with pytest.raises(ValueError, match="Invalid Phase APP_SECRET"):
+        Phase(APP_ID_INVALID, invalid_app_secret)