Coverage for tests/test_args.py : 34%

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#
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 = '--id "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.datasetKey
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.datasetKey
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.datasetKey
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 --id parses correctly.
92 """
93 args = '--dataset %s --output tests/output/foo --id "visit=54123" --id "filter=x"' \
94 % CommandLineTestSuite.datasetKey
95 parsed = self._parseString(args)
96 self.assertEqual(parsed.dataIds, ["visit=54123", "filter=x"])
98 def testEmptyDataId(self):
99 """Test that an --id argument may be not followed by a data ID.
100 """
101 minArgs = f'--dataset {self.datasetKey} --output tests/output/foo'
103 # Nothing after --id
104 parsed = self._parseString(minArgs + ' --id')
105 self.assertEqual(parsed.dataIds, [])
107 # --dataset immediately after --id
108 parsed = self._parseString('--id ' + minArgs)
109 self.assertEqual(parsed.dataIds, [])
111 def testBadDataset(self):
112 """Verify that a command line with an unregistered dataset is rejected.
113 """
114 args = '--dataset FooScope --output tests/output/foo --id "visit=54123"'
115 with self.assertRaises(SystemExit):
116 self._parseString(args)
118 def testBadKeyMain(self):
119 """Verify that a command line with unsupported arguments is rejected.
120 """
121 args = '--dataset %s --output tests/output/foo --id "visit=54123" --clobber' \
122 % CommandLineTestSuite.datasetKey
123 with self.assertRaises(SystemExit):
124 self._parseString(args)
126 def testBadKeyIngest(self):
127 """Verify that a command line with unsupported arguments is rejected.
128 """
129 args = '--dataset %s --output tests/output/foo --id "visit=54123"' \
130 % CommandLineTestSuite.datasetKey
131 with self.assertRaises(SystemExit):
132 self._parseString(args, ap_verify._IngestOnlyParser())
135class MemoryTester(lsst.utils.tests.MemoryTestCase):
136 pass
139def setup_module(module):
140 lsst.utils.tests.init()
143if __name__ == "__main__": 143 ↛ 144line 143 didn't jump to line 144, because the condition on line 143 was never true
144 lsst.utils.tests.init()
145 unittest.main()