45 lines
1.8 KiB
Python
45 lines
1.8 KiB
Python
# Copyright 2023 (c) Anna Schumaker.
|
|
"""Tests the `xfstestsdb new` command."""
|
|
import io
|
|
import unittest
|
|
import unittest.mock
|
|
import xfstestsdb.new
|
|
|
|
|
|
class TestNew(unittest.TestCase):
|
|
"""Tests the `xfstestsdb new` command."""
|
|
|
|
def setUp(self):
|
|
"""Set up common variables."""
|
|
self.xfstestsdb = xfstestsdb.Command()
|
|
self.new = self.xfstestsdb.commands["new"]
|
|
|
|
def test_init(self):
|
|
"""Check that the new command was set up properly."""
|
|
self.assertIsInstance(self.new, xfstestsdb.commands.Command)
|
|
self.assertIsInstance(self.new, xfstestsdb.new.Command)
|
|
self.assertEqual(self.xfstestsdb.subparser.choices["new"],
|
|
self.new.parser)
|
|
|
|
@unittest.mock.patch("sys.stdout", new_callable=io.StringIO)
|
|
def test_new(self, mock_stdout: io.StringIO):
|
|
"""Test running the `xfstestsdb new` command with correct input."""
|
|
self.xfstestsdb.run(["new", "/dev/vdb1"])
|
|
self.assertRegex(mock_stdout.getvalue(),
|
|
r"created run #1 with test device '/dev/vdb1' "
|
|
r"\[[\d\-\: ]+\]\n")
|
|
|
|
@unittest.mock.patch("sys.stdout", new_callable=io.StringIO)
|
|
def test_new_terse(self, mock_stdout: io.StringIO):
|
|
"""Test running `xfstestsdb new --terse`."""
|
|
self.xfstestsdb.run(["new", "--terse", "/dev/vdb1"])
|
|
self.assertEqual(mock_stdout.getvalue(), "1\n")
|
|
|
|
@unittest.mock.patch("sys.stderr", new_callable=io.StringIO)
|
|
def test_new_error(self, mock_stderr: io.StringIO):
|
|
"""Test running the `xfstestsdb new` command with invalid input."""
|
|
with self.assertRaises(SystemExit):
|
|
self.xfstestsdb.run(["new"])
|
|
self.assertRegex(mock_stderr.getvalue(),
|
|
"error: the following arguments are required: device")
|