import sys
import unittest
import xmlrunner


def env_set(environment):
    test_env = 'http://lolo-test.com'
    dev_env = 'http://lolo-dev.com'
    prod_env = 'http://lolo-prod.com'

    if environment:
        if environment == 'test':
            return test_env
        elif environment == 'dev':
            return dev_env
        elif environment == 'prod':
            return prod_env
        else:
            print(f'There is no environment like: {environment}, setting up default test env: {test_env}')
            return test_env
    else:
        print(f'No environment specified, setting up default test env: {test_env}')
        return test_env


class TestLogin(unittest.TestCase):
    ENV = ''

    def setUp(self):
        self.base_url = env_set(self.ENV)

    def test_logins_or_something(self):
        print(self.base_url)


if __name__ == "__main__":
    if len(sys.argv) > 1:
        TestLogin.ENV = sys.argv.pop()
    unittest.main(
        testRunner=xmlrunner.XMLTestRunner(output='test-reports'),
        # these make sure that some options that are not applicable
        # remain hidden from the help menu.
        failfast=False, buffer=False, catchbreak=False)