def set_username(username: Optional[str] = None, prompt: Optional[str] = None) -> str: """Sets the username in the registry. Optionally prompts if there is no username supplied as a parameter. Args: username: Value to set as the username in registry. prompt: Custom string to append to username prompt. Returns: username: The determined username. Raises: Error: Failed to set username in registry. """ if not username: username = interact.GetUsername(prompt) try: registry.set_value('Username', username, path=constants.REG_ROOT) except registry.Error as e: raise Error(e) return username
def testGetUsername(self, raw): raw.side_effect = iter(['invalid-name', '', ' ', 'username1']) self.assertEqual(interact.GetUsername(), 'username1')
def _SetUsername(self): self._username = interact.GetUsername()
def testGetUsernamePurpose(self, prompt): prompt.return_value = 'username1' self.assertEqual(interact.GetUsername('domain join'), 'username1') prompt.assert_called_with( 'Please enter your username for domain join: ', validator='^[a-zA-Z0-9]+$')