def test_get_access_token__unknown_user(self, Command):
sf = mock.Mock()
sf.query_all.return_value = {"records": []}
config = ScratchOrgConfig({}, "test")
with mock.patch("cumulusci.core.config.OrgConfig.salesforce_client", sf):
with pytest.raises(
SfdxOrgException,
match="Couldn't find a username for the specified user",
):
config.get_access_token(alias="dadvisor")
def test_get_access_token__no_access_token(self, Command):
sf = mock.Mock()
sf.query_all.return_value = {"records": [{"Username": "******"}]}
sfdx_response = mock.Mock(returncode=1)
sfdx_response.stdout_text.read.return_value = '{"message": "blah blah..."}'
sfdx = mock.Mock(return_value=sfdx_response)
config = ScratchOrgConfig({}, "test")
with mock.patch("cumulusci.core.config.OrgConfig.salesforce_client", sf):
with mock.patch("cumulusci.core.config.sfdx_org_config.sfdx", sfdx):
exception = (
"Unable to find access token for [email protected]\nblah blah..."
)
with pytest.raises(SfdxOrgException, match=exception):
config.get_access_token(alias="dadvisor")
def test_get_access_token__default(self, Command):
"""Verify that with no args, get_access_token returns the default token"""
config = ScratchOrgConfig({}, "test")
_marker = object()
config._sfdx_info = {"access_token": _marker}
access_token = config.get_access_token()
assert access_token is _marker
def test_get_access_token__multiple_users(self, Command):
sf = mock.Mock()
sf.query_all.return_value = {
"records": [
{"Username": "******"},
{"Username": "******"},
]
}
config = ScratchOrgConfig({}, "test")
with mock.patch("cumulusci.core.config.OrgConfig.salesforce_client", sf):
with pytest.raises(
SfdxOrgException,
match="More than one user matched the search critiera.",
):
config.get_access_token(alias="dadvisor")
def test_get_access_token(self, Command):
"""Verify that get_access_token calls out to sfdx"""
sf = mock.Mock()
sf.query_all.return_value = {"records": [{"Username": "******"}]}
sfdx_response = mock.Mock(returncode=0)
sfdx_response.stdout_text.read.return_value = (
'{"result": {"accessToken": "the-token"}}'
)
sfdx = mock.Mock(return_value=sfdx_response)
config = ScratchOrgConfig({}, "test")
with mock.patch("cumulusci.core.config.OrgConfig.salesforce_client", sf):
with mock.patch("cumulusci.core.config.sfdx_org_config.sfdx", sfdx):
access_token = config.get_access_token(alias="dadvisor")
sfdx.assert_called_once_with(
"force:org:display [email protected] --json"
)
assert access_token == "the-token"
