105 lines
4.3 KiB
Python
105 lines
4.3 KiB
Python
import pytest
|
|
from unittest.mock import MagicMock, patch
|
|
from datetime import datetime, timezone
|
|
import main
|
|
|
|
# convert_to_timestamp 함수 테스트
|
|
@pytest.mark.parametrize("input_date, expected_timestamp", [
|
|
("2025-12-25", int(datetime(2025, 12, 25, tzinfo=timezone.utc).timestamp())),
|
|
(1735084800, 1735084800),
|
|
("1735084800", 1735084800),
|
|
(None, None),
|
|
("", None),
|
|
("invalid-date", None),
|
|
])
|
|
def test_convert_to_timestamp(input_date, expected_timestamp):
|
|
"""
|
|
다양한 형식의 날짜 입력에 대해 convert_to_timestamp가 정확한 타임스탬프를 반환하는지 테스트합니다.
|
|
"""
|
|
assert main.convert_to_timestamp(input_date) == expected_timestamp
|
|
|
|
# _prepare_user_data 함수 테스트
|
|
def test_prepare_user_data():
|
|
"""
|
|
_prepare_user_data 함수가 사용자 정보를 올바르게 파싱하고,
|
|
특히 egBimLExpiryDate를 CLI 인자와 CSV 값에 따라 우선순위를 부여하여 처리하는지 테스트합니다.
|
|
"""
|
|
user_data = {
|
|
'login_id': 'test@example.com',
|
|
'tenants': '["tenant1", "tenant2"]',
|
|
'role_name': 'editor',
|
|
'company': 'Test Inc.',
|
|
'egBimLExpiryDate': '2024-12-31'
|
|
}
|
|
|
|
# 1. CLI expiry_timestamp가 주어질 경우
|
|
cli_timestamp = int(datetime(2025, 1, 1, tzinfo=timezone.utc).timestamp())
|
|
tenants, custom_attrs = main._prepare_user_data(user_data, expiry_timestamp=cli_timestamp)
|
|
assert custom_attrs['egBimLExpiryDate'] == cli_timestamp
|
|
assert custom_attrs['company'] == 'Test Inc.'
|
|
assert len(tenants) == 2
|
|
assert tenants[0].tenant_id == 'tenant1'
|
|
assert tenants[0].role_names == ['editor']
|
|
|
|
# 2. CLI expiry_timestamp가 없고, CSV에 값이 있을 경우
|
|
csv_timestamp = int(datetime(2024, 12, 31, tzinfo=timezone.utc).timestamp())
|
|
tenants, custom_attrs = main._prepare_user_data(user_data, expiry_timestamp=None)
|
|
assert custom_attrs['egBimLExpiryDate'] == csv_timestamp
|
|
|
|
# 3. 두 값 모두 없을 경우
|
|
user_data_no_date = user_data.copy()
|
|
del user_data_no_date['egBimLExpiryDate']
|
|
tenants, custom_attrs = main._prepare_user_data(user_data_no_date, expiry_timestamp=None)
|
|
assert 'egBimLExpiryDate' not in custom_attrs
|
|
|
|
# update_or_create_users 함수 테스트 (DescopeClient 모킹)
|
|
@patch('main.get_descope_client')
|
|
def test_update_or_create_users_flow(mock_get_client):
|
|
"""
|
|
update_or_create_users 함수가 사용자의 존재 여부에 따라
|
|
user.update 또는 user.create를 올바르게 호출하는지 테스트합니다.
|
|
"""
|
|
# 가짜 DescopeClient 및 mgmt.user 객체 설정
|
|
mock_descope_client = MagicMock()
|
|
mock_user_mgmt = MagicMock()
|
|
mock_descope_client.mgmt.user = mock_user_mgmt
|
|
mock_get_client.return_value = mock_descope_client
|
|
|
|
users_to_process = [
|
|
{'login_id': 'existing.user@example.com', 'display_name': 'Existing User Updated'},
|
|
{'login_id': 'new.user@example.com', 'display_name': 'New User Created'},
|
|
]
|
|
|
|
# 'existing.user@example.com'에 대해서는 load가 성공하고,
|
|
# 'new.user@example.com'에 대해서는 AuthException을 발생시키도록 설정
|
|
def load_user_side_effect(login_id):
|
|
if login_id == 'existing.user@example.com':
|
|
return {'user': 'details'} # 성공 시 반환값 (내용은 중요하지 않음)
|
|
elif login_id == 'new.user@example.com':
|
|
# Descope SDK는 사용자를 찾지 못하면 AuthException 발생
|
|
from descope import AuthException
|
|
raise AuthException("user not found", "E062103", 404)
|
|
|
|
mock_user_mgmt.load.side_effect = load_user_side_effect
|
|
|
|
# 테스트 실행
|
|
main.update_or_create_users(users_to_process)
|
|
|
|
# 검증: update가 'existing.user@example.com'에 대해 한 번 호출되었는지 확인
|
|
mock_user_mgmt.update.assert_called_once_with(
|
|
login_id='existing.user@example.com',
|
|
email=None,
|
|
display_name='Existing User Updated',
|
|
user_tenants=[],
|
|
custom_attributes={'completeForm': True}
|
|
)
|
|
|
|
# 검증: create가 'new.user@example.com'에 대해 한 번 호출되었는지 확인
|
|
mock_user_mgmt.create.assert_called_once_with(
|
|
login_id='new.user@example.com',
|
|
email=None,
|
|
display_name='New User Created',
|
|
user_tenants=[],
|
|
custom_attributes={'completeForm': True}
|
|
)
|