Implement create_with_seed and relevant test (#69)

Co-authored-by: Lucas Bruder <lbruder@me.com>
michaelhly · May 31, 2021 · 964aa25 · 964aa25
1 parent f468483
commit 964aa25
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/solana/publickey.py b/solana/publickey.py
@@ -66,7 +66,8 @@ def to_base58(self) -> bytes:
     @staticmethod
     def create_with_seed(from_public_key: PublicKey, seed: str, program_id: PublicKey) -> PublicKey:
         """Derive a public key from another key, a seed, and a program ID."""
-        raise NotImplementedError("create_with_seed not implemented")
+        buf = bytes(from_public_key) + seed.encode("utf-8") + bytes(program_id)
+        return PublicKey(sha256(buf).digest())
 
     @staticmethod
     def create_program_address(seeds: List[bytes], program_id: PublicKey) -> PublicKey:

diff --git a/tests/unit/test_publickey.py b/tests/unit/test_publickey.py
@@ -252,3 +252,10 @@ def test_is_on_curve():
 
     off_curve = PublicKey("12rqwuEgBYiGhBrDJStCiqEtzQpTTiZbh7teNVLuYcFA")
     assert not PublicKey._is_on_curve(pubkey_bytes=bytes(off_curve))  # pylint: disable=protected-access
+
+
+def test_create_with_seed():
+    """Test create with seed"""
+    default_public_key = PublicKey("11111111111111111111111111111111")
+    derived_key = PublicKey.create_with_seed(default_public_key, "limber chicken: 4/45", default_public_key)
+    assert derived_key == PublicKey("9h1HyLCW5dZnBVap8C5egQ9Z6pHyjsh5MNy83iPqqRuq")