Estoy tratando de hacer una prueba simple en Python, pero no puedo entender cómo lograr el proceso de burla.

Este es el código de clase y definición:

class FileRemoveOp(...)
    @apply_defaults
    def __init__(
            self,
            source_conn_keys,
            source_conn_id='conn_default',
            *args, **kwargs):
        super(v4FileRemoveOperator, self).__init__(*args, **kwargs)
        self.source_conn_keys = source_conn_keys
        self.source_conn_id = source_conn_id


    def execute (self, context)
          source_conn = Connection(conn_id)
          try:
              for source_conn_key in self.source_keys:
                  if not source_conn.check_for_key(source_conn_key):    
                      logging.info("The source key does not exist")  
                  source_conn.remove_file(source_conn_key,'')
          finally:
              logging.info("Remove operation successful.")

Y esta es mi prueba para la función de ejecución:

@mock.patch('main.Connection')
def test_remove_execute(self,MockConn):
    mock_coon = MockConn.return_value
    mock_coon.value = #I'm not sure what to put here#
    remove_operator = FileRemoveOp(...)
    remove_operator.execute(self)

Desde elejecutar Intento hacer una conexión, necesito burlarme de eso, no quiero hacer una conexión real, solo devolver algo simulado. ¿Cómo puedo hacer eso? Estoy acostumbrado a hacer pruebas en Java, pero nunca lo hice en Python.