Coverage for python/lsst/obs/base/cli/butler_cmd_test.py : 55%

Hot-keys on this page
r m x p toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
1# This file is part of obs_base.
2#
3# Developed for the LSST Data Management System.
4# This product includes software developed by the LSST Project
5# (https://www.lsst.org).
6# See the COPYRIGHT file at the top-level directory of this distribution
7# for details of code ownership.
8#
9# This program is free software: you can redistribute it and/or modify
10# it under the terms of the GNU General Public License as published by
11# the Free Software Foundation, either version 3 of the License, or
12# (at your option) any later version.
13#
14# This program is distributed in the hope that it will be useful,
15# but WITHOUT ANY WARRANTY; without even the implied warranty of
16# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17# GNU General Public License for more details.
18#
19# You should have received a copy of the GNU General Public License
20# along with this program. If not, see <https://www.gnu.org/licenses/>.
22"""Base class for writing CLI butler command tests.
23"""
25__all__ = ("ButlerCmdTestBase",)
27import abc
28import click
29import click.testing
31from lsst.daf.butler.cli import butler
34class ButlerCmdTestBase(metaclass=abc.ABCMeta):
35 """Base class for tests of butler command line interface subcommands.
36 Subclass from this, then `unittest.TestCase` to get a working test suite.
37 """
39 @staticmethod
40 @abc.abstractmethod
41 def instrumentClass():
42 """Get the fully qualified instrument class.
44 Returns
45 -------
46 `str`
47 The fully qualified instrument class.
48 """
49 pass
51 @staticmethod
52 @abc.abstractmethod
53 def instrumentName():
54 """Get the instrument name.
56 Returns
57 -------
58 `str`
59 The name of the instrument.
60 """
61 pass
63 def test_cli(self):
64 runner = click.testing.CliRunner()
65 with runner.isolated_filesystem():
66 result = runner.invoke(butler.cli, ["create", "here"])
67 self.assertEqual(result.exit_code, 0, f"output: {result.output} exception: {result.exception}")
68 result = runner.invoke(butler.cli, ["register-instrument",
69 "here",
70 "-i", self.instrumentClass()])
71 self.assertEqual(result.exit_code, 0, f"output: {result.output} exception: {result.exception}")
72 result = runner.invoke(butler.cli, ["write-curated-calibrations",
73 "here",
74 "-i", self.instrumentName(),
75 "--output-run", "output_run"])
76 self.assertEqual(result.exit_code, 0, f"output: {result.output} exception: {result.exception}")