def METHOD_NAME(self, mock_cluster_repo):
    job_state_data = {"1": "things", "2": "things"}
    mesos_state_data = {"3": "things", "4": "things"}
    state_data = {
        "mesos_state": mesos_state_data,
        "job_state": job_state_data,
    }
    self.mcp.state_watcher.restore.return_value = state_data
    action_runner = mock.Mock()
    self.mcp.restore_state(action_runner)
    mock_cluster_repo.restore_state.assert_called_with(mesos_state_data)
    self.mcp.jobs.restore_state.assert_called_with(
        job_state_data, action_runner,
    )