# test_insert_carriers.py
from src.database_module import Carrier, Base
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

def insert_initial_carriers():
    # Database connection
    engine = create_engine('sqlite:///orders.db')  # Update with your database URL
    Base.metadata.create_all(engine)
    Session = sessionmaker(bind=engine)
    session = Session()

    # List of carriers with their service names and charges
    carriers = [
        {'service_name': 'SK텔레콤', 'sms_charge': 22, 'mms_charge': 110},
        {'service_name': 'KT', 'sms_charge': 22, 'mms_charge': 110},
        {'service_name': 'LG유플러스', 'sms_charge': 22, 'mms_charge': 110},
        {'service_name': '모빙', 'sms_charge': 20, 'mms_charge': 100},  # 유니컴즈의 서비스명
        {'service_name': '헬로모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '알뜰폰', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '이야기모바일', 'sms_charge': 22, 'mms_charge': 220},
        {'service_name': 'U+알뜰모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': 'KT엠모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': 'SK세븐모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '프리티모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '에스원모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '마이월드', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': 'KCT모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': 'CJ헬로모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '온국민폰', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '아이즈모바일', 'sms_charge': 22, 'mms_charge': 110},
        {'service_name': '에넥스텔레콤', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '스노우맨', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': 'U+유모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': 'KT파워텔', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': 'SK텔링크', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '이마트알뜰폰', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '홈플러스모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': 'GS25모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '롯데모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': 'CU알뜰폰', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '세븐일레븐모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '미니스톱모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '하나모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '우리모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '신한카드모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': 'KB국민은행모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': 'NH농협모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': 'IBK기업은행모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '하나은행모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '우리은행모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '신한은행모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '카카오뱅크모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '케이뱅크모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '토스뱅크모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '우체국알뜰폰', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '공공와이파이모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '공공알뜰폰', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '국민행복모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '행복한안심모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '안심모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '착한텔레콤', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '착한모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '좋은텔레콤', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '좋은모바일', 'sms_charge': 20, 'mms_charge': 100},
        {'service_name': '행복한텔레콤', 'sms_charge': 20, 'mms_charge': 100},
    ]

    # Insert carriers into the database
    carrier_objects = [Carrier(**carrier) for carrier in carriers]
    session.bulk_save_objects(carrier_objects)
    session.commit()
    session.close()

    print("Initial carriers have been inserted into the database.")

if __name__ == '__main__':
    insert_initial_carriers()