Zombar de toda a classe python
Estou tentando fazer um teste simples em python, mas não consigo descobrir como realizar o processo de zombaria.
Este é o código de classe e definição:
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.")
E este é o meu teste para a função execute:
@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 oexecutar método tenta fazer uma conexão, eu preciso zombar disso, eu não quero fazer uma conexão real, basta retornar algo simulado. Como posso fazer isso? Estou acostumado a fazer testes em Java, mas nunca fiz em python ..