chore: factory init (#169)

yearn · Jun 5, 2023 · 3b0999d · 3b0999d
1 parent 3e51c08
commit 3b0999d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/contracts/VaultFactory.vy b/contracts/VaultFactory.vy
@@ -29,7 +29,6 @@
     The protocol fees will be sent to the designated fee_recipient and
     then (X - protocol_fees) will be sent to the vault/strategy specific
     fee recipient.
-
 """
 
 from vyper.interfaces import ERC20
@@ -96,10 +95,10 @@ custom_protocol_fee: public(HashMap[address, uint16])
 use_custom_protocol_fee: public(HashMap[address, bool])
 
 @external
-def __init__(name: String[64], vault_blueprint: address):
+def __init__(name: String[64], vault_blueprint: address, governance: address):
     self.name = name
     VAULT_BLUEPRINT = vault_blueprint
-    self.governance = msg.sender
+    self.governance = governance
 
 @external
 def deploy_new_vault(

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -159,7 +159,10 @@ def vault_blueprint(project, gov):
 @pytest.fixture(scope="session")
 def vault_factory(project, gov, vault_blueprint):
     return gov.deploy(
-        project.VaultFactory, "Vault V3 Factory 3.0.1-beta", vault_blueprint
+        project.VaultFactory,
+        "Vault V3 Factory 3.0.1-beta",
+        vault_blueprint,
+        gov.address,
     )