diff options
Diffstat (limited to 'Lib')
| -rw-r--r-- | Lib/nntplib.py | 11 | ||||
| -rw-r--r-- | Lib/test/test_nntplib.py | 35 |
2 files changed, 37 insertions, 9 deletions
diff --git a/Lib/nntplib.py b/Lib/nntplib.py index fde339a..d5786e2 100644 --- a/Lib/nntplib.py +++ b/Lib/nntplib.py @@ -571,14 +571,19 @@ class _NNTPBase: cmd = 'NEWNEWS {0} {1} {2}'.format(group, date_str, time_str) return self._longcmdstring(cmd, file) - def list(self, *, file=None): - """Process a LIST command. Argument: + def list(self, group_pattern=None, *, file=None): + """Process a LIST or LIST ACTIVE command. Arguments: + - group_pattern: a pattern indicating which groups to query - file: Filename string or file object to store the result in Returns: - resp: server response if successful - list: list of (group, last, first, flag) (strings) """ - resp, lines = self._longcmdstring('LIST', file) + if group_pattern is not None: + command = 'LIST ACTIVE ' + group_pattern + else: + command = 'LIST' + resp, lines = self._longcmdstring(command, file) return resp, self._grouplist(lines) def _getdescriptions(self, group_pattern, return_all): diff --git a/Lib/test/test_nntplib.py b/Lib/test/test_nntplib.py index 8da901f1..f9a4cdc 100644 --- a/Lib/test/test_nntplib.py +++ b/Lib/test/test_nntplib.py @@ -22,16 +22,22 @@ class NetworkedNNTPTestsMixin: self.assertEqual(str, type(welcome)) def test_help(self): - resp, list = self.server.help() + resp, lines = self.server.help() self.assertTrue(resp.startswith("100 "), resp) - for line in list: + for line in lines: self.assertEqual(str, type(line)) def test_list(self): - resp, list = self.server.list() - if len(list) > 0: - self.assertEqual(GroupInfo, type(list[0])) - self.assertEqual(str, type(list[0].group)) + resp, groups = self.server.list() + if len(groups) > 0: + self.assertEqual(GroupInfo, type(groups[0])) + self.assertEqual(str, type(groups[0].group)) + + def test_list_active(self): + resp, groups = self.server.list(self.GROUP_PAT) + if len(groups) > 0: + self.assertEqual(GroupInfo, type(groups[0])) + self.assertEqual(str, type(groups[0].group)) def test_unknown_command(self): with self.assertRaises(nntplib.NNTPPermanentError) as cm: @@ -383,6 +389,17 @@ class NNTPv1Handler: free.it.comp.lang.python.learner 0000000000 0000000001 y tw.bbs.comp.lang.python 0000000304 0000000304 y .""") + elif action == "ACTIVE": + if param == "*distutils*": + self.push_lit("""\ + 215 Newsgroups in form "group high low flags" + gmane.comp.python.distutils.devel 0000014104 0000000001 m + gmane.comp.python.distutils.cvs 0000000000 0000000001 m + .""") + else: + self.push_lit("""\ + 215 Newsgroups in form "group high low flags" + .""") elif action == "OVERVIEW.FMT": self.push_lit("""\ 215 Order of fields in overview database. @@ -608,6 +625,12 @@ class NNTPv1v2TestsMixin: self.assertEqual(g, GroupInfo("comp.lang.python.announce", "0000001153", "0000000993", "m")) + resp, groups = self.server.list("*distutils*") + self.assertEqual(len(groups), 2) + g = groups[0] + self.assertEqual(g, + GroupInfo("gmane.comp.python.distutils.devel", "0000014104", + "0000000001", "m")) def test_stat(self): resp, art_num, message_id = self.server.stat(3000234) |