diff --git a/pyoaev/contracts/contract_config.py b/pyoaev/contracts/contract_config.py index 8de69ad..07b9d6d 100644 --- a/pyoaev/contracts/contract_config.py +++ b/pyoaev/contracts/contract_config.py @@ -50,6 +50,7 @@ class ContractOutputType(str, Enum): Credentials: str = "credentials" Username: str = "username" Share: str = "share" + File: str = "file" AdminUsername: str = "admin_username" Group: str = "group" Computer: str = "computer" diff --git a/test/contracts/test_contract_output_types.py b/test/contracts/test_contract_output_types.py new file mode 100644 index 0000000..42c99ca --- /dev/null +++ b/test/contracts/test_contract_output_types.py @@ -0,0 +1,15 @@ +import unittest + +from pyoaev.contracts.contract_config import ContractOutputType + + +class ContractOutputTypeTest(unittest.TestCase): + def test_file_wire_label(self): + # The wire label is a public contract shared with the platform enum and every + # injector that declares a `file` output; it must stay exactly "file". + self.assertEqual(ContractOutputType.File.value, "file") + self.assertEqual(ContractOutputType.File, "file") + + +if __name__ == "__main__": + unittest.main()