Coverage for tests/test_args.py: 29%
54 statements
« prev ^ index » next coverage.py v6.5.0, created at 2022-10-25 07:48 +0000
« prev ^ index » next coverage.py v6.5.0, created at 2022-10-25 07:48 +0000
1#
2# This file is part of ap_verify.
3#
4# Developed for the LSST Data Management System.
5# This product includes software developed by the LSST Project
6# (http://www.lsst.org).
7# See the COPYRIGHT file at the top-level directory of this distribution
8# for details of code ownership.
9#
10# This program is free software: you can redistribute it and/or modify
11# it under the terms of the GNU General Public License as published by
12# the Free Software Foundation, either version 3 of the License, or
13# (at your option) any later version.
14#
15# This program is distributed in the hope that it will be useful,
16# but WITHOUT ANY WARRANTY; without even the implied warranty of
17# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18# GNU General Public License for more details.
19#
20# You should have received a copy of the GNU General Public License
21# along with this program. If not, see <http://www.gnu.org/licenses/>.
22#
24import shlex
25import unittest
27import lsst.utils.tests
28import lsst.ap.verify.ap_verify as ap_verify
29import lsst.ap.verify.testUtils
32class CommandLineTestSuite(lsst.ap.verify.testUtils.DataTestCase):
33 # DataTestCase's test dataset is needed for successful parsing of the --dataset argument
35 def _parseString(self, commandLine, parser=None):
36 """Tokenize and parse a command line string.
38 Parameters
39 ----------
40 commandLine : `str`
41 a string containing Unix-style command line arguments, but not the
42 name of the program
43 parser : `argparse.ArgumentParser`, optional
44 the parser to use. Defaults to ``ap_verify``'s primary parser.
46 Returns
47 -------
48 parsed : `argparse.Namespace`
49 The parsed command line.
50 """
51 if not parser:
52 parser = ap_verify._ApVerifyParser()
54 return parser.parse_args(shlex.split(commandLine))
56 def testMissingMain(self):
57 """Verify that a command line consisting missing required arguments is rejected.
58 """
59 args = '--data-query "visit=54123" --output tests/output/foo'
60 with self.assertRaises(SystemExit):
61 self._parseString(args)
63 def testMissingIngest(self):
64 """Verify that a command line consisting missing required arguments is rejected.
65 """
66 args = '--dataset %s' % CommandLineTestSuite.testDataset
67 with self.assertRaises(SystemExit):
68 self._parseString(args, ap_verify._IngestOnlyParser())
70 def testMinimumMain(self):
71 """Verify that a command line consisting only of required arguments parses correctly.
72 """
73 args = '--dataset %s --output tests/output/foo' % CommandLineTestSuite.testDataset
74 parsed = self._parseString(args)
75 self.assertEqual(parsed.dataset.datasetRoot,
76 lsst.utils.getPackageDir(CommandLineTestSuite.testDataset))
77 self.assertEqual(parsed.output, "tests/output/foo")
78 self.assertEqual(parsed.dataIds, [])
80 def testMinimumIngest(self):
81 """Verify that a command line consisting only of required arguments parses correctly.
82 """
83 args = '--dataset %s --output tests/output/foo' % CommandLineTestSuite.testDataset
84 parsed = self._parseString(args, ap_verify._IngestOnlyParser())
85 self.assertEqual(parsed.dataset.datasetRoot,
86 lsst.utils.getPackageDir(CommandLineTestSuite.testDataset))
87 self.assertEqual(parsed.output, "tests/output/foo")
89 def testDataId(self):
90 """Verify that a command line consisting only of required arguments and
91 --data-query parses correctly.
92 """
93 args = '--dataset %s --output tests/output/foo --data-query "visit=54123" --data-query "filter=x"' \
94 % CommandLineTestSuite.testDataset
95 parsed = self._parseString(args)
96 self.assertEqual(parsed.dataIds, ["visit=54123", "filter=x"])
98 def testEmptyDataId(self):
99 """Test that an --data-query argument must be followed by a data ID.
100 """
101 minArgs = f'--dataset {self.testDataset} --output tests/output/foo'
103 # Nothing after --data-query
104 with self.assertRaises(SystemExit):
105 self._parseString(minArgs + ' --data-query')
107 # --dataset immediately after --data-query
108 with self.assertRaises(SystemExit):
109 self._parseString('--data-query ' + minArgs)
111 def testBadKeyMain(self):
112 """Verify that a command line with unsupported arguments is rejected.
113 """
114 args = '--dataset %s --output tests/output/foo --data-query "visit=54123" --clobber' \
115 % CommandLineTestSuite.testDataset
116 with self.assertRaises(SystemExit):
117 self._parseString(args)
119 def testBadKeyIngest(self):
120 """Verify that a command line with unsupported arguments is rejected.
121 """
122 args = '--dataset %s --output tests/output/foo --data-query "visit=54123"' \
123 % CommandLineTestSuite.testDataset
124 with self.assertRaises(SystemExit):
125 self._parseString(args, ap_verify._IngestOnlyParser())
128class MemoryTester(lsst.utils.tests.MemoryTestCase):
129 pass
132def setup_module(module):
133 lsst.utils.tests.init()
136if __name__ == "__main__": 136 ↛ 137line 136 didn't jump to line 137, because the condition on line 136 was never true
137 lsst.utils.tests.init()
138 unittest.main()