test: add test

ape_vyper/compiler.py

@@ -418,10 +418,29 @@ def compile(
 
         return contract_types
 
+    def compile_code(
+        self,
+        code: str,
+        base_path: Optional[Path] = None,
+        contract_type_overrides: Optional[Dict] = None,
+    ) -> ContractType:
+        base_path = base_path or self.project_manager.contracts_folder
+        try:
+            result = vvm.compile_source(code, base_path=base_path)
+        except Exception as err:
+            raise VyperCompileError(str(err)) from err
+
+        output = result.get("<stdin>", {})
+        return ContractType(
+            abi=output["abi"],
+            deploymentBytecode={"bytecode": output["bytecode"]},
+            runtimeBytecode={"bytecode": output["bytecode_runtime"]},
+            **(contract_type_overrides or {}),
+        )
+
     def get_optimization_pragma_map(
-        self, contract_filepaths: List[Path], base_path: Optional[Path] = None
+        self, contract_filepaths: List[Path]
     ) -> Dict[Union[str, bool], Set[Path]]:
-        base_path = base_path or self.config_manager.contracts_folder
         optimization_pragma_map: Dict[Union[str, bool], Set[Path]] = {}
         for path in contract_filepaths:
             pragma = get_optimization_pragma(path) or True

tests/test_compiler.py

@@ -2,6 +2,7 @@
 
 import pytest
 from ape.exceptions import ContractLogicError
+from ethpm_types import ContractType
 from packaging.version import Version
 from vvm import compile_source  # type: ignore
 from vvm.exceptions import VyperError  # type: ignore
@@ -459,3 +460,13 @@ def test_compile_with_version_set(config, projects_path, compiler, mocker):
         # Show it uses this version in the compiler.
         project.load_contracts(use_cache=False)
         assert str(spy.call_args[1]["vyper_version"]) == version_from_config
+
+
+def test_compile_code(compiler, dev_revert_source):
+    code = dev_revert_source.read_text()
+    actual = compiler.compile_code(code, contract_type_overrides={"contractName": "MyContract"})
+    assert isinstance(actual, ContractType)
+    assert actual.name == "MyContract"
+    assert len(actual.abi) > 1
+    assert len(actual.deployment_bytecode.bytecode) > 1
+    assert len(actual.runtime_bytecode.bytecode) > 1