"""Create fixtures in db prior to the tests."""

import mysql.connector

cnx = mysql.connector.connect(
    host='ledger-accounting-run-balance-mysql',
    port='3306',
    database='royalty_accounting',
    user='db_user',
    password='db_pass'
)

cursor = cnx.cursor()

query = """
INSERT INTO royalty_accounting.accounting_period (
    accounting_period_id,
    statement_period_id,
    accounting_period_name,
    accounting_period_status,
    contract_type,
    closed_date,
    created_by,
    created_at,
    last_modified_by,
    last_modified
) VALUES (
    274,
    274,
    'period',
    'locked',
    'distribution',
    null,
    'vz',
    '2021-05-05 09:39:01',
    'vz',
    '2021-05-05 09:39:04'
);
"""
cursor.execute(query)
cnx.commit()

query = """
INSERT INTO royalty_accounting.accounting_run (
    accounting_run_id,
    accounting_period_id,
    run_controller_id,
    run_status,
    start_date,
    summary_export_url,
    created_by,
    created_at,
    last_modified_by,
    last_modified
) VALUES (
    1473,
    274,
    2686,
    'running',
    NOW(),
    null,
    'vz',
    '2021-05-05 09:45:47',
    'vz',
    '2021-05-05 09:45:53'
);
"""
cursor.execute(query)
cnx.commit()


query = """
INSERT INTO royalty_accounting.contract(
    contract_id,
    reference_signing_entity_id,
    reference_sap_profit_center_id,
    contract_name,
    term_start,
    term_end,
    created_by,
    created_at,
    last_modified_by,
    last_modified
) VALUES
    (28, 1, 1, 'test', '2021-07-09', '2021-07-28', 'vz', NOW(), 'vz', NOW()),
    (29, 1, 1, 'other test', '2021-07-09', '2021-07-28', 'dana', NOW(), 'dana', NOW());
"""
cursor.execute(query)
cnx.commit()


query = """
INSERT INTO royalty_accounting.abacus_event (
    abacus_event_id,
    statement_period_id,
    event_name,
    target_type,
    target_id,
    event_date
) VALUES (
    1,
    274,
    'accounting_run_calculate',
    'accounting_run',
    1473,
    '2015-03-31 00:00:00.000000'
);
"""
cursor.execute(query)
cnx.commit()

cnx.close()

print('Added required fixtures to docker DB.')