def test_cli_with_utils(self):
"""Test that the command line works with a simple simulation"""
runner = CliRunner()
runner.echo_stdin = True
result = runner.invoke(cli.simulate, [
"--sol-file", "tests/sols/basic.p", "--output-prefix", "clitest",
"--output-dir", ".", "--workload-length", "3600", "--quantum",
"3600", "--save-utils", "true"
])
assert result.exit_code == 0
files = glob.glob("clitest*")
assert 'clitest.csv' in files
assert 'clitest_out.txt' in files
assert 'clitest_evs.csv' not in files
assert 'clitest_reqs.csv' not in files
assert 'clitest_utils.csv' in files
df = pd.read_csv('clitest_utils.csv')
assert len(df) == 3600
util0_row = df.iloc[0]
assert util0_row.vm_name == 'VM 0'
assert util0_row.ic == 'priv'
assert util0_row.util == 1
last_util_row = df.iloc[-1]
assert last_util_row.vm_name == 'VM 3599'
assert last_util_row.ic == 'priv'
assert last_util_row.util == 1
def test_cli_with_evs(self):
"""Test that the command line works with a simple simulation"""
runner = CliRunner()
runner.echo_stdin = True
result = runner.invoke(cli.simulate, [
"--sol-file", "tests/sols/basic.p", "--output-prefix", "clitest",
"--output-dir", ".", "--workload-length", "3600", "--quantum",
"3600", "--save-evs", "true"
])
assert result.exit_code == 0
files = glob.glob("clitest*")
assert 'clitest.csv' in files
assert 'clitest_out.txt' in files
assert 'clitest_reqs.csv' in files
assert 'clitest_utils.csv' not in files
df = pd.read_csv('clitest_reqs.csv')
assert len(df) == 10800
req1_row = df.iloc[1]
assert req1_row.req == 1
assert req1_row.creation == 0
assert req1_row.start == 1200
assert req1_row.end == 2400
assert req1_row.app == 'a0'
assert req1_row.vm == 'VM 0'
assert req1_row.ic == 'priv'
assert req1_row.lost == False
def test_cli_yaml(self):
"""Test that the command line works with a YAML file"""
runner = CliRunner()
runner.echo_stdin = True
result = runner.invoke(cli.simulate, [
"--sol-file", "tests/sols/3vm.yaml", "--output-prefix", "clitest",
"--output-dir", ".", "--quantum", "3600"
])
assert result.exit_code == 0
def test_list_repos(self):
runner = CliRunner()
runner.echo_stdin = True
# create a repo and a component
with runner.isolated_filesystem():
result = runner.invoke(cli.create_repo,
["my_repo", "--gpgkey", "123456"])
self.assertEqual(result.exit_code, 0)
result = runner.invoke(cli.add_component, ["my_repo", "main"])
self.assertEqual(result.exit_code, 0)
# make sure list_repos is showing the correct output
result = runner.invoke(cli.list_repos)
self.assertEqual(result.exit_code, 0)
self.assertEqual(result.output, LIST_REPOS_OUTPUT)
def test_list_repos(self):
runner = CliRunner()
runner.echo_stdin = True
# create a repo and a component
with runner.isolated_filesystem():
result = runner.invoke(
cli.create_repo, ["my_repo", "--gpgkey", "123456"])
self.assertEqual(result.exit_code, 0)
result = runner.invoke(cli.add_component, ["my_repo", "main"])
self.assertEqual(result.exit_code, 0)
# make sure list_repos is showing the correct output
result = runner.invoke(cli.list_repos)
self.assertEqual(result.exit_code, 0)
self.assertEqual(result.output, LIST_REPOS_OUTPUT)
def test_cli_with_config_file(self):
runner = CliRunner()
runner.echo_stdin = True
result = runner.invoke(cli.simulate, ["--config", "tests/config_test"])
assert result.exit_code == 0