Test-TP_Integration/buildprbody_test.py at main · SushTP/Test-TP_Integration · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52


__author__ = "partheniou@google.com (Anthonios Partheniou)"

import pathlib
import shutil
import unittest

from buildprbody import BuildPrBody
from changesummary import ChangeType

SCRIPTS_DIR = pathlib.Path(__file__).parent.resolve()
CHANGE_SUMMARY_DIR = SCRIPTS_DIR / "test_resources" / "buildprbody_resources"

EXPECTED_PR_BODY_OUTPUT = """\
## Deleted keys were detected in the following stable discovery artifacts:
bigquery v2 https://github.com/googleapis/google-api-python-client/commit/123
cloudtasks v2 https://github.com/googleapis/google-api-python-client/commit/456

## Discovery Artifact Change Summary:
feat(bigquery): update the api https://github.com/googleapis/google-api-python-client/commit/123
feat(cloudtasks): update the api https://github.com/googleapis/google-api-python-client/commit/456
feat(drive): update the api https://github.com/googleapis/google-api-python-client/commit/789
"""


class TestBuildPrBody(unittest.TestCase):
    def setUp(self):
        self.buildprbody = BuildPrBody(change_summary_directory=CHANGE_SUMMARY_DIR)

    def test_get_commit_uri_returns_correct_string(self):
        base_uri = "https://github.com/googleapis/google-api-python-client/commit/"

        expected_uri = "".join([base_uri, "123"])
        result = self.buildprbody.get_commit_uri(name="bigquery")
        self.assertEqual(result, expected_uri)

        expected_uri = "".join([base_uri, "456"])
        result = self.buildprbody.get_commit_uri(name="cloudtasks")
        self.assertEqual(result, expected_uri)

        expected_uri = "".join([base_uri, "789"])
        result = self.buildprbody.get_commit_uri(name="drive")
        self.assertEqual(result, expected_uri)

    def test_generate_pr_body(self):
        self.buildprbody.generate_pr_body()

        with open(CHANGE_SUMMARY_DIR / "allapis.summary") as f:
            pr_body = "".join(f.readlines())
        self.assertEqual(pr_body, EXPECTED_PR_BODY_OUTPUT)