61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
# Copyright 2023 (c) Anna Schumaker.
|
|
"""Tests the `xfstestsdb` command."""
|
|
import io
|
|
import argparse
|
|
import unittest
|
|
import unittest.mock
|
|
import xfstestsdb
|
|
|
|
|
|
class TestXfstestsdb(unittest.TestCase):
|
|
"""Tests the xfstestsdb command operation."""
|
|
|
|
def setUp(self):
|
|
"""Set up common variables."""
|
|
self.xfstestsdb = xfstestsdb.Command()
|
|
|
|
def test_init(self):
|
|
"""Test that the xfstestsdb instance is set up properly."""
|
|
self.assertIsInstance(self.xfstestsdb.parser, argparse.ArgumentParser)
|
|
self.assertIsInstance(self.xfstestsdb.subparser, argparse.Action)
|
|
self.assertIsInstance(self.xfstestsdb.sql,
|
|
xfstestsdb.sqlite.Connection)
|
|
|
|
def test_close(self):
|
|
"""Test that the xfstestsdb instance closes the database."""
|
|
self.xfstestsdb.__del__()
|
|
self.assertFalse(self.xfstestsdb.sql.connected)
|
|
|
|
@unittest.mock.patch("sys.stdout", new_callable=io.StringIO)
|
|
def test_run_empty(self, mock_stdout: io.StringIO):
|
|
"""Test calling xfstestdb.run() with an empty argument list."""
|
|
self.xfstestsdb.run([])
|
|
self.assertRegex(mock_stdout.getvalue(),
|
|
r"^usage: pytest \[\-h\] \[\-\-version\]"
|
|
r"\s+\{.*?}\s+...$")
|
|
|
|
def test_run(self):
|
|
"""Test running the xfstestsdb."""
|
|
parser = self.xfstestsdb.subparser.add_parser("test-run", help="help")
|
|
test_done = unittest.mock.Mock()
|
|
test_passed = False
|
|
|
|
def test_func(args: argparse.Namespace) -> None:
|
|
nonlocal test_passed
|
|
self.assertTrue(self.xfstestsdb.sql.sql.in_transaction)
|
|
test_passed = True
|
|
parser.set_defaults(function=test_func, done=test_done)
|
|
|
|
self.xfstestsdb.run(["test-run"])
|
|
self.assertTrue(test_passed)
|
|
test_done.assert_called()
|
|
|
|
@unittest.mock.patch("sys.stdout", new_callable=io.StringIO)
|
|
def test_version(self, mock_stdout: io.StringIO):
|
|
"""Test printing version information."""
|
|
self.assertEqual(xfstestsdb.MAJOR, 1)
|
|
self.assertEqual(xfstestsdb.MINOR, 6)
|
|
|
|
self.xfstestsdb.run(["--version"])
|
|
self.assertEqual(mock_stdout.getvalue(), "xfstestsdb v1.6-debug\n")
|