def test_upload_existing_task(self, mock_assert_owner,
mock_assert_original, mock_clone,
mock_upgrade_graph):
user = UserNode()
task = TaskInst()
target_task = TaskInst()
new_task = user.upload(task, target_task)
self.assertIs(new_task, target_task)
mock_assert_owner.assert_called_once_with(task)
mock_assert_original.assert_called_once()
mock_clone.assert_not_called()
mock_upgrade_graph.assert_called_once()
def test_upload(self, mock_assert_owner, mock_assert_original, mock_clone,
mock_upgrade_graph):
user = UserNode()
task = TaskInst()
new_task = TaskInst()
mock_clone.return_value = new_task
result = user.upload(task)
self.assertTrue(new_task is result)
mock_assert_original.assert_called_once()
mock_assert_owner.assert_called_once_with(task)
mock_clone.assert_called_once()
mock_upgrade_graph.assert_not_called()