feat: impl compile code

ApeWorX · Nov 9, 2023 · 41a0235 · 41a0235
1 parent c24acae
commit 41a0235
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/ape_solidity/compiler.py b/ape_solidity/compiler.py
@@ -544,7 +544,15 @@ def compile_code(
         except SolcError as err:
             raise CompilerError(str(err)) from err
 
+        output = result[next(iter(result.keys()))]
         return ContractType(
+            abi=output["abi"],
+            ast=output["ast"],
+            deploymentBytecode={"bytecode": HexBytes(output["bin"])},
+            devdoc=load_dict(output["devdoc"]),
+            runtimeBytecode={"bytecode": HexBytes(output["bin-runtime"])},
+            sourcemap=output["srcmap"],
+            userdoc=load_dict(output["userdoc"]),
             **kwargs,
         )
 

diff --git a/tests/test_compiler.py b/tests/test_compiler.py
@@ -535,6 +535,9 @@ def test_compile_code(compiler):
     }
 }
 """
-    actual = compiler.compile_code(code, contractName="python")
-    assert actual.name == "python"
-    assert actual.abi == ["asdf"]
+    actual = compiler.compile_code(code, contractName="TestContractName")
+    assert actual.name == "TestContractName"
+    assert len(actual.abi) > 0
+    assert actual.ast is not None
+    assert len(actual.runtime_bytecode.bytecode) > 0
+    assert len(actual.deployment_bytecode.bytecode) > 0