"""Test System User.""" from os import environ from content_utils.connectors import system_user def test_system_user_default(): """Test a default system user.""" mock_user = system_user.SystemUser() assert mock_user.identity_id == system_user.DEFAULT_IDENTITY_ID assert mock_user.identity_uuid == system_user.DEFAULT_IDENTITY_ID assert mock_user.profile_id == system_user.DEFAULT_PROFILE_ID assert mock_user.profile_uuid == system_user.DEFAULT_PROFILE_UUID assert mock_user.profile_type == system_user.DEFAULT_PROFILE_TYPE assert mock_user.roles == system_user.DEFAULT_ROLES def test_system_user_env_vars(): """Test that env vars override defaults.""" environ['SYSTEM_USER_IDENTITY_ID'] = 'foo' environ['SYSTEM_USER_IDENTITY_UUID'] = 'bar' environ['SYSTEM_USER_PROFILE_ID'] = 'baz' environ['SYSTEM_USER_PROFILE_UUID'] = 'quz' environ['SYSTEM_USER_PROFILE_TYPE'] = 'zok' environ['SYSTEM_USER_ROLES'] = 'pil' mock_user = system_user.SystemUser() assert mock_user.identity_id == 'foo' assert mock_user.identity_uuid == 'bar' assert mock_user.profile_id == 'baz' assert mock_user.profile_uuid == 'quz' assert mock_user.profile_type == 'zok' assert mock_user.roles == 'pil' def test_system_user_empty_env_vars_are_ignored(): """Test that env vars override defaults.""" environ['SYSTEM_USER_IDENTITY_ID'] = '' environ['SYSTEM_USER_IDENTITY_UUID'] = '' environ['SYSTEM_USER_PROFILE_ID'] = '' environ['SYSTEM_USER_PROFILE_UUID'] = '' environ['SYSTEM_USER_PROFILE_TYPE'] = '' environ['SYSTEM_USER_ROLES'] = '' mock_user = system_user.SystemUser() assert mock_user.identity_id == system_user.DEFAULT_IDENTITY_ID assert mock_user.identity_uuid == system_user.DEFAULT_IDENTITY_ID assert mock_user.profile_id == system_user.DEFAULT_PROFILE_ID assert mock_user.profile_uuid == system_user.DEFAULT_PROFILE_UUID assert mock_user.profile_type == system_user.DEFAULT_PROFILE_TYPE assert mock_user.roles == system_user.DEFAULT_ROLES def test_system_user_args(): """Test that arguments are used if supplied.""" environ['SYSTEM_USER_IDENTITY_ID'] = 'foo' environ['SYSTEM_USER_IDENTITY_UUID'] = 'bar' environ['SYSTEM_USER_PROFILE_ID'] = 'baz' environ['SYSTEM_USER_PROFILE_UUID'] = 'quz' environ['SYSTEM_USER_PROFILE_TYPE'] = 'zok' environ['SYSTEM_USER_ROLES'] = 'pil' mock_user = system_user.SystemUser( identity_id='mock_identity_id', identity_uuid='mock_identity_uuid', profile_id='mock_profile_id', profile_uuid='mock_profile_uuid', profile_type='mock_profile_type', roles='mock_roles' ) assert mock_user.identity_id == 'mock_identity_id' assert mock_user.identity_uuid == 'mock_identity_uuid' assert mock_user.profile_id == 'mock_profile_id' assert mock_user.profile_uuid == 'mock_profile_uuid' assert mock_user.profile_type == 'mock_profile_type' assert mock_user.roles == 'mock_roles'