Remove test command

Author: Uxio0

safe_transaction_service/history/management/commands/safe_contract.py

@@ -20,7 +20,6 @@
     IndexingStatus,
     InternalTxDecoded,
     ProxyFactory,
-    SafeContract,
     SafeLastStatus,
     SafeMasterCopy,
 )
@@ -739,196 +738,3 @@ def test_validate_tx_integrity(self):
         )
         self.assertNotIn("is not matching", text)
         self.assertNotIn("is not valid for multisig transaction", text)
-
-    @mock.patch(
-        "safe_transaction_service.history.management.commands.safe_contract.get_auto_ethereum_client"
-    )
-    @mock.patch(
-        "safe_transaction_service.history.management.commands.safe_contract.SafeEventsIndexer"
-    )
-    @mock.patch(
-        "safe_transaction_service.history.management.commands.safe_contract.SafeTxProcessorProvider"
-    )
-    def test_safe_contract_add(
-        self,
-        mock_tx_processor_provider: MagicMock,
-        mock_safe_events_indexer: MagicMock,
-        mock_get_ethereum_client: MagicMock,
-    ):
-        """Test adding a SafeContract entry processes events through the indexer."""
-        command = "safe_contract"
-
-        safe_address = Account.create().address
-        tx_hash = "0x" + "ab" * 32
-
-        # Mock the ethereum client and its get_transaction_receipt method
-        mock_client = MagicMock()
-        mock_get_ethereum_client.return_value = mock_client
-        mock_client.get_transaction_receipt.return_value = {
-            "logs": [{"logIndex": 0, "transactionHash": tx_hash}],
-            "transactionHash": tx_hash,
-        }
-
-        # Mock indexer instance
-        mock_indexer_instance = MagicMock()
-        mock_safe_events_indexer.return_value = mock_indexer_instance
-
-        # Mock tx processor
-        mock_tx_processor = MagicMock()
-        mock_tx_processor_provider.return_value = mock_tx_processor
-
-        self.assertEqual(SafeContract.objects.count(), 0)
-
-        # Add SafeContract - command will process through indexer
-        buf = StringIO()
-        call_command(command, "add", safe_address, tx_hash, stdout=buf)
-        output = buf.getvalue()
-
-        # Verify get_transaction_receipt was called
-        mock_client.get_transaction_receipt.assert_called_once()
-
-        # Verify SafeEventsIndexer was created with ignored_initiators=set()
-        mock_safe_events_indexer.assert_called_once()
-        call_kwargs = mock_safe_events_indexer.call_args.kwargs
-        self.assertEqual(call_kwargs.get("ignored_initiators"), set())
-
-        # Verify process_elements was called on the indexer instance
-        mock_indexer_instance.process_elements.assert_called_once_with(
-            [{"logIndex": 0, "transactionHash": tx_hash}]
-        )
-
-        # Verify SafeTxProcessor.process_decoded_transactions was called
-        mock_tx_processor.process_decoded_transactions.assert_called_once()
-
-        self.assertIn("Processing events through SafeEventsIndexer", output)
-        self.assertIn("Processing decoded transactions to create SafeContract", output)
-
-    @mock.patch(
-        "safe_transaction_service.history.management.commands.safe_contract.get_auto_ethereum_client"
-    )
-    def test_safe_contract_add_already_exists(
-        self, mock_get_ethereum_client: MagicMock
-    ):
-        """Test adding a SafeContract that already exists shows warning."""
-        command = "safe_contract"
-
-        safe_contract = SafeContractFactory()
-        safe_address = safe_contract.address
-        tx_hash = "0x" + "ab" * 32
-
-        self.assertEqual(SafeContract.objects.count(), 1)
-
-        # Add same address again (should show warning without fetching)
-        buf = StringIO()
-        call_command(command, "add", safe_address, tx_hash, stdout=buf)
-        output = buf.getvalue()
-        self.assertIn(
-            f"SafeContract already exists for address: {safe_address}", output
-        )
-        self.assertEqual(SafeContract.objects.count(), 1)
-
-        # Should not have called ethereum client
-        mock_get_ethereum_client.return_value.get_transaction_receipt.assert_not_called()
-
-    def test_safe_contract_add_invalid_address(self):
-        """Test adding with invalid address shows error."""
-        command = "safe_contract"
-        tx_hash = "0x" + "ab" * 32
-
-        with self.assertRaises(CommandError) as context:
-            call_command(command, "add", "invalid-address", tx_hash)
-        self.assertIn("Invalid Ethereum address", str(context.exception))
-
-    def test_safe_contract_add_invalid_tx_hash(self):
-        """Test adding with invalid tx hash shows error."""
-        command = "safe_contract"
-        address = Account.create().address
-
-        with self.assertRaises(CommandError) as context:
-            call_command(command, "add", address, "invalid-tx-hash")
-        self.assertIn("Invalid transaction hash", str(context.exception))
-
-    @mock.patch(
-        "safe_transaction_service.history.management.commands.safe_contract.get_auto_ethereum_client"
-    )
-    def test_safe_contract_add_nonexistent_tx(
-        self, mock_get_ethereum_client: MagicMock
-    ):
-        """Test adding with non-existent tx hash shows error when receipt not found."""
-        command = "safe_contract"
-        address = Account.create().address
-        fake_tx_hash = "0x" + "a" * 64
-
-        # Mock the ethereum client to return None for receipt
-        mock_client = MagicMock()
-        mock_get_ethereum_client.return_value = mock_client
-        mock_client.get_transaction_receipt.return_value = None
-
-        with self.assertRaises(CommandError) as context:
-            call_command(command, "add", address, fake_tx_hash)
-        self.assertIn("not found", str(context.exception))
-
-    @mock.patch(
-        "safe_transaction_service.history.management.commands.safe_contract.get_auto_ethereum_client"
-    )
-    def test_safe_contract_add_no_logs(self, mock_get_ethereum_client: MagicMock):
-        """Test adding with tx that has no logs shows error."""
-        command = "safe_contract"
-        address = Account.create().address
-        tx_hash = "0x" + "ab" * 32
-
-        # Mock the ethereum client to return receipt with no logs
-        mock_client = MagicMock()
-        mock_get_ethereum_client.return_value = mock_client
-        mock_client.get_transaction_receipt.return_value = {
-            "logs": [],
-            "transactionHash": tx_hash,
-        }
-
-        with self.assertRaises(CommandError) as context:
-            call_command(command, "add", address, tx_hash)
-        self.assertIn("has no logs", str(context.exception))
-
-    def test_safe_contract_remove(self):
-        """Test removing SafeContract entries."""
-        command = "safe_contract"
-
-        safe_contract1 = SafeContractFactory()
-        safe_contract2 = SafeContractFactory()
-        address1 = safe_contract1.address
-        address2 = safe_contract2.address
-
-        self.assertEqual(SafeContract.objects.count(), 2)
-
-        # Remove single address
-        buf = StringIO()
-        call_command(command, "remove", address1, stdout=buf)
-        output = buf.getvalue()
-        self.assertIn(f"Removed SafeContract: {address1}", output)
-        self.assertIn("1 removed, 0 not found", output)
-        self.assertEqual(SafeContract.objects.count(), 1)
-        self.assertFalse(SafeContract.objects.filter(address=address1).exists())
-
-        # Remove non-existent address
-        buf = StringIO()
-        call_command(command, "remove", address1, stdout=buf)
-        output = buf.getvalue()
-        self.assertIn(f"SafeContract not found: {address1}", output)
-        self.assertIn("0 removed, 1 not found", output)
-
-        # Remove multiple addresses (one exists, one doesn't)
-        buf = StringIO()
-        call_command(command, "remove", address1, address2, stdout=buf)
-        output = buf.getvalue()
-        self.assertIn(f"SafeContract not found: {address1}", output)
-        self.assertIn(f"Removed SafeContract: {address2}", output)
-        self.assertIn("1 removed, 1 not found", output)
-        self.assertEqual(SafeContract.objects.count(), 0)
-
-    def test_safe_contract_no_action(self):
-        """Test that missing action shows error."""
-        command = "safe_contract"
-
-        with self.assertRaises(CommandError) as context:
-            call_command(command)
-        self.assertIn("Please specify an action", str(context.exception))