def test_calculate_landable_subgraphs_extra_check(phabdouble):
phab = phabdouble.get_phabricator_client()
repo = phabdouble.repo()
r1 = phabdouble.revision(repo=repo)
r2 = phabdouble.revision(repo=repo, depends_on=[r1])
r3 = phabdouble.revision(repo=repo, depends_on=[r2])
r4 = phabdouble.revision(repo=repo, depends_on=[r3])
nodes, edges = build_stack_graph(phab, r1['phid'])
ext_data = request_extended_revision_data(
phab, [r1['phid'], r2['phid'], r3['phid'], r4['phid']])
REASON = "Blocked by custom check."
def custom_check(*, revision, diff, repo):
return REASON if revision['id'] == r3['id'] else None
landable, blocked = calculate_landable_subgraphs(
ext_data, edges, {repo['phid']}, other_checks=[custom_check])
assert landable == [
[r1['phid'], r2['phid']],
]
assert r3['phid'] in blocked and r4['phid'] in blocked
assert blocked[r3['phid']] == REASON
def _assess_transplant_request(phab, landing_path):
nodes, edges = _find_stack_from_landing_path(phab, landing_path)
stack_data = request_extended_revision_data(phab, [phid for phid in nodes])
landing_path = _convert_path_id_to_phid(landing_path, stack_data)
supported_repos = get_repos_for_env(current_app.config.get('ENVIRONMENT'))
landable_repos = get_landable_repos_for_revision_data(
stack_data, supported_repos)
landable, blocked = calculate_landable_subgraphs(
stack_data,
edges,
landable_repos,
other_checks=DEFAULT_OTHER_BLOCKER_CHECKS)
assessment = check_landing_blockers(
g.auth0_user,
landing_path,
stack_data,
landable,
landable_repos,
)
if assessment.blocker is not None:
return (assessment, None, None, None)
# We have now verified that landable_path is valid and is indeed
# landable (in the sense that it is a landable_subgraph, with no
# revisions being blocked). Make this clear by using a different
# value, and assume it going forward.
valid_path = landing_path
# Now that we know this is a valid path we can convert it into a list
# of (revision, diff) tuples.
to_land = [stack_data.revisions[r_phid] for r_phid, _ in valid_path]
to_land = [
(r, stack_data.diffs[PhabricatorClient.expect(r, 'fields',
'diffPHID')])
for r in to_land
]
# To be a landable path the entire path must have the same
# repository, so we can get away with checking only one.
repo = stack_data.repositories[to_land[0][0]['fields']['repositoryPHID']]
landing_repo = landable_repos[repo['phid']]
involved_phids = set()
for revision, _ in to_land:
involved_phids.update(gather_involved_phids(revision))
involved_phids = list(involved_phids)
users = lazy_user_search(phab, involved_phids)()
projects = lazy_project_search(phab, involved_phids)()
reviewers = {
revision['phid']: get_collated_reviewers(revision)
for revision, _ in to_land
}
assessment = check_landing_warnings(g.auth0_user, to_land, repo,
landing_repo, reviewers, users,
projects)
return (assessment, to_land, landing_repo, stack_data)
def test_calculate_landable_subgraphs_diverging_paths_merge(phabdouble):
phab = phabdouble.get_phabricator_client()
repo = phabdouble.repo()
r1 = phabdouble.revision(repo=repo)
r2 = phabdouble.revision(repo=repo, depends_on=[r1])
r3 = phabdouble.revision(repo=repo, depends_on=[r2])
r4 = phabdouble.revision(repo=repo, depends_on=[r1])
r5 = phabdouble.revision(repo=repo, depends_on=[r4])
r6 = phabdouble.revision(repo=repo, depends_on=[r1])
r7 = phabdouble.revision(repo=repo, depends_on=[r3, r5, r6])
nodes, edges = build_stack_graph(phab, r1['phid'])
ext_data = request_extended_revision_data(phab, [
r1['phid'],
r2['phid'],
r3['phid'],
r4['phid'],
r5['phid'],
r6['phid'],
r7['phid'],
])
landable, _ = calculate_landable_subgraphs(ext_data, edges, {repo['phid']})
assert len(landable) == 3
assert [r1['phid'], r2['phid'], r3['phid']] in landable
assert [r1['phid'], r4['phid'], r5['phid']] in landable
assert [r1['phid'], r6['phid']] in landable
def test_calculate_landable_subgraphs_allows_distinct_repo_paths(phabdouble):
phab = phabdouble.get_phabricator_client()
repo1 = phabdouble.repo(name='repo1')
r1 = phabdouble.revision(repo=repo1)
r2 = phabdouble.revision(repo=repo1, depends_on=[r1])
repo2 = phabdouble.repo(name='repo2')
r3 = phabdouble.revision(repo=repo2)
r4 = phabdouble.revision(repo=repo2, depends_on=[r3])
r5 = phabdouble.revision(repo=repo1, depends_on=[r2, r4])
nodes, edges = build_stack_graph(phab, r1['phid'])
ext_data = request_extended_revision_data(phab, [
r1['phid'],
r2['phid'],
r3['phid'],
r4['phid'],
r5['phid'],
])
landable, _ = calculate_landable_subgraphs(ext_data, edges,
{repo1['phid'], repo2['phid']})
assert len(landable) == 2
assert [r1['phid'], r2['phid']] in landable
assert [r3['phid'], r4['phid']] in landable
def test_calculate_landable_subgraphs_no_edges_closed(phabdouble):
phab = phabdouble.get_phabricator_client()
repo = phabdouble.repo()
revision = phabdouble.revision(repo=repo, status=RevisionStatus.PUBLISHED)
ext_data = request_extended_revision_data(phab, [revision['phid']])
landable, _ = calculate_landable_subgraphs(ext_data, [], {repo['phid']})
assert not landable
def test_calculate_landable_subgraphs_no_edges_open(phabdouble):
phab = phabdouble.get_phabricator_client()
repo = phabdouble.repo()
revision = phabdouble.revision(repo=repo)
ext_data = request_extended_revision_data(phab, [revision['phid']])
landable, _ = calculate_landable_subgraphs(ext_data, [], {repo['phid']})
assert len(landable) == 1
assert landable[0] == [revision['phid']]
def test_calculate_landable_subgraphs_closed_root(phabdouble):
phab = phabdouble.get_phabricator_client()
repo = phabdouble.repo()
r1 = phabdouble.revision(repo=repo, status=RevisionStatus.PUBLISHED)
r2 = phabdouble.revision(repo=repo, depends_on=[r1])
nodes, edges = build_stack_graph(phab, r1["phid"])
ext_data = request_extended_revision_data(phab, [r1["phid"], r2["phid"]])
landable, _ = calculate_landable_subgraphs(ext_data, edges, {repo["phid"]})
assert landable == [[r2["phid"]]]
def test_calculate_landable_subgraphs_stops_multiple_repo_paths(phabdouble):
phab = phabdouble.get_phabricator_client()
repo1 = phabdouble.repo(name="repo1")
repo2 = phabdouble.repo(name="repo2")
r1 = phabdouble.revision(repo=repo1)
r2 = phabdouble.revision(repo=repo1, depends_on=[r1])
r3 = phabdouble.revision(repo=repo2, depends_on=[r2])
nodes, edges = build_stack_graph(phab, r1["phid"])
ext_data = request_extended_revision_data(
phab, [r1["phid"], r2["phid"], r3["phid"]])
landable, _ = calculate_landable_subgraphs(ext_data, edges,
{repo1["phid"], repo2["phid"]})
assert landable == [[r1["phid"], r2["phid"]]]
def test_calculate_landable_subgraphs_different_repo_closed_parent(phabdouble):
phab = phabdouble.get_phabricator_client()
repo1 = phabdouble.repo(name="repo1")
r1 = phabdouble.revision(repo=repo1, status=RevisionStatus.PUBLISHED)
repo2 = phabdouble.repo(name="repo2")
r2 = phabdouble.revision(repo=repo2)
r3 = phabdouble.revision(repo=repo2, depends_on=[r1, r2])
nodes, edges = build_stack_graph(phab, r1["phid"])
ext_data = request_extended_revision_data(
phab, [r1["phid"], r2["phid"], r3["phid"]])
landable, _ = calculate_landable_subgraphs(ext_data, edges,
{repo1["phid"], repo2["phid"]})
assert len(landable) == 1
assert [r2["phid"], r3["phid"]] in landable
def test_calculate_landable_subgraphs_missing_repo(phabdouble):
"""Test to assert a missing repository for a revision is
blocked with an appropriate error
"""
phab = phabdouble.get_phabricator_client()
repo1 = phabdouble.repo()
r1 = phabdouble.revision(repo=None)
nodes, edges = build_stack_graph(phab, r1["phid"])
revision_data = request_extended_revision_data(phab, [r1["phid"]])
landable, blocked = calculate_landable_subgraphs(revision_data, edges,
{repo1["phid"]})
repo_unset_warning = ("Revision's repository unset. Specify a target using"
'"Edit revision" in Phabricator')
assert not landable
assert r1["phid"] in blocked
assert blocked[r1["phid"]] == repo_unset_warning
def test_calculate_landable_subgraphs_closed_root_child_merges(phabdouble):
phab = phabdouble.get_phabricator_client()
repo = phabdouble.repo()
r1 = phabdouble.revision(repo=repo)
r2 = phabdouble.revision(repo=repo, depends_on=[r1])
r3 = phabdouble.revision(repo=repo, status=RevisionStatus.PUBLISHED)
r4 = phabdouble.revision(repo=repo, depends_on=[r2, r3])
nodes, edges = build_stack_graph(phab, r1['phid'])
ext_data = request_extended_revision_data(phab, [
r1['phid'],
r2['phid'],
r3['phid'],
r4['phid'],
])
landable, _ = calculate_landable_subgraphs(ext_data, edges, {repo['phid']})
assert [r3['phid']] not in landable
assert [r3['phid'], r4['phid']] not in landable
assert [r4['phid']] not in landable
assert landable == [[r1['phid'], r2['phid'], r4['phid']]]
def test_calculate_landable_subgraphs_complex_graph(phabdouble):
phab = phabdouble.get_phabricator_client()
repoA = phabdouble.repo(name='repoA')
repoB = phabdouble.repo(name='repoB')
repoC = phabdouble.repo(name='repoC')
# Revision stack to construct:
# * rB4
# |\
# | * rB3
# * | rB2 (CLOSED)
# | * rC1
# |/
# * rA10
# /|\
# * | | rB1
# | * rA9 (CLOSED)
# * rA8
# | * rA7
# |/
# * rA6
# /|
# | * rA5
# | * rA4
# * |\ rA3 (CLOSED)
# | * | rA2
# \|/
# * rA1 (CLOSED)
rA1 = phabdouble.revision(repo=repoA, status=RevisionStatus.PUBLISHED)
rA2 = phabdouble.revision(repo=repoA, depends_on=[rA1])
rA3 = phabdouble.revision(repo=repoA,
status=RevisionStatus.PUBLISHED,
depends_on=[rA1])
rA4 = phabdouble.revision(repo=repoA, depends_on=[rA1, rA2])
rA5 = phabdouble.revision(repo=repoA, depends_on=[rA4])
rA6 = phabdouble.revision(repo=repoA, depends_on=[rA3, rA5])
rA7 = phabdouble.revision(repo=repoA, depends_on=[rA6])
rA8 = phabdouble.revision(repo=repoA, depends_on=[rA6])
rA9 = phabdouble.revision(repo=repoA, status=RevisionStatus.PUBLISHED)
rB1 = phabdouble.revision(repo=repoB)
rA10 = phabdouble.revision(repo=repoA, depends_on=[rA8, rA9, rB1])
rC1 = phabdouble.revision(repo=repoC, depends_on=[rA10])
rB2 = phabdouble.revision(repo=repoB, status=RevisionStatus.PUBLISHED)
rB3 = phabdouble.revision(repo=repoB, depends_on=[rA10])
rB4 = phabdouble.revision(repo=repoB, depends_on=[rB2, rB3])
nodes, edges = build_stack_graph(phab, rA1['phid'])
ext_data = request_extended_revision_data(phab, [
rA1['phid'],
rA2['phid'],
rA3['phid'],
rA4['phid'],
rA5['phid'],
rA6['phid'],
rA7['phid'],
rA8['phid'],
rA9['phid'],
rA10['phid'],
rB1['phid'],
rB2['phid'],
rB3['phid'],
rB4['phid'],
rC1['phid'],
])
landable, _ = calculate_landable_subgraphs(ext_data, edges,
{repoA['phid'], repoB['phid']})
assert len(landable) == 3
assert [
rA2['phid'],
rA4['phid'],
rA5['phid'],
rA6['phid'],
rA7['phid'],
] in landable
assert [
rA2['phid'],
rA4['phid'],
rA5['phid'],
rA6['phid'],
rA8['phid'],
] in landable
assert [rB1['phid']] in landable
def get(revision_id):
"""Get the stack a revision is part of.
Args:
revision_id: (string) ID of the revision in 'D{number}' format
"""
revision_id = revision_id_to_int(revision_id)
phab = g.phabricator
revision = phab.call_conduit("differential.revision.search",
constraints={"ids": [revision_id]})
revision = phab.single(revision, "data", none_when_empty=True)
if revision is None:
return problem(
404,
"Revision not found",
"The requested revision does not exist",
type="https://developer.mozilla.org/en-US/docs/Web/HTTP/Status/404",
)
# TODO: This assumes that all revisions and related objects in the stack
# have uniform view permissions for the requesting user. Some revisions
# being restricted could cause this to fail.
nodes, edges = build_stack_graph(phab, phab.expect(revision, "phid"))
stack_data = request_extended_revision_data(phab, [phid for phid in nodes])
supported_repos = get_repos_for_env(current_app.config.get("ENVIRONMENT"))
landable_repos = get_landable_repos_for_revision_data(
stack_data, supported_repos)
other_checks = get_blocker_checks(
repositories=supported_repos,
relman_group_phid=get_relman_group_phid(phab))
landable, blocked = calculate_landable_subgraphs(stack_data,
edges,
landable_repos,
other_checks=other_checks)
uplift_repos = [
name for name, repo in supported_repos.items()
if repo.approval_required
]
involved_phids = set()
for revision in stack_data.revisions.values():
involved_phids.update(gather_involved_phids(revision))
involved_phids = list(involved_phids)
users = user_search(phab, involved_phids)
projects = project_search(phab, involved_phids)
secure_project_phid = get_secure_project_phid(phab)
sec_approval_project_phid = get_sec_approval_project_phid(phab)
revisions_response = []
for _phid, revision in stack_data.revisions.items():
revision_phid = PhabricatorClient.expect(revision, "phid")
fields = PhabricatorClient.expect(revision, "fields")
diff_phid = PhabricatorClient.expect(fields, "diffPHID")
diff = stack_data.diffs[diff_phid]
human_revision_id = "D{}".format(
PhabricatorClient.expect(revision, "id"))
revision_url = urllib.parse.urljoin(
current_app.config["PHABRICATOR_URL"], human_revision_id)
secure = revision_is_secure(revision, secure_project_phid)
commit_description = find_title_and_summary_for_display(
phab, revision, secure)
bug_id = get_bugzilla_bug(revision)
reviewers = get_collated_reviewers(revision)
accepted_reviewers = reviewers_for_commit_message(
reviewers, users, projects, sec_approval_project_phid)
commit_message_title, commit_message = format_commit_message(
commit_description.title,
bug_id,
accepted_reviewers,
commit_description.summary,
revision_url,
)
author_response = serialize_author(phab.expect(fields, "authorPHID"),
users)
revisions_response.append({
"id":
human_revision_id,
"phid":
revision_phid,
"status":
serialize_status(revision),
"blocked_reason":
blocked.get(revision_phid, ""),
"bug_id":
bug_id,
"title":
commit_description.title,
"url":
revision_url,
"date_created":
PhabricatorClient.to_datetime(
PhabricatorClient.expect(revision, "fields",
"dateCreated")).isoformat(),
"date_modified":
PhabricatorClient.to_datetime(
PhabricatorClient.expect(revision, "fields",
"dateModified")).isoformat(),
"summary":
commit_description.summary,
"commit_message_title":
commit_message_title,
"commit_message":
commit_message,
"repo_phid":
PhabricatorClient.expect(fields, "repositoryPHID"),
"diff":
serialize_diff(diff),
"author":
author_response,
"reviewers":
serialize_reviewers(reviewers, users, projects, diff_phid),
"is_secure":
secure,
"is_using_secure_commit_message":
commit_description.sanitized,
})
repositories = []
for phid in stack_data.repositories.keys():
short_name = PhabricatorClient.expect(stack_data.repositories[phid],
"fields", "shortName")
repo = supported_repos.get(short_name)
if repo is None:
landing_supported, approval_required = False, None
else:
landing_supported, approval_required = True, repo.approval_required
url = ("{phabricator_url}/source/{short_name}".format(
phabricator_url=current_app.config["PHABRICATOR_URL"],
short_name=short_name,
) if not landing_supported else supported_repos[short_name].url)
repositories.append({
"phid": phid,
"short_name": short_name,
"url": url,
"landing_supported": landing_supported,
"approval_required": approval_required,
})
return {
"repositories": repositories,
"revisions": revisions_response,
"edges": [e for e in edges],
"landable_paths": landable,
"uplift_repositories": uplift_repos,
}
def get(revision_id):
"""Get the stack a revision is part of.
Args:
revision_id: (string) ID of the revision in 'D{number}' format
"""
revision_id = revision_id_to_int(revision_id)
phab = g.phabricator
revision = phab.call_conduit(
'differential.revision.search',
constraints={'ids': [revision_id]},
)
revision = phab.single(revision, 'data', none_when_empty=True)
if revision is None:
return problem(
404,
'Revision not found',
'The requested revision does not exist',
type='https://developer.mozilla.org/en-US/docs/Web/HTTP/Status/404'
)
# TODO: This assumes that all revisions and related objects in the stack
# have uniform view permissions for the requesting user. Some revisions
# being restricted could cause this to fail.
nodes, edges = build_stack_graph(phab, phab.expect(revision, 'phid'))
stack_data = request_extended_revision_data(phab, [phid for phid in nodes])
supported_repos = get_repos_for_env(current_app.config.get('ENVIRONMENT'))
landable_repos = get_landable_repos_for_revision_data(
stack_data, supported_repos
)
landable, blocked = calculate_landable_subgraphs(
stack_data,
edges,
landable_repos,
other_checks=DEFAULT_OTHER_BLOCKER_CHECKS
)
involved_phids = set()
for revision in stack_data.revisions.values():
involved_phids.update(gather_involved_phids(revision))
involved_phids = list(involved_phids)
users = lazy_user_search(phab, involved_phids)()
projects = lazy_project_search(phab, involved_phids)()
revisions_response = []
for phid, revision in stack_data.revisions.items():
revision_phid = PhabricatorClient.expect(revision, 'phid')
fields = PhabricatorClient.expect(revision, 'fields')
diff_phid = PhabricatorClient.expect(fields, 'diffPHID')
diff = stack_data.diffs[diff_phid]
human_revision_id = 'D{}'.format(
PhabricatorClient.expect(revision, 'id')
)
revision_url = urllib.parse.urljoin(
current_app.config['PHABRICATOR_URL'], human_revision_id
)
title = PhabricatorClient.expect(fields, 'title')
summary = PhabricatorClient.expect(fields, 'summary')
bug_id = get_bugzilla_bug(revision)
reviewers = get_collated_reviewers(revision)
accepted_reviewers = [
reviewer_identity(phid, users, projects).identifier
for phid, r in reviewers.items()
if r['status'] is ReviewerStatus.ACCEPTED
]
commit_message_title, commit_message = format_commit_message(
title, bug_id, accepted_reviewers, summary, revision_url
)
author_response = serialize_author(
phab.expect(fields, 'authorPHID'), users
)
revisions_response.append({
'id': human_revision_id,
'phid': revision_phid,
'status': serialize_status(revision),
'blocked_reason': blocked.get(revision_phid, ''),
'bug_id': bug_id,
'title': title,
'url': revision_url,
'date_created': PhabricatorClient.to_datetime(
PhabricatorClient.expect(revision, 'fields', 'dateCreated')
).isoformat(),
'date_modified': PhabricatorClient.to_datetime(
PhabricatorClient.expect(revision, 'fields', 'dateModified')
).isoformat(),
'summary': summary,
'commit_message_title': commit_message_title,
'commit_message': commit_message,
'diff': serialize_diff(diff),
'author': author_response,
'reviewers': serialize_reviewers(
reviewers, users, projects, diff_phid
),
}) # yapf: disable
return {
'revisions': revisions_response,
'edges': [e for e in edges],
'landable_paths': landable,
}
def get(revision_id):
"""Get the stack a revision is part of.
Args:
revision_id: (string) ID of the revision in 'D{number}' format
"""
revision_id = revision_id_to_int(revision_id)
phab = g.phabricator
revision = phab.call_conduit("differential.revision.search",
constraints={"ids": [revision_id]})
revision = phab.single(revision, "data", none_when_empty=True)
if revision is None:
return not_found_problem
try:
nodes, edges = build_stack_graph(phab, phab.expect(revision, "phid"))
except PhabricatorAPIException:
# If a revision within the stack causes an API exception, treat the whole stack
# as not found.
return not_found_problem
stack_data = request_extended_revision_data(phab, [phid for phid in nodes])
supported_repos = get_repos_for_env(current_app.config.get("ENVIRONMENT"))
landable_repos = get_landable_repos_for_revision_data(
stack_data, supported_repos)
other_checks = get_blocker_checks(
repositories=supported_repos,
relman_group_phid=get_relman_group_phid(phab))
landable, blocked = calculate_landable_subgraphs(stack_data,
edges,
landable_repos,
other_checks=other_checks)
uplift_repos = [
name for name, repo in supported_repos.items()
if repo.approval_required
]
involved_phids = set()
for revision in stack_data.revisions.values():
involved_phids.update(gather_involved_phids(revision))
involved_phids = list(involved_phids)
users = user_search(phab, involved_phids)
projects = project_search(phab, involved_phids)
secure_project_phid = get_secure_project_phid(phab)
sec_approval_project_phid = get_sec_approval_project_phid(phab)
revisions_response = []
for _phid, revision in stack_data.revisions.items():
revision_phid = PhabricatorClient.expect(revision, "phid")
fields = PhabricatorClient.expect(revision, "fields")
diff_phid = PhabricatorClient.expect(fields, "diffPHID")
diff = stack_data.diffs[diff_phid]
human_revision_id = "D{}".format(
PhabricatorClient.expect(revision, "id"))
revision_url = urllib.parse.urljoin(
current_app.config["PHABRICATOR_URL"], human_revision_id)
secure = revision_is_secure(revision, secure_project_phid)
commit_description = find_title_and_summary_for_display(
phab, revision, secure)
bug_id = get_bugzilla_bug(revision)
reviewers = get_collated_reviewers(revision)
accepted_reviewers = reviewers_for_commit_message(
reviewers, users, projects, sec_approval_project_phid)
commit_message_title, commit_message = format_commit_message(
commit_description.title,
bug_id,
accepted_reviewers,
commit_description.summary,
revision_url,
)
author_response = serialize_author(phab.expect(fields, "authorPHID"),
users)
revisions_response.append({
"id":
human_revision_id,
"phid":
revision_phid,
"status":
serialize_status(revision),
"blocked_reason":
blocked.get(revision_phid, ""),
"bug_id":
bug_id,
"title":
commit_description.title,
"url":
revision_url,
"date_created":
PhabricatorClient.to_datetime(
PhabricatorClient.expect(revision, "fields",
"dateCreated")).isoformat(),
"date_modified":
PhabricatorClient.to_datetime(
PhabricatorClient.expect(revision, "fields",
"dateModified")).isoformat(),
"summary":
commit_description.summary,
"commit_message_title":
commit_message_title,
"commit_message":
commit_message,
"repo_phid":
PhabricatorClient.expect(fields, "repositoryPHID"),
"diff":
serialize_diff(diff),
"author":
author_response,
"reviewers":
serialize_reviewers(reviewers, users, projects, diff_phid),
"is_secure":
secure,
"is_using_secure_commit_message":
commit_description.sanitized,
})
repositories = []
for phid in stack_data.repositories.keys():
short_name = PhabricatorClient.expect(stack_data.repositories[phid],
"fields", "shortName")
repo = supported_repos.get(short_name)
landing_supported = repo is not None
url = (repo.url if landing_supported else
f"{current_app.config['PHABRICATOR_URL']}/source/{short_name}")
repositories.append({
"approval_required": landing_supported and repo.approval_required,
"commit_flags": repo.commit_flags if repo else [],
"landing_supported": landing_supported,
"phid": phid,
"short_name": short_name,
"url": url,
})
return {
"repositories": repositories,
"revisions": revisions_response,
"edges": [e for e in edges],
"landable_paths": landable,
"uplift_repositories": uplift_repos,
}