container/tools/actions/review_change.py

# Copyright (C) 2019 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import random

import requests

from . import abstract
from .query_changes import QueryChangesAction
from .query_change_files import QueryChangeFilesAction

# pylint: disable=W0703
class ReviewChangeAction(abstract.AbstractAction):
    def __init__(self, url, user, pwd, probability=1):
        super().__init__(url, user, pwd, probability)
        self.change_id = self._get_change_id()
        self.revision_id = 1

    def _execute_action(self):
        rest_url = self._assemble_review_url()
        requests.post(rest_url, auth=(self.user, self.pwd), json=self._assemble_body())
        self.was_executed = True
        self._log_result()

    def _get_change_id(self):
        try:
            return QueryChangesAction(self.url, self.user, self.pwd, 1.0).execute()[
                "change_id"
            ]
        except Exception:
            return None

    def _assemble_review_url(self):
        return "%s/a/changes/%s/revisions/%s/review" % (
            self.url,
            self.change_id,
            self.revision_id,
        )

    def _assemble_body(self):
        file_to_comment = random.choice(self._list_files())
        label = random.randint(-2, 2)
        return {
            "tag": "loadtest",
            "message": "Yet another comment.",
            "labels": {"Code-Review": label},
            "comments": {file_to_comment: [{"line": 1, "message": "Gibberish!"}]},
        }

    def _list_files(self):
        return QueryChangeFilesAction(
            self.change_id, self.url, self.user, self.pwd, 1.0
        ).execute()