def test_update_in_nested_three_levels(_, set_org_repo_kwargs):
selection = _.blah[_.baz.foo[_.bar.repository(owner="testrepoowner",
name="testreponame")[_.id]]]
expected_selection = _.blah[_.baz.foo[_.bar.repository(
owner="owner", name="repo_name")[_.id]]]
assert (m.update_in(selection, ["blah", "foo", "repository"],
set_org_repo_kwargs) == expected_selection)
def add_org_repo_to_repository_field(
selection: m.quiz.Selection, ) -> m.quiz.Selection:
return m.update_in(
selection,
["repository"],
functools.partial(m.upsert_kwargs, "repository",
dict(owner="owner", name="repo_name")),
)
def test_update_in_top_level_by_index(_, set_org_repo_kwargs):
selection = _.foo.repository(owner="testrepoowner",
name="testreponame")[_.id]
assert (m.update_in(selection, [1],
set_org_repo_kwargs) == _.foo.repository(
owner="owner", name="repo_name")[_.id])
def test_update_in_nested_by_name_then_index(_, set_org_repo_kwargs):
selection = _.foo[_.bar.repository(owner="testrepoowner",
name="testreponame")[_.id]]
assert (m.update_in(selection, ["foo", 1],
set_org_repo_kwargs) == _.foo[_.repository(
owner="owner", name="repo_name")[_.id]])