"""Generation of taurus yml file."""

import argparse

import yaml

parser = argparse.ArgumentParser()
parser.add_argument(
    "--users_count", "-u", help="a number of users per simulation", type=int, default=3
)
parser.add_argument(
    "--holdfor",
    "-hld",
    help="expected time for an asset to transcode in seconds",
    type=int,
    default=240,
)
parser.add_argument(
    "--iterations", "-i", help="number of iterations per user", type=int, default=1
)

args = parser.parse_args()

data = {
    "execution": [
        {
            "hold-for": "{}s".format(args.holdfor),
            "iterations": args.iterations,
            "scenario": "test",
        }
    ]
    * args.users_count,
    "reporting": ["final-stats", "console", "junit-xml", "blazemeter"],
    "settings": {
        "artifacts-dir": "tests/load/output/output_%Y-%m-%d_%H-%M-%S.%f",
        "default-executor": "pytest",
    },
    "scenarios": {
        "test": {
            "script": "tests/integration/test_assets.py",
            "additional-args": '-p no:owsrequest -k "test_get_asset_status_main"',
        }
    },
}


class NoAliasDumper(yaml.Dumper):
    def ignore_aliases(self, config_data: object) -> bool:
        return True


with open("tests/load/load.yml", "w") as outfile:
    yaml.dump(data, outfile, default_flow_style=False, Dumper=NoAliasDumper)