Add EncryptionMode enum and integrate index key derivation

src/tests/test_key_derivation.py

@@ -4,6 +4,8 @@ from utils.key_derivation import (
     derive_key_from_password,
     derive_index_key_seed_only,
     derive_index_key_seed_plus_pw,
+    derive_index_key,
+    EncryptionMode,
 )
 
 
@@ -36,3 +38,17 @@ def test_seed_plus_pw_differs_from_seed_only():
     k1 = derive_index_key_seed_only(seed)
     k2 = derive_index_key_seed_plus_pw(seed, pw)
     assert k1 != k2
+
+
+def test_derive_index_key_modes():
+    seed = "abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon about"
+    pw = "hunter2"
+    assert derive_index_key(
+        seed, pw, EncryptionMode.SEED_ONLY
+    ) == derive_index_key_seed_only(seed)
+    assert derive_index_key(
+        seed, pw, EncryptionMode.SEED_PLUS_PW
+    ) == derive_index_key_seed_plus_pw(seed, pw)
+    assert derive_index_key(
+        seed, pw, EncryptionMode.PW_ONLY
+    ) == derive_key_from_password(pw)