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()