Fix tests

Substra · Jul 10, 2020 · 552b836 · 552b836
1 parent 05f2b39
commit 552b836
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/backend/substrapp/tests/tests_tasks.py b/backend/substrapp/tests/tests_tasks.py
@@ -381,7 +381,8 @@ def __init__(self):
                 mock.patch('substrapp.tasks.tasks.uncompress_content'), \
                 mock.patch('substrapp.tasks.tasks.json.loads') as mjson_loads, \
                 mock.patch('substrapp.tasks.tasks.AsyncResult') as masyncres, \
-                mock.patch('substrapp.tasks.tasks.get_owner') as get_owner:
+                mock.patch('substrapp.tasks.tasks.get_owner') as get_owner,\
+                mock.patch('substrapp.tasks.tasks.find_training_step_tuple_from_key') as gettuple:
 
             msettings.return_value = FakeSettings()
             mget_hash.return_value = 'owkinhash'
@@ -392,6 +393,7 @@ def __init__(self):
             mprepare_opener.return_value = 'opener'
             mprepare_data_sample.return_value = 'data'
             get_owner.return_value = 'foo'
+            gettuple.return_value = None, subtuple[0]
 
             masyncres.return_value.state = 'PENDING'