chia-blockchain 2.4.4__py3-none-any.whl

chia/_tests/core/util/test_keyring_wrapper.py

@@ -0,0 +1,490 @@
+from __future__ import annotations
+
+import logging
+from typing import Type
+
+import pytest
+
+from chia.simulator.keyring import TempKeyring
+from chia.util.errors import KeychainFingerprintNotFound, KeychainLabelError, KeychainLabelExists, KeychainLabelInvalid
+from chia.util.file_keyring import Key
+from chia.util.keyring_wrapper import DEFAULT_PASSPHRASE_IF_NO_MASTER_PASSPHRASE, KeyringWrapper
+
+log = logging.getLogger(__name__)
+
+
+# TODO: might need to use the anyio_backend fixture per https://anyio.readthedocs.io/en/stable/testing.html
+@pytest.fixture(autouse=True, scope="function")
+def setup_keyring_wrapper():
+    yield
+    KeyringWrapper.cleanup_shared_instance()
+    assert KeyringWrapper.get_shared_instance(create_if_necessary=False) is None
+
+
+class TestKeyringWrapper:
+    def test_shared_instance(self):
+        """
+        Using KeyringWrapper's get_shared_instance() method should return the same
+        instance each time it's called
+        """
+        # When: multiple calls to the shared accessor are made
+        kw1 = KeyringWrapper.get_shared_instance()
+        kw2 = KeyringWrapper.get_shared_instance()
+
+        # Expect: the shared instance should exist
+        assert kw1 is not None
+        # Expect: multiple references should point to the same instance
+        assert id(kw1) == id(kw2)
+
+        # When: destroying the shared instance
+        KeyringWrapper.cleanup_shared_instance()
+
+        # Expect: the shared instance should be cleared
+        assert KeyringWrapper.get_shared_instance(create_if_necessary=False) is None
+
+    # When: creating a new/unpopulated file keyring
+    def test_empty_file_keyring_doesnt_have_master_passphrase(self, empty_temp_file_keyring: TempKeyring):
+        """
+        A new/unpopulated file keyring should not have a master passphrase set
+        """
+        # Expect: no master passphrase set
+        assert KeyringWrapper.get_shared_instance().has_master_passphrase() is False
+
+    # When: using a populated file keyring
+    def test_populated_file_keyring_has_master_passphrase(self, populated_temp_file_keyring: TempKeyring):
+        """
+        Populated keyring should have the default master passphrase set
+        """
+        # Expect: master passphrase is set
+        assert KeyringWrapper.get_shared_instance().has_master_passphrase() is True
+
+    # When: creating a new file keyring
+    def test_default_cached_master_passphrase(self, empty_temp_file_keyring: TempKeyring):
+        """
+        The default passphrase DEFAULT_PASSPHRASE_IF_NO_MASTER_PASSPHRASE is set
+        """
+        # Expect: cached passphrase set to DEFAULT_PASSPHRASE_IF_NO_MASTER_PASSPHRASE by default
+        assert KeyringWrapper.get_shared_instance().get_cached_master_passphrase() == (
+            DEFAULT_PASSPHRASE_IF_NO_MASTER_PASSPHRASE,
+            False,
+        )
+        assert KeyringWrapper.get_shared_instance().has_cached_master_passphrase() is True
+
+    # When: using a file keyring
+    def test_set_cached_master_passphrase(self, empty_temp_file_keyring: TempKeyring):
+        """
+        Setting and retrieving the cached master passphrase should work
+        """
+        # When: setting the cached master passphrase
+        KeyringWrapper.get_shared_instance().set_cached_master_passphrase("testing one two three")
+
+        # Expect: cached passphrase should match
+        assert KeyringWrapper.get_shared_instance().get_cached_master_passphrase() == ("testing one two three", False)
+
+        # When: setting a validated (successfully decrypted the content) master passphrase
+        KeyringWrapper.get_shared_instance().set_cached_master_passphrase("apple banana orange grape", validated=True)
+
+        # Expect: cached passphrase should match and be validated
+        assert KeyringWrapper.get_shared_instance().get_cached_master_passphrase() == (
+            "apple banana orange grape",
+            True,
+        )
+
+    # When: using a populated file keyring
+    def test_master_passphrase_is_valid(self, populated_temp_file_keyring: TempKeyring):
+        """
+        The default master passphrase should unlock the populated keyring (without any keys)
+        """
+        # Expect: default master passphrase should validate
+        assert (
+            KeyringWrapper.get_shared_instance().master_passphrase_is_valid(DEFAULT_PASSPHRASE_IF_NO_MASTER_PASSPHRASE)
+            is True
+        )
+
+        # Expect: bogus passphrase should not validate
+        assert KeyringWrapper.get_shared_instance().master_passphrase_is_valid("foobarbaz") is False
+
+    # When: creating a new unpopulated keyring
+    def test_set_master_passphrase_on_empty_keyring(self, empty_temp_file_keyring: TempKeyring):
+        """
+        Setting a master passphrase should cache the passphrase and be usable to unlock
+        the keyring. Using an old passphrase should not unlock the keyring.
+        """
+        # When: setting the master passphrase
+        KeyringWrapper.get_shared_instance().set_master_passphrase(None, "testing one two three")
+
+        # Expect: the master passphrase is cached and can be validated
+        assert KeyringWrapper.get_shared_instance().get_cached_master_passphrase() == ("testing one two three", True)
+        assert KeyringWrapper.get_shared_instance().master_passphrase_is_valid("testing one two three") is True
+
+        # When: changing the master passphrase
+        KeyringWrapper.get_shared_instance().set_master_passphrase("testing one two three", "potato potato potato")
+
+        # Expect: the new master passphrase is cached and can be validated
+        assert KeyringWrapper.get_shared_instance().get_cached_master_passphrase() == ("potato potato potato", True)
+        assert KeyringWrapper.get_shared_instance().master_passphrase_is_valid("potato potato potato") is True
+
+        # Expect: old passphrase should not validate
+        assert KeyringWrapper.get_shared_instance().master_passphrase_is_valid("testing one two three") is False
+
+    # When: using a populated keyring
+    def test_set_master_passphrase_on_keyring(self, populated_temp_file_keyring: TempKeyring):
+        """
+        Setting a master passphrase should cache the passphrase and be usable to unlock
+        the keyring. Using an old passphrase should not unlock the keyring.
+        """
+        # When: setting the master passphrase
+        KeyringWrapper.get_shared_instance().set_master_passphrase(
+            DEFAULT_PASSPHRASE_IF_NO_MASTER_PASSPHRASE, "testing one two three"
+        )
+
+        # Expect: the master passphrase is cached and can be validated
+        assert KeyringWrapper.get_shared_instance().get_cached_master_passphrase() == ("testing one two three", True)
+        assert KeyringWrapper.get_shared_instance().master_passphrase_is_valid("testing one two three") is True
+
+        # When: changing the master passphrase
+        KeyringWrapper.get_shared_instance().set_master_passphrase("testing one two three", "potato potato potato")
+
+        # Expect: the new master passphrase is cached and can be validated
+        assert KeyringWrapper.get_shared_instance().get_cached_master_passphrase() == ("potato potato potato", True)
+        assert KeyringWrapper.get_shared_instance().master_passphrase_is_valid("potato potato potato") is True
+
+        # Expect: old passphrase should not validate
+        assert KeyringWrapper.get_shared_instance().master_passphrase_is_valid("testing one two three") is False
+
+    # When: using a new empty keyring
+    def test_remove_master_passphrase_from_empty_keyring(self, empty_temp_file_keyring: TempKeyring):
+        """
+        An empty keyring doesn't require a current passphrase to remove the master passphrase.
+        Removing the master passphrase will set the default master passphrase on the keyring.
+        """
+        # When: removing the master passphrase from an empty keyring, current passphrase isn't necessary
+        KeyringWrapper.get_shared_instance().remove_master_passphrase(None)
+
+        # Expect: default master passphrase is set
+        assert KeyringWrapper.get_shared_instance().get_cached_master_passphrase() == (
+            DEFAULT_PASSPHRASE_IF_NO_MASTER_PASSPHRASE,
+            True,
+        )
+        assert (
+            KeyringWrapper.get_shared_instance().master_passphrase_is_valid(DEFAULT_PASSPHRASE_IF_NO_MASTER_PASSPHRASE)
+            is True
+        )
+
+    # When: using a populated keyring
+    def test_remove_master_passphrase_from_populated_keyring(self, populated_temp_file_keyring: TempKeyring):
+        """
+        A populated keyring will require a current passphrase when removing the master passphrase.
+        Removing the master passphrase will set the default master passphrase on the keyring.
+        """
+        # When: the master passphrase is set
+        KeyringWrapper.get_shared_instance().set_master_passphrase(
+            DEFAULT_PASSPHRASE_IF_NO_MASTER_PASSPHRASE, "It's dangerous to go alone, take this!"
+        )
+
+        # When: removing the master passphrase
+        KeyringWrapper.get_shared_instance().remove_master_passphrase("It's dangerous to go alone, take this!")
+
+        # Expect: default master passphrase is set, old passphrase doesn't validate
+        assert KeyringWrapper.get_shared_instance().get_cached_master_passphrase() == (
+            DEFAULT_PASSPHRASE_IF_NO_MASTER_PASSPHRASE,
+            True,
+        )
+        assert (
+            KeyringWrapper.get_shared_instance().master_passphrase_is_valid(DEFAULT_PASSPHRASE_IF_NO_MASTER_PASSPHRASE)
+            is True
+        )
+        assert (
+            KeyringWrapper.get_shared_instance().master_passphrase_is_valid("It's dangerous to go alone, take this!")
+            is False
+        )
+
+    # When: using a new empty keyring
+    def test_get_key(self, empty_temp_file_keyring: TempKeyring):
+        """
+        Simple key setting and retrieval
+        """
+        # Expect: key lookup should return None
+        assert KeyringWrapper.get_shared_instance().keyring.get_key("service-abc", "user-xyz") is None
+
+        # When: setting a key
+        KeyringWrapper.get_shared_instance().keyring.set_key(
+            "service-abc", "user-xyz", Key(b"super secret key", {"foo": "bar"})
+        )
+
+        # Expect: key lookup should succeed
+        assert KeyringWrapper.get_shared_instance().keyring.get_key("service-abc", "user-xyz") == Key(
+            b"super secret key", {"foo": "bar"}
+        )
+
+        # Expect: non-existent key lookup should fail
+        assert KeyringWrapper.get_shared_instance().keyring.get_key("service-123", "some non-existent key") is None
+
+    # When: using a new empty keyring
+    def test_set_key_overwrite(self, empty_temp_file_keyring: TempKeyring):
+        """
+        Overwriting a previously-set key should work
+        """
+        # When: initially setting the key
+        KeyringWrapper.get_shared_instance().keyring.set_key("service-xyz", "user-123", Key(b"initial key"))
+
+        # Expect: key lookup should succeed
+        assert KeyringWrapper.get_shared_instance().keyring.get_key("service-xyz", "user-123") == Key(b"initial key")
+
+        # When: updating the same key
+        KeyringWrapper.get_shared_instance().keyring.set_key("service-xyz", "user-123", Key(b"updated key"))
+
+        # Expect: the updated key should be retrieved
+        assert KeyringWrapper.get_shared_instance().keyring.get_key("service-xyz", "user-123") == Key(b"updated key")
+
+    # When: using a new empty keyring
+    def test_delete_key(self, empty_temp_file_keyring: TempKeyring):
+        """
+        Deleting a non-existent key should fail gracefully (no exceptions)
+        """
+        # Expect: deleting a non-existent key should fail gracefully
+        KeyringWrapper.get_shared_instance().keyring.delete_key("some service", "some user")
+
+        # When: setting a key
+        KeyringWrapper.get_shared_instance().keyring.set_key("some service", "some user", Key(b"500p3r 53cr37"))
+
+        # Expect: key retrieval should succeed
+        assert KeyringWrapper.get_shared_instance().keyring.get_key("some service", "some user") == Key(
+            b"500p3r 53cr37"
+        )
+
+        # When: deleting the key
+        KeyringWrapper.get_shared_instance().keyring.delete_key("some service", "some user")
+
+        # Expect: key retrieval should fail gracefully
+        assert KeyringWrapper.get_shared_instance().keyring.get_key("some service", "some user") is None
+
+        # Check that metadata is properly deleted
+        from chia.cmds.passphrase_funcs import obtain_current_passphrase
+
+        passphrase = obtain_current_passphrase(use_passphrase_cache=True)
+        assert KeyringWrapper.get_shared_instance().keyring.cached_file_content.get_decrypted_data_dict(passphrase) == {
+            "keys": {},
+            "labels": {},
+            "metadata": {},
+        }
+
+    def test_emoji_master_passphrase(self, empty_temp_file_keyring: TempKeyring):
+        """
+        Emoji master passphrases should just work 😀
+        """
+        # When: setting a passphrase containing emojis
+        KeyringWrapper.get_shared_instance().set_master_passphrase(None, "🥳🤩🤪🤯😎😝😀")
+
+        # Expect: the master passphrase is cached and can be validated
+        assert KeyringWrapper.get_shared_instance().get_cached_master_passphrase() == ("🥳🤩🤪🤯😎😝😀", True)
+        assert KeyringWrapper.get_shared_instance().master_passphrase_is_valid("🥳🤩🤪🤯😎😝😀") is True
+
+        # Expect: an invalid passphrase containing an emoji should fail validation
+        assert KeyringWrapper.get_shared_instance().get_cached_master_passphrase() != ("🦄🦄🦄🦄🦄🦄🦄🦄", True)
+        assert KeyringWrapper.get_shared_instance().master_passphrase_is_valid("🦄🦄🦄🦄🦄🦄🦄🦄") is False
+
+    def test_japanese_master_passphrase(self, empty_temp_file_keyring: TempKeyring):
+        """
+        Non-ascii master passphrases should just work
+        """
+        # When: setting a passphrase containing non-ascii characters
+        KeyringWrapper.get_shared_instance().set_master_passphrase(None, "私は幸せな農夫です")
+
+        # Expect: the master passphrase is cached and can be validated
+        assert KeyringWrapper.get_shared_instance().get_cached_master_passphrase() == ("私は幸せな農夫です", True)
+        assert KeyringWrapper.get_shared_instance().master_passphrase_is_valid("私は幸せな農夫です") is True
+
+        # Expect: an invalid passphrase containing an non-ascii characters should fail validation
+        assert KeyringWrapper.get_shared_instance().get_cached_master_passphrase() != (
+            "私は幸せな農夫ではありません",
+            True,
+        )
+        assert KeyringWrapper.get_shared_instance().master_passphrase_is_valid("私は幸せな農夫ではありません") is False
+
+    # When: using a new empty keyring
+    def test_set_master_passphrase_with_hint(self, empty_temp_file_keyring: TempKeyring):
+        """
+        Setting a passphrase hint at the same time as setting the passphrase
+        """
+        # When: setting the master passphrase with a hint
+        KeyringWrapper.get_shared_instance().set_master_passphrase(
+            None, "new master passphrase", passphrase_hint="some passphrase hint"
+        )
+
+        # Expect: hint can be retrieved
+        assert KeyringWrapper.get_shared_instance().get_master_passphrase_hint() == "some passphrase hint"
+
+    def test_passphrase_hint(self, empty_temp_file_keyring: TempKeyring):
+        """
+        Setting and retrieving the passphrase hint
+        """
+        # Expect: no hint set by default
+        assert KeyringWrapper.get_shared_instance().get_master_passphrase_hint() is None
+
+        # When: setting the passphrase hint while setting the master passphrase
+        KeyringWrapper.get_shared_instance().set_master_passphrase(
+            None, "passphrase", passphrase_hint="rhymes with bassphrase"
+        )
+
+        # Expect: to retrieve the passphrase hint that was just set
+        assert KeyringWrapper.get_shared_instance().get_master_passphrase_hint() == "rhymes with bassphrase"
+
+        # When: writing the keyring again
+        KeyringWrapper.get_shared_instance().keyring.write_keyring()
+
+        # Expect: the hint is still set
+        assert KeyringWrapper.get_shared_instance().get_master_passphrase_hint() == "rhymes with bassphrase"
+
+    def test_passphrase_hint_removal(self, empty_temp_file_keyring: TempKeyring):
+        """
+        Removing a passphrase hint
+        """
+        # When: setting the passphrase hint while setting the master passphrase
+        KeyringWrapper.get_shared_instance().set_master_passphrase(
+            None, "12345", passphrase_hint="President Skroob's luggage combination"
+        )
+
+        # Expect: to retrieve the passphrase hint that was just set
+        assert (
+            KeyringWrapper.get_shared_instance().get_master_passphrase_hint()
+            == "President Skroob's luggage combination"
+        )
+
+        # When: removing the passphrase hint
+        KeyringWrapper.get_shared_instance().set_master_passphrase("12345", "12345", passphrase_hint=None)
+
+        # Expect: passphrase hint has been removed
+        assert KeyringWrapper.get_shared_instance().get_master_passphrase_hint() is None
+
+    def test_passphrase_hint_update(self, empty_temp_file_keyring: TempKeyring):
+        """
+        Updating a passphrase hint
+        """
+        # When: setting the passphrase hint while setting the master passphrase
+        KeyringWrapper.get_shared_instance().set_master_passphrase(
+            None, "i like turtles", passphrase_hint="My deepest darkest secret"
+        )
+
+        # Expect: to retrieve the passphrase hint that was just set
+        assert KeyringWrapper.get_shared_instance().get_master_passphrase_hint() == "My deepest darkest secret"
+
+        # When: updating the passphrase hint
+        KeyringWrapper.get_shared_instance().set_master_passphrase(
+            "i like turtles", "i like turtles", passphrase_hint="Something you wouldn't expect The Shredder to say"
+        )
+
+        # Expect: to retrieve the passphrase hint that was just set
+        assert (
+            KeyringWrapper.get_shared_instance().get_master_passphrase_hint()
+            == "Something you wouldn't expect The Shredder to say"
+        )
+
+    def test_get_label(self, empty_temp_file_keyring: TempKeyring):
+        keyring_wrapper = KeyringWrapper.get_shared_instance()
+        # label lookup for 1, 2, 3 should return None
+        assert keyring_wrapper.keyring.get_label(1) is None
+        assert keyring_wrapper.keyring.get_label(2) is None
+        assert keyring_wrapper.keyring.get_label(3) is None
+
+        # Set and validate a label for 1
+        keyring_wrapper.keyring.set_label(1, "one")
+        assert keyring_wrapper.keyring.get_label(1) == "one"
+
+        # Set and validate a label for 3
+        keyring_wrapper.keyring.set_label(3, "three")
+
+        # And validate all match the expected values
+        assert keyring_wrapper.keyring.get_label(1) == "one"
+        assert keyring_wrapper.keyring.get_label(2) is None
+        assert keyring_wrapper.keyring.get_label(3) == "three"
+
+    def test_set_label(self, empty_temp_file_keyring: TempKeyring):
+        keyring_wrapper = KeyringWrapper.get_shared_instance()
+        # Set and validate a label for 1
+        keyring_wrapper.keyring.set_label(1, "one")
+        assert keyring_wrapper.keyring.get_label(1) == "one"
+
+        # Set and validate a label for 2
+        keyring_wrapper.keyring.set_label(2, "two")
+        assert keyring_wrapper.keyring.get_label(2) == "two"
+
+        # Change the label of 2
+        keyring_wrapper.keyring.set_label(2, "two!")
+        assert keyring_wrapper.keyring.get_label(2) == "two!"
+        # 1 should still have the same label
+        assert keyring_wrapper.keyring.get_label(1) == "one"
+
+        # Change the label of 2 again
+        keyring_wrapper.keyring.set_label(2, "two!!")
+        assert keyring_wrapper.keyring.get_label(2) == "two!!"
+        # 1 should still have the same label
+        assert keyring_wrapper.keyring.get_label(1) == "one"
+
+        # Change the label of 1
+        keyring_wrapper.keyring.set_label(1, "one!")
+        assert keyring_wrapper.keyring.get_label(1) == "one!"
+        # 2 should still have the same label
+        assert keyring_wrapper.keyring.get_label(2) == "two!!"
+
+    @pytest.mark.parametrize(
+        "label",
+        [
+            "🥳🤩🤪🤯😎😝😀",
+            "私は幸せな農夫です",
+            "لتفاصيل لتكتشف حقيقة وأساس ت",
+        ],
+    )
+    def test_set_special_labels(self, label: str, empty_temp_file_keyring: TempKeyring):
+        keyring_wrapper = KeyringWrapper.get_shared_instance()
+        keyring_wrapper.keyring.set_label(1, label)
+        assert keyring_wrapper.keyring.get_label(1) == label
+
+    @pytest.mark.parametrize(
+        "label, exception, message",
+        [
+            ("one", KeychainLabelExists, "label 'one' already exists for fingerprint '1"),
+            ("", KeychainLabelInvalid, "label can't be empty or whitespace only"),
+            ("   ", KeychainLabelInvalid, "label can't be empty or whitespace only"),
+            ("a\nb", KeychainLabelInvalid, "label can't contain newline or tab"),
+            ("a\tb", KeychainLabelInvalid, "label can't contain newline or tab"),
+            ("a label ", KeychainLabelInvalid, "label can't contain leading or trailing whitespaces"),
+            (" a label", KeychainLabelInvalid, "label can't contain leading or trailing whitespaces"),
+            (" a label ", KeychainLabelInvalid, "label can't contain leading or trailing whitespaces"),
+            ("  a label ", KeychainLabelInvalid, "label can't contain leading or trailing whitespaces"),
+            ("a" * 66, KeychainLabelInvalid, "label exceeds max length: 66/65"),
+            ("a" * 70, KeychainLabelInvalid, "label exceeds max length: 70/65"),
+        ],
+    )
+    def test_set_label_failures(
+        self, label: str, exception: Type[KeychainLabelError], message: str, empty_temp_file_keyring: TempKeyring
+    ) -> None:
+        keyring_wrapper = KeyringWrapper.get_shared_instance()
+        keyring_wrapper.keyring.set_label(1, "one")
+        with pytest.raises(exception, match=message) as e:
+            keyring_wrapper.keyring.set_label(1, label)
+        assert e.value.label == label
+        if isinstance(e.value, KeychainLabelExists):
+            assert e.value.label == "one"
+            assert e.value.fingerprint == 1
+
+    def test_delete_label(self, empty_temp_file_keyring: TempKeyring) -> None:
+        keyring_wrapper = KeyringWrapper.get_shared_instance()
+        # Set labels for 1,2 and validate them
+        keyring_wrapper.keyring.set_label(1, "one")
+        keyring_wrapper.keyring.set_label(2, "two")
+        assert keyring_wrapper.keyring.get_label(1) == "one"
+        assert keyring_wrapper.keyring.get_label(2) == "two"
+        # Remove the label of 1
+        keyring_wrapper.keyring.delete_label(1)
+        assert keyring_wrapper.keyring.get_label(1) is None
+        assert keyring_wrapper.keyring.get_label(2) == "two"
+        # Remove the label of 2
+        keyring_wrapper.keyring.delete_label(2)
+        assert keyring_wrapper.keyring.get_label(1) is None
+        assert keyring_wrapper.keyring.get_label(2) is None
+        # Make sure the deletion fails for 0-2
+        for i in range(3):
+            with pytest.raises(KeychainFingerprintNotFound) as e:
+                keyring_wrapper.keyring.delete_label(i)
+            assert e.value.fingerprint == i