From abc24eba8f039a0cb174ceeaf9991a0ed876407c Mon Sep 17 00:00:00 2001
From: Kieran Prasch <kieran@nucypher.com>
Date: Wed, 14 Oct 2020 20:56:13 -0700
Subject: [PATCH] Assertion for staking interface router ownership transfer via
 CLI.

---
 .../cli/deploy/test_deploy_cli_commands.py    | 20 +++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/tests/acceptance/cli/deploy/test_deploy_cli_commands.py b/tests/acceptance/cli/deploy/test_deploy_cli_commands.py
index a6287d2d6..a6a18cf11 100644
--- a/tests/acceptance/cli/deploy/test_deploy_cli_commands.py
+++ b/tests/acceptance/cli/deploy/test_deploy_cli_commands.py
@@ -32,7 +32,7 @@ from nucypher.blockchain.eth.constants import (
     POLICY_MANAGER_CONTRACT_NAME,
     STAKING_ESCROW_CONTRACT_NAME
 )
-from nucypher.blockchain.eth.deployers import StakingEscrowDeployer
+from nucypher.blockchain.eth.deployers import StakingEscrowDeployer, StakingInterfaceDeployer
 from nucypher.blockchain.eth.registry import InMemoryContractRegistry, LocalContractRegistry
 from nucypher.cli.commands.deploy import deploy
 from nucypher.config.constants import TEMPORARY_DOMAIN
@@ -160,11 +160,16 @@ def test_transfer_ownership(click_runner, testerchain, agency_local_registry):
     assert policy_agent.owner == testerchain.etherbase_account
     assert adjudicator_agent.owner == testerchain.etherbase_account
 
-    #### BEGIN HOTFIX TEST
+    # Test transfer ownersh
+
+
+def test_transfer_ownership_staking_interface_router(click_runner, testerchain, agency_local_registry):
+
+    maclane = testerchain.unassigned_accounts[0]
 
     ownership_command = ('transfer-ownership',
                          '--registry-infile', agency_local_registry.filepath,
-                         '--contract-name', 'StakingInterface',
+                         '--contract-name', StakingInterfaceDeployer.contract_name,
                          '--provider', TEST_PROVIDER_URI,
                          '--network', TEMPORARY_DOMAIN,
                          '--target-address', maclane,
@@ -180,14 +185,9 @@ def test_transfer_ownership(click_runner, testerchain, agency_local_registry):
                                  catch_exceptions=False)
     assert result.exit_code == 0, result.output
 
-    # All of these are unchanged
-    assert staking_agent.owner == michwill
-    assert policy_agent.owner == testerchain.etherbase_account
-    assert adjudicator_agent.owner == testerchain.etherbase_account
-
-    # TODO: Assert the new owner is correct
     # This owner is updated
-    # assert interface_agent.owner == maclane
+    interface_deployer = StakingInterfaceDeployer(registry=agency_local_registry)
+    assert interface_deployer.owner == maclane
 
 
 def test_bare_contract_deployment_to_alternate_registry(click_runner, agency_local_registry):