def test_change_fallback_handler(self):
safe_operator = self.setup_operator()
safe = Safe(safe_operator.address, self.ethereum_client)
current_fallback_handler = safe.retrieve_fallback_handler()
with self.assertRaises(SameFallbackHandlerException):
safe_operator.change_fallback_handler(current_fallback_handler)
new_fallback_handler = Account.create().address
self.assertTrue(
safe_operator.change_fallback_handler(new_fallback_handler))
self.assertEqual(safe_operator.safe_cli_info.fallback_handler,
new_fallback_handler)
self.assertEqual(safe.retrieve_fallback_handler(),
new_fallback_handler)
safe_operator.change_master_copy(self.safe_old_contract_address)
with self.assertRaises(FallbackHandlerNotSupportedException):
safe_operator.change_fallback_handler(Account.create().address)
