def test_create_session_user(self, generate_session_token_mock, server_caps_mock):
"""
Makes sure that create_session_user does correct input validation.
:param generate_session_token_mock: Mocked so we can skip communicating
with the Shotgun server.
"""
generate_session_token_mock.return_value = "session_token"
# No login should throw
with self.assertRaises(IncompleteCredentials):
ShotgunAuthenticator(TestDefaultManager()).create_session_user("", "session_token")
# No password or session token should throw
with self.assertRaises(IncompleteCredentials):
ShotgunAuthenticator(TestDefaultManager()).create_session_user("login")
# Passing a password should generate a session token
user = ShotgunAuthenticator(TestDefaultManager()).create_session_user(
"login", password="******", host="https://host.shotgunstudio.com"
)
self.assertEquals(generate_session_token_mock.call_count, 1)
self.assertEquals(user.impl.get_session_token(), "session_token")
connection = user.create_sg_connection()
self.assertEqual(connection.config.session_token, "session_token")
def test_create_session_user(self, generate_session_token_mock,
server_caps_mock):
"""
Makes sure that create_session_user does correct input validation.
:param generate_session_token_mock: Mocked so we can skip communicating
with the Shotgun server.
"""
generate_session_token_mock.return_value = "session_token"
# No login should throw
with self.assertRaises(IncompleteCredentials):
ShotgunAuthenticator(TestDefaultManager()).create_session_user(
"", "session_token")
# No password or session token should throw
with self.assertRaises(IncompleteCredentials):
ShotgunAuthenticator(
TestDefaultManager()).create_session_user("login")
# Passing a password should generate a session token
user = ShotgunAuthenticator(TestDefaultManager()).create_session_user(
"login",
password="******",
host="https://host.shotgunstudio.com")
self.assertEquals(generate_session_token_mock.call_count, 1)
self.assertEquals(user.impl.get_session_token(), "session_token")
connection = user.create_sg_connection()
self.assertEqual(connection.config.session_token, "session_token")
def test_create_script_user(self, server_caps_mock):
"""
Makes sure that create_script_user does correct input validation.
"""
# No script name should throw
with self.assertRaises(IncompleteCredentials):
ShotgunAuthenticator(TestDefaultManager()).create_script_user("", "api_key")
# No script key should throw
with self.assertRaises(IncompleteCredentials):
ShotgunAuthenticator(TestDefaultManager()).create_script_user("api_script", "")
# With valid values it should work
user = ShotgunAuthenticator(TestDefaultManager()).create_script_user(
"api_script", "api_key", "https://host.shotgunstudio.com", None
)
connection = user.create_sg_connection()
self.assertEqual(connection.config.script_name, "api_script")
self.assertEqual(connection.config.api_key, "api_key")
def test_create_script_user(self, server_caps_mock):
"""
Makes sure that create_script_user does correct input validation.
"""
# No script name should throw
with self.assertRaises(IncompleteCredentials):
ShotgunAuthenticator(TestDefaultManager()).create_script_user(
"", "api_key")
# No script key should throw
with self.assertRaises(IncompleteCredentials):
ShotgunAuthenticator(TestDefaultManager()).create_script_user(
"api_script", "")
# With valid values it should work
user = ShotgunAuthenticator(TestDefaultManager()).create_script_user(
"api_script", "api_key", "https://host.shotgunstudio.com", None)
connection = user.create_sg_connection()
self.assertEqual(connection.config.script_name, "api_script")
self.assertEqual(connection.config.api_key, "api_key")
