Commit 8b93aec8 authored by JunHyung An's avatar JunHyung An
Browse files

Add order on user input

parent 54ef4b1e
Loading
Loading
Loading
Loading
Loading
+10 −10
Original line number Diff line number Diff line
@@ -19,19 +19,19 @@ class MultipleTest(unittest.TestCase):
    def _get_user_inputs(self):
        """Helper method to get user inputs and set up initial configuration."""
        # Test case user input
        print("[필수 입력] Chrome 위치 (e.g., /opt/chrome-linux64/chrome, D:\\tmp\\chrome-win64\\chrome.exe): ")
        print("1. [필수 입력] Chrome 위치 (e.g., /opt/chrome-linux64/chrome, D:\\tmp\\chrome-win64\\chrome.exe): ")
        self.binary_location = input().strip()

        print("\n[필수 입력] webdriver 위치 (e.g., /opt/chromedriver-linux64/chromedriver, D:\\tmp\\chromedriver-win64\\chromedriver.exe): ")
        print("\n2. [필수 입력] webdriver 위치 (e.g., /opt/chromedriver-linux64/chromedriver, D:\\tmp\\chromedriver-win64\\chromedriver.exe): ")
        self.webdriver_path = input().strip()

        print("\n[필수 입력] 테스트 대상 웹 (e.g., https://example.com:8443/): ")
        print("\n3. [필수 입력] 테스트 대상 웹 (e.g., https://example.com:8443/): ")
        self.dc_address = input()

        # A list is created to store multiple account configurations.
        self.account_configs = []
        # print("\n[필수 입력] 계정 정보를 한 줄씩 입력하세요. (포맷: 유형 / 아이디, 비번 / 세션 수)")
        print("\n[필수 입력] 계정유형(1-어드민,2-(지역)관리자,3-사용자) / 계정 접속정보 / 커넥션 비중")
        print("\n4. [필수 입력] 계정유형(1-어드민,2-(지역)관리자,3-사용자) / 계정 접속정보 / 커넥션 비중")
        print("ex) 1 / admin, admin / 3")
        print("ex) 2 / watchall, watchall / 3")
        print("ex) 3 / monitor, monitor / 4")
@@ -66,27 +66,27 @@ class MultipleTest(unittest.TestCase):
                print(f"잘못된 입력 형식입니다: '{user_input}'. 해당 줄은 무시됩니다.")

        # New inputs for connection management
        print("\n[선택(디폴트:일반접속)] 테스트 유형 : 일반 접속(1), 일시 접속(2), 부하 화면 접근(3)")
        print("\n5. [선택] 테스트 유형(기본값: 일반접속): 일반 접속(1), 일시 접속(2), 부하 화면 접근(3)")
        try:
            self.function_to_execute = int(input().strip())
        except ValueError:
            print("잘못된 입력입니다. 기본값 일반 접속(1)으로 설정합니다.")
            print("잘못된 입력입니다. 기본값 '일반 접속(1)'으로 설정합니다.")
            self.function_to_execute = 1

        print("\n[선택(기본값: 120분)] 연결 유지 시간 (분): ")
        print("\n6. [선택] 연결 유지 시간(분)(기본값: 120)")
        try:
            self.keep_connection_time = int(input().strip()) * 60
        except ValueError:
            print("잘못된 입력입니다. 기본값 120분으로 설정합니다.")
            print("잘못된 입력입니다. 기본값 '120분'으로 설정합니다.")
            self.keep_connection_time = 120 * 60

        print("\n[선택(기본값: 50 / 30)] 최대 연결 수 / 최대 연결 수 도달 시간 (분): ")
        print("\n7. [선택] 유지 커넥션 개수 / 최대 커넥션 수 까지 도달 시간(분)(기본값: 50 / 30)")
        try:
            pool_input = input().strip().split('/')
            self.max_connections = int(pool_input[0].strip())
            self.ramp_up_time = int(pool_input[1].strip()) * 60
        except (ValueError, IndexError):
            print("잘못된 입력입니다. 기본값 50개 / 30분으로 설정합니다.")
            print("잘못된 입력입니다. 기본값 '50개 / 30분'으로 설정합니다.")
            self.max_connections = 50
            self.ramp_up_time = 30 * 60