Mercurial > cpython-lru_cache-weakref

annotate Lib/test/test_fileinput.py @ 79691:05c8928362ee

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Merge __next__ method link fixes with 3.3.
author Ezio Melotti <ezio.melotti@gmail.com>
date Fri, 12 Oct 2012 13:46:37 +0300
parents c2b3e8b437a4
children 7d69d04522e3
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
18484
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
1 '''
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
2 Tests for fileinput module.
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
3 Nick Mathewson
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
4 '''
68496
937b1fbb402f import style nits
Benjamin Peterson <benjamin@python.org>
parents: 68495
diff changeset
5 import os
937b1fbb402f import style nits
Benjamin Peterson <benjamin@python.org>
parents: 68495
diff changeset
6 import sys
937b1fbb402f import style nits
Benjamin Peterson <benjamin@python.org>
parents: 68495
diff changeset
7 import re
937b1fbb402f import style nits
Benjamin Peterson <benjamin@python.org>
parents: 68495
diff changeset
8 import fileinput
937b1fbb402f import style nits
Benjamin Peterson <benjamin@python.org>
parents: 68495
diff changeset
9 import collections
73434
c2b3e8b437a4 fileinput: replace this last occurence of codecs.open with builtins.open.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 70096
diff changeset
10 import builtins
68496
937b1fbb402f import style nits
Benjamin Peterson <benjamin@python.org>
parents: 68495
diff changeset
11 import unittest
18484
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
12
68669
bb645cc39e60 Fix #11596. When bz2 isn't available, skip test_bz2_ext_fake.
briancurtin <brian.curtin@gmail.com>
parents: 68530
diff changeset
13 try:
bb645cc39e60 Fix #11596. When bz2 isn't available, skip test_bz2_ext_fake.
briancurtin <brian.curtin@gmail.com>
parents: 68530
diff changeset
14 import bz2
bb645cc39e60 Fix #11596. When bz2 isn't available, skip test_bz2_ext_fake.
briancurtin <brian.curtin@gmail.com>
parents: 68530
diff changeset
15 except ImportError:
bb645cc39e60 Fix #11596. When bz2 isn't available, skip test_bz2_ext_fake.
briancurtin <brian.curtin@gmail.com>
parents: 68530
diff changeset
16 bz2 = None
70096
e0b4ed75b8e7 Skip test in test_fileinput when zlib is missing.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 68669
diff changeset
17 try:
e0b4ed75b8e7 Skip test in test_fileinput when zlib is missing.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 68669
diff changeset
18 import gzip
e0b4ed75b8e7 Skip test in test_fileinput when zlib is missing.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 68669
diff changeset
19 except ImportError:
e0b4ed75b8e7 Skip test in test_fileinput when zlib is missing.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 68669
diff changeset
20 gzip = None
68669
bb645cc39e60 Fix #11596. When bz2 isn't available, skip test_bz2_ext_fake.
briancurtin <brian.curtin@gmail.com>
parents: 68530
diff changeset
21
68496
937b1fbb402f import style nits
Benjamin Peterson <benjamin@python.org>
parents: 68495
diff changeset
22 from io import StringIO
937b1fbb402f import style nits
Benjamin Peterson <benjamin@python.org>
parents: 68495
diff changeset
23 from fileinput import FileInput, hook_encoded
937b1fbb402f import style nits
Benjamin Peterson <benjamin@python.org>
parents: 68495
diff changeset
24
47201
9c930a32cebb #2621 rename test.test_support to test.support
Benjamin Peterson <benjamin@python.org>
parents: 46047
diff changeset
25 from test.support import verbose, TESTFN, run_unittest
9c930a32cebb #2621 rename test.test_support to test.support
Benjamin Peterson <benjamin@python.org>
parents: 46047
diff changeset
26 from test.support import unlink as safe_unlink
68496
937b1fbb402f import style nits
Benjamin Peterson <benjamin@python.org>
parents: 68495
diff changeset
27
18484
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
28
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
29 # The fileinput module has 2 interfaces: the FileInput class which does
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
30 # all the work, and a few functions (input, etc.) that use a global _state
68495
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
31 # variable.
18484
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
32
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
33 # Write lines (a list of lines) to temp file number i, and return the
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
34 # temp file's name.
36607
1a98798a4b1d Repair new test failures on Windows due to
Tim Peters <tim.peters@gmail.com>
parents: 36596
diff changeset
35 def writeTmp(i, lines, mode='w'): # opening in text mode is the default
18484
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
36 name = TESTFN + str(i)
36607
1a98798a4b1d Repair new test failures on Windows due to
Tim Peters <tim.peters@gmail.com>
parents: 36596
diff changeset
37 f = open(name, mode)
42111
32fe2b9e9f11 Fix a buch of shallow test failures.
Guido van Rossum <guido@python.org>
parents: 41575
diff changeset
38 for line in lines:
32fe2b9e9f11 Fix a buch of shallow test failures.
Guido van Rossum <guido@python.org>
parents: 41575
diff changeset
39 f.write(line)
18484
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
40 f.close()
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
41 return name
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
42
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
43 def remove_tempfiles(*names):
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
44 for name in names:
42918
9220931e4741 More changes needed to make things work once bytes and str are truly divorced.
Guido van Rossum <guido@python.org>
parents: 42501
diff changeset
45 if name:
9220931e4741 More changes needed to make things work once bytes and str are truly divorced.
Guido van Rossum <guido@python.org>
parents: 42501
diff changeset
46 safe_unlink(name)
18484
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
47
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
48 class BufferSizesTests(unittest.TestCase):
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
49 def test_buffer_sizes(self):
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
50 # First, run the tests with default and teeny buffer size.
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
51 for round, bs in (0, 0), (1, 30):
46047
df90774f92b0 Ensure cleanup does not reference variables that don't yet exist.
Neal Norwitz <nnorwitz@gmail.com>
parents: 45471
diff changeset
52 t1 = t2 = t3 = t4 = None
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
53 try:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
54 t1 = writeTmp(1, ["Line %s of file 1\n" % (i+1) for i in range(15)])
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
55 t2 = writeTmp(2, ["Line %s of file 2\n" % (i+1) for i in range(10)])
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
56 t3 = writeTmp(3, ["Line %s of file 3\n" % (i+1) for i in range(5)])
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
57 t4 = writeTmp(4, ["Line %s of file 4\n" % (i+1) for i in range(1)])
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
58 self.buffer_size_test(t1, t2, t3, t4, bs, round)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
59 finally:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
60 remove_tempfiles(t1, t2, t3, t4)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
61
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
62 def buffer_size_test(self, t1, t2, t3, t4, bs=0, round=0):
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
63 pat = re.compile(r'LINE (\d+) OF FILE (\d+)')
18484
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
64
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
65 start = 1 + round*6
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
66 if verbose:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
67 print('%s. Simple iteration (bs=%s)' % (start+0, bs))
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
68 fi = FileInput(files=(t1, t2, t3, t4), bufsize=bs)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
69 lines = list(fi)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
70 fi.close()
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
71 self.assertEqual(len(lines), 31)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
72 self.assertEqual(lines[4], 'Line 5 of file 1\n')
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
73 self.assertEqual(lines[30], 'Line 1 of file 4\n')
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
74 self.assertEqual(fi.lineno(), 31)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
75 self.assertEqual(fi.filename(), t4)
18484
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
76
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
77 if verbose:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
78 print('%s. Status variables (bs=%s)' % (start+1, bs))
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
79 fi = FileInput(files=(t1, t2, t3, t4), bufsize=bs)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
80 s = "x"
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
81 while s and s != 'Line 6 of file 2\n':
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
82 s = fi.readline()
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
83 self.assertEqual(fi.filename(), t2)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
84 self.assertEqual(fi.lineno(), 21)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
85 self.assertEqual(fi.filelineno(), 6)
55593
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47201
diff changeset
86 self.assertFalse(fi.isfirstline())
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47201
diff changeset
87 self.assertFalse(fi.isstdin())
18484
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
88
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
89 if verbose:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
90 print('%s. Nextfile (bs=%s)' % (start+2, bs))
18484
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
91 fi.nextfile()
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
92 self.assertEqual(fi.readline(), 'Line 1 of file 3\n')
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
93 self.assertEqual(fi.lineno(), 22)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
94 fi.close()
18484
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
95
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
96 if verbose:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
97 print('%s. Stdin (bs=%s)' % (start+3, bs))
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
98 fi = FileInput(files=(t1, t2, t3, t4, '-'), bufsize=bs)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
99 savestdin = sys.stdin
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
100 try:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
101 sys.stdin = StringIO("Line 1 of stdin\nLine 2 of stdin\n")
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
102 lines = list(fi)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
103 self.assertEqual(len(lines), 33)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
104 self.assertEqual(lines[32], 'Line 2 of stdin\n')
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
105 self.assertEqual(fi.filename(), '<stdin>')
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
106 fi.nextfile()
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
107 finally:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
108 sys.stdin = savestdin
18484
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
109
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
110 if verbose:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
111 print('%s. Boundary conditions (bs=%s)' % (start+4, bs))
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
112 fi = FileInput(files=(t1, t2, t3, t4), bufsize=bs)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
113 self.assertEqual(fi.lineno(), 0)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
114 self.assertEqual(fi.filename(), None)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
115 fi.nextfile()
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
116 self.assertEqual(fi.lineno(), 0)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
117 self.assertEqual(fi.filename(), None)
18484
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
118
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
119 if verbose:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
120 print('%s. Inplace (bs=%s)' % (start+5, bs))
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
121 savestdout = sys.stdout
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
122 try:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
123 fi = FileInput(files=(t1, t2, t3, t4), inplace=1, bufsize=bs)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
124 for line in fi:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
125 line = line[:-1].upper()
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
126 print(line)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
127 fi.close()
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
128 finally:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
129 sys.stdout = savestdout
18484
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
130
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
131 fi = FileInput(files=(t1, t2, t3, t4), bufsize=bs)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
132 for line in fi:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
133 self.assertEqual(line[-1], '\n')
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
134 m = pat.match(line[:-1])
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
135 self.assertNotEqual(m, None)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
136 self.assertEqual(int(m.group(1)), fi.filelineno())
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
137 fi.close()
18484
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
138
68495
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
139 class UnconditionallyRaise:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
140 def __init__(self, exception_type):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
141 self.exception_type = exception_type
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
142 self.invoked = False
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
143 def __call__(self, *args, **kwargs):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
144 self.invoked = True
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
145 raise self.exception_type()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
146
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
147 class FileInputTests(unittest.TestCase):
68495
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
148
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
149 def test_zero_byte_files(self):
46047
df90774f92b0 Ensure cleanup does not reference variables that don't yet exist.
Neal Norwitz <nnorwitz@gmail.com>
parents: 45471
diff changeset
150 t1 = t2 = t3 = t4 = None
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
151 try:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
152 t1 = writeTmp(1, [""])
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
153 t2 = writeTmp(2, [""])
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
154 t3 = writeTmp(3, ["The only line there is.\n"])
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
155 t4 = writeTmp(4, [""])
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
156 fi = FileInput(files=(t1, t2, t3, t4))
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
157
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
158 line = fi.readline()
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
159 self.assertEqual(line, 'The only line there is.\n')
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
160 self.assertEqual(fi.lineno(), 1)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
161 self.assertEqual(fi.filelineno(), 1)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
162 self.assertEqual(fi.filename(), t3)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
163
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
164 line = fi.readline()
55593
0bb232078b8c Merged revisions 73715 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47201
diff changeset
165 self.assertFalse(line)
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
166 self.assertEqual(fi.lineno(), 1)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
167 self.assertEqual(fi.filelineno(), 0)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
168 self.assertEqual(fi.filename(), t4)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
169 fi.close()
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
170 finally:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
171 remove_tempfiles(t1, t2, t3, t4)
18484
e061155f99e8 SF patch #440170: Tests for fileinput module.
Tim Peters <tim.peters@gmail.com>
parents:
diff changeset
172
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
173 def test_files_that_dont_end_with_newline(self):
46047
df90774f92b0 Ensure cleanup does not reference variables that don't yet exist.
Neal Norwitz <nnorwitz@gmail.com>
parents: 45471
diff changeset
174 t1 = t2 = None
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
175 try:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
176 t1 = writeTmp(1, ["A\nB\nC"])
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
177 t2 = writeTmp(2, ["D\nE\nF"])
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
178 fi = FileInput(files=(t1, t2))
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
179 lines = list(fi)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
180 self.assertEqual(lines, ["A\n", "B\n", "C", "D\n", "E\n", "F"])
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
181 self.assertEqual(fi.filelineno(), 3)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
182 self.assertEqual(fi.lineno(), 6)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
183 finally:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
184 remove_tempfiles(t1, t2)
36590
0d5f8077bd24 Patch #1337756: fileinput now accepts Unicode filenames.
Georg Brandl <georg@python.org>
parents: 24364
diff changeset
185
42111
32fe2b9e9f11 Fix a buch of shallow test failures.
Guido van Rossum <guido@python.org>
parents: 41575
diff changeset
186 ## def test_unicode_filenames(self):
32fe2b9e9f11 Fix a buch of shallow test failures.
Guido van Rossum <guido@python.org>
parents: 41575
diff changeset
187 ## # XXX A unicode string is always returned by writeTmp.
32fe2b9e9f11 Fix a buch of shallow test failures.
Guido van Rossum <guido@python.org>
parents: 41575
diff changeset
188 ## # So is this needed?
32fe2b9e9f11 Fix a buch of shallow test failures.
Guido van Rossum <guido@python.org>
parents: 41575
diff changeset
189 ## try:
32fe2b9e9f11 Fix a buch of shallow test failures.
Guido van Rossum <guido@python.org>
parents: 41575
diff changeset
190 ## t1 = writeTmp(1, ["A\nB"])
32fe2b9e9f11 Fix a buch of shallow test failures.
Guido van Rossum <guido@python.org>
parents: 41575
diff changeset
191 ## encoding = sys.getfilesystemencoding()
32fe2b9e9f11 Fix a buch of shallow test failures.
Guido van Rossum <guido@python.org>
parents: 41575
diff changeset
192 ## if encoding is None:
32fe2b9e9f11 Fix a buch of shallow test failures.
Guido van Rossum <guido@python.org>
parents: 41575
diff changeset
193 ## encoding = 'ascii'
32fe2b9e9f11 Fix a buch of shallow test failures.
Guido van Rossum <guido@python.org>
parents: 41575
diff changeset
194 ## fi = FileInput(files=str(t1, encoding))
32fe2b9e9f11 Fix a buch of shallow test failures.
Guido van Rossum <guido@python.org>
parents: 41575
diff changeset
195 ## lines = list(fi)
32fe2b9e9f11 Fix a buch of shallow test failures.
Guido van Rossum <guido@python.org>
parents: 41575
diff changeset
196 ## self.assertEqual(lines, ["A\n", "B"])
32fe2b9e9f11 Fix a buch of shallow test failures.
Guido van Rossum <guido@python.org>
parents: 41575
diff changeset
197 ## finally:
32fe2b9e9f11 Fix a buch of shallow test failures.
Guido van Rossum <guido@python.org>
parents: 41575
diff changeset
198 ## remove_tempfiles(t1)
36594
781654be55ef Patch #1215184: fileinput now has a fileno() function for getting the
Georg Brandl <georg@python.org>
parents: 36590
diff changeset
199
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
200 def test_fileno(self):
46047
df90774f92b0 Ensure cleanup does not reference variables that don't yet exist.
Neal Norwitz <nnorwitz@gmail.com>
parents: 45471
diff changeset
201 t1 = t2 = None
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
202 try:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
203 t1 = writeTmp(1, ["A\nB"])
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
204 t2 = writeTmp(2, ["C\nD"])
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
205 fi = FileInput(files=(t1, t2))
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
206 self.assertEqual(fi.fileno(), -1)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
207 line =next( fi)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
208 self.assertNotEqual(fi.fileno(), -1)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
209 fi.nextfile()
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
210 self.assertEqual(fi.fileno(), -1)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
211 line = list(fi)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
212 self.assertEqual(fi.fileno(), -1)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
213 finally:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
214 remove_tempfiles(t1, t2)
36595
c6fba456227b Patch #1212287: fileinput.input() now has a mode parameter for
Georg Brandl <georg@python.org>
parents: 36594
diff changeset
215
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
216 def test_opening_mode(self):
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
217 try:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
218 # invalid mode, should raise ValueError
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
219 fi = FileInput(mode="w")
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
220 self.fail("FileInput should reject invalid mode argument")
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
221 except ValueError:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
222 pass
42918
9220931e4741 More changes needed to make things work once bytes and str are truly divorced.
Guido van Rossum <guido@python.org>
parents: 42501
diff changeset
223 t1 = None
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
224 try:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
225 # try opening in universal newline mode
42918
9220931e4741 More changes needed to make things work once bytes and str are truly divorced.
Guido van Rossum <guido@python.org>
parents: 42501
diff changeset
226 t1 = writeTmp(1, [b"A\nB\r\nC\rD"], mode="wb")
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
227 fi = FileInput(files=t1, mode="U")
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
228 lines = list(fi)
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
229 self.assertEqual(lines, ["A\n", "B\n", "C\n", "D"])
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
230 finally:
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
231 remove_tempfiles(t1)
36596
e57377f0a167 Patch #1215184: FileInput now can be given an opening hook which can
Georg Brandl <georg@python.org>
parents: 36595
diff changeset
232
42918
9220931e4741 More changes needed to make things work once bytes and str are truly divorced.
Guido van Rossum <guido@python.org>
parents: 42501
diff changeset
233 def test_file_opening_hook(self):
9220931e4741 More changes needed to make things work once bytes and str are truly divorced.
Guido van Rossum <guido@python.org>
parents: 42501
diff changeset
234 try:
9220931e4741 More changes needed to make things work once bytes and str are truly divorced.
Guido van Rossum <guido@python.org>
parents: 42501
diff changeset
235 # cannot use openhook and inplace mode
9220931e4741 More changes needed to make things work once bytes and str are truly divorced.
Guido van Rossum <guido@python.org>
parents: 42501
diff changeset
236 fi = FileInput(inplace=1, openhook=lambda f, m: None)
9220931e4741 More changes needed to make things work once bytes and str are truly divorced.
Guido van Rossum <guido@python.org>
parents: 42501
diff changeset
237 self.fail("FileInput should raise if both inplace "
9220931e4741 More changes needed to make things work once bytes and str are truly divorced.
Guido van Rossum <guido@python.org>
parents: 42501
diff changeset
238 "and openhook arguments are given")
9220931e4741 More changes needed to make things work once bytes and str are truly divorced.
Guido van Rossum <guido@python.org>
parents: 42501
diff changeset
239 except ValueError:
9220931e4741 More changes needed to make things work once bytes and str are truly divorced.
Guido van Rossum <guido@python.org>
parents: 42501
diff changeset
240 pass
9220931e4741 More changes needed to make things work once bytes and str are truly divorced.
Guido van Rossum <guido@python.org>
parents: 42501
diff changeset
241 try:
9220931e4741 More changes needed to make things work once bytes and str are truly divorced.
Guido van Rossum <guido@python.org>
parents: 42501
diff changeset
242 fi = FileInput(openhook=1)
9220931e4741 More changes needed to make things work once bytes and str are truly divorced.
Guido van Rossum <guido@python.org>
parents: 42501
diff changeset
243 self.fail("FileInput should check openhook for being callable")
9220931e4741 More changes needed to make things work once bytes and str are truly divorced.
Guido van Rossum <guido@python.org>
parents: 42501
diff changeset
244 except ValueError:
9220931e4741 More changes needed to make things work once bytes and str are truly divorced.
Guido van Rossum <guido@python.org>
parents: 42501
diff changeset
245 pass
68495
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
246
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
247 class CustomOpenHook:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
248 def __init__(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
249 self.invoked = False
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
250 def __call__(self, *args):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
251 self.invoked = True
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
252 return open(*args)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
253
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
254 t = writeTmp(1, ["\n"])
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
255 self.addCleanup(remove_tempfiles, t)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
256 custom_open_hook = CustomOpenHook()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
257 with FileInput([t], openhook=custom_open_hook) as fi:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
258 fi.readline()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
259 self.assertTrue(custom_open_hook.invoked, "openhook not invoked")
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
260
63379
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
261 def test_context_manager(self):
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
262 try:
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
263 t1 = writeTmp(1, ["A\nB\nC"])
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
264 t2 = writeTmp(2, ["D\nE\nF"])
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
265 with FileInput(files=(t1, t2)) as fi:
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
266 lines = list(fi)
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
267 self.assertEqual(lines, ["A\n", "B\n", "C", "D\n", "E\n", "F"])
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
268 self.assertEqual(fi.filelineno(), 3)
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
269 self.assertEqual(fi.lineno(), 6)
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
270 self.assertEqual(fi._files, ())
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
271 finally:
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
272 remove_tempfiles(t1, t2)
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
273
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
274 def test_close_on_exception(self):
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
275 try:
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
276 t1 = writeTmp(1, [""])
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
277 with FileInput(files=t1) as fi:
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
278 raise IOError
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
279 except IOError:
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
280 self.assertEqual(fi._files, ())
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
281 finally:
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
282 remove_tempfiles(t1)
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
283
68495
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
284 def test_empty_files_list_specified_to_constructor(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
285 with FileInput(files=[]) as fi:
68530
3e783a029869 Move off of assertEquals() and over to assertEqual().
Brett Cannon <brett@python.org>
parents: 68497
diff changeset
286 self.assertEqual(fi._files, ('-',))
68495
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
287
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
288 def test__getitem__(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
289 """Tests invoking FileInput.__getitem__() with the current
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
290 line number"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
291 t = writeTmp(1, ["line1\n", "line2\n"])
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
292 self.addCleanup(remove_tempfiles, t)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
293 with FileInput(files=[t]) as fi:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
294 retval1 = fi[0]
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
295 self.assertEqual(retval1, "line1\n")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
296 retval2 = fi[1]
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
297 self.assertEqual(retval2, "line2\n")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
298
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
299 def test__getitem__invalid_key(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
300 """Tests invoking FileInput.__getitem__() with an index unequal to
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
301 the line number"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
302 t = writeTmp(1, ["line1\n", "line2\n"])
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
303 self.addCleanup(remove_tempfiles, t)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
304 with FileInput(files=[t]) as fi:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
305 with self.assertRaises(RuntimeError) as cm:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
306 fi[1]
68530
3e783a029869 Move off of assertEquals() and over to assertEqual().
Brett Cannon <brett@python.org>
parents: 68497
diff changeset
307 self.assertEqual(cm.exception.args, ("accessing lines out of order",))
68495
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
308
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
309 def test__getitem__eof(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
310 """Tests invoking FileInput.__getitem__() with the line number but at
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
311 end-of-input"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
312 t = writeTmp(1, [])
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
313 self.addCleanup(remove_tempfiles, t)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
314 with FileInput(files=[t]) as fi:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
315 with self.assertRaises(IndexError) as cm:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
316 fi[0]
68530
3e783a029869 Move off of assertEquals() and over to assertEqual().
Brett Cannon <brett@python.org>
parents: 68497
diff changeset
317 self.assertEqual(cm.exception.args, ("end of input reached",))
68495
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
318
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
319 def test_nextfile_oserror_deleting_backup(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
320 """Tests invoking FileInput.nextfile() when the attempt to delete
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
321 the backup file would raise OSError. This error is expected to be
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
322 silently ignored"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
323
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
324 os_unlink_orig = os.unlink
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
325 os_unlink_replacement = UnconditionallyRaise(OSError)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
326 try:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
327 t = writeTmp(1, ["\n"])
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
328 self.addCleanup(remove_tempfiles, t)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
329 with FileInput(files=[t], inplace=True) as fi:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
330 next(fi) # make sure the file is opened
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
331 os.unlink = os_unlink_replacement
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
332 fi.nextfile()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
333 finally:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
334 os.unlink = os_unlink_orig
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
335
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
336 # sanity check to make sure that our test scenario was actually hit
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
337 self.assertTrue(os_unlink_replacement.invoked,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
338 "os.unlink() was not invoked")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
339
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
340 def test_readline_os_fstat_raises_OSError(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
341 """Tests invoking FileInput.readline() when os.fstat() raises OSError.
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
342 This exception should be silently discarded."""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
343
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
344 os_fstat_orig = os.fstat
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
345 os_fstat_replacement = UnconditionallyRaise(OSError)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
346 try:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
347 t = writeTmp(1, ["\n"])
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
348 self.addCleanup(remove_tempfiles, t)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
349 with FileInput(files=[t], inplace=True) as fi:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
350 os.fstat = os_fstat_replacement
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
351 fi.readline()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
352 finally:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
353 os.fstat = os_fstat_orig
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
354
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
355 # sanity check to make sure that our test scenario was actually hit
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
356 self.assertTrue(os_fstat_replacement.invoked,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
357 "os.fstat() was not invoked")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
358
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
359 @unittest.skipIf(not hasattr(os, "chmod"), "os.chmod does not exist")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
360 def test_readline_os_chmod_raises_OSError(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
361 """Tests invoking FileInput.readline() when os.chmod() raises OSError.
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
362 This exception should be silently discarded."""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
363
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
364 os_chmod_orig = os.chmod
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
365 os_chmod_replacement = UnconditionallyRaise(OSError)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
366 try:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
367 t = writeTmp(1, ["\n"])
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
368 self.addCleanup(remove_tempfiles, t)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
369 with FileInput(files=[t], inplace=True) as fi:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
370 os.chmod = os_chmod_replacement
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
371 fi.readline()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
372 finally:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
373 os.chmod = os_chmod_orig
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
374
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
375 # sanity check to make sure that our test scenario was actually hit
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
376 self.assertTrue(os_chmod_replacement.invoked,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
377 "os.fstat() was not invoked")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
378
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
379 def test_fileno_when_ValueError_raised(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
380 class FilenoRaisesValueError(UnconditionallyRaise):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
381 def __init__(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
382 UnconditionallyRaise.__init__(self, ValueError)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
383 def fileno(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
384 self.__call__()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
385
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
386 unconditionally_raise_ValueError = FilenoRaisesValueError()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
387 t = writeTmp(1, ["\n"])
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
388 self.addCleanup(remove_tempfiles, t)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
389 with FileInput(files=[t]) as fi:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
390 file_backup = fi._file
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
391 try:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
392 fi._file = unconditionally_raise_ValueError
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
393 result = fi.fileno()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
394 finally:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
395 fi._file = file_backup # make sure the file gets cleaned up
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
396
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
397 # sanity check to make sure that our test scenario was actually hit
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
398 self.assertTrue(unconditionally_raise_ValueError.invoked,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
399 "_file.fileno() was not invoked")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
400
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
401 self.assertEqual(result, -1, "fileno() should return -1")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
402
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
403 class MockFileInput:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
404 """A class that mocks out fileinput.FileInput for use during unit tests"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
405
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
406 def __init__(self, files=None, inplace=False, backup="", bufsize=0,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
407 mode="r", openhook=None):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
408 self.files = files
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
409 self.inplace = inplace
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
410 self.backup = backup
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
411 self.bufsize = bufsize
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
412 self.mode = mode
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
413 self.openhook = openhook
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
414 self._file = None
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
415 self.invocation_counts = collections.defaultdict(lambda: 0)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
416 self.return_values = {}
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
417
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
418 def close(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
419 self.invocation_counts["close"] += 1
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
420
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
421 def nextfile(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
422 self.invocation_counts["nextfile"] += 1
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
423 return self.return_values["nextfile"]
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
424
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
425 def filename(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
426 self.invocation_counts["filename"] += 1
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
427 return self.return_values["filename"]
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
428
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
429 def lineno(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
430 self.invocation_counts["lineno"] += 1
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
431 return self.return_values["lineno"]
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
432
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
433 def filelineno(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
434 self.invocation_counts["filelineno"] += 1
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
435 return self.return_values["filelineno"]
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
436
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
437 def fileno(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
438 self.invocation_counts["fileno"] += 1
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
439 return self.return_values["fileno"]
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
440
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
441 def isfirstline(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
442 self.invocation_counts["isfirstline"] += 1
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
443 return self.return_values["isfirstline"]
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
444
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
445 def isstdin(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
446 self.invocation_counts["isstdin"] += 1
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
447 return self.return_values["isstdin"]
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
448
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
449 class BaseFileInputGlobalMethodsTest(unittest.TestCase):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
450 """Base class for unit tests for the global function of
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
451 the fileinput module."""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
452
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
453 def setUp(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
454 self._orig_state = fileinput._state
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
455 self._orig_FileInput = fileinput.FileInput
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
456 fileinput.FileInput = MockFileInput
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
457
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
458 def tearDown(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
459 fileinput.FileInput = self._orig_FileInput
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
460 fileinput._state = self._orig_state
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
461
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
462 def assertExactlyOneInvocation(self, mock_file_input, method_name):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
463 # assert that the method with the given name was invoked once
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
464 actual_count = mock_file_input.invocation_counts[method_name]
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
465 self.assertEqual(actual_count, 1, method_name)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
466 # assert that no other unexpected methods were invoked
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
467 actual_total_count = len(mock_file_input.invocation_counts)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
468 self.assertEqual(actual_total_count, 1)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
469
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
470 class Test_fileinput_input(BaseFileInputGlobalMethodsTest):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
471 """Unit tests for fileinput.input()"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
472
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
473 def test_state_is_not_None_and_state_file_is_not_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
474 """Tests invoking fileinput.input() when fileinput._state is not None
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
475 and its _file attribute is also not None. Expect RuntimeError to
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
476 be raised with a meaningful error message and for fileinput._state
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
477 to *not* be modified."""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
478 instance = MockFileInput()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
479 instance._file = object()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
480 fileinput._state = instance
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
481 with self.assertRaises(RuntimeError) as cm:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
482 fileinput.input()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
483 self.assertEqual(("input() already active",), cm.exception.args)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
484 self.assertIs(instance, fileinput._state, "fileinput._state")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
485
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
486 def test_state_is_not_None_and_state_file_is_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
487 """Tests invoking fileinput.input() when fileinput._state is not None
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
488 but its _file attribute *is* None. Expect it to create and return
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
489 a new fileinput.FileInput object with all method parameters passed
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
490 explicitly to the __init__() method; also ensure that
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
491 fileinput._state is set to the returned instance."""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
492 instance = MockFileInput()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
493 instance._file = None
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
494 fileinput._state = instance
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
495 self.do_test_call_input()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
496
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
497 def test_state_is_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
498 """Tests invoking fileinput.input() when fileinput._state is None
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
499 Expect it to create and return a new fileinput.FileInput object
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
500 with all method parameters passed explicitly to the __init__()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
501 method; also ensure that fileinput._state is set to the returned
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
502 instance."""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
503 fileinput._state = None
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
504 self.do_test_call_input()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
505
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
506 def do_test_call_input(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
507 """Tests that fileinput.input() creates a new fileinput.FileInput
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
508 object, passing the given parameters unmodified to
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
509 fileinput.FileInput.__init__(). Note that this test depends on the
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
510 monkey patching of fileinput.FileInput done by setUp()."""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
511 files = object()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
512 inplace = object()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
513 backup = object()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
514 bufsize = object()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
515 mode = object()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
516 openhook = object()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
517
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
518 # call fileinput.input() with different values for each argument
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
519 result = fileinput.input(files=files, inplace=inplace, backup=backup,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
520 bufsize=bufsize,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
521 mode=mode, openhook=openhook)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
522
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
523 # ensure fileinput._state was set to the returned object
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
524 self.assertIs(result, fileinput._state, "fileinput._state")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
525
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
526 # ensure the parameters to fileinput.input() were passed directly
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
527 # to FileInput.__init__()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
528 self.assertIs(files, result.files, "files")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
529 self.assertIs(inplace, result.inplace, "inplace")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
530 self.assertIs(backup, result.backup, "backup")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
531 self.assertIs(bufsize, result.bufsize, "bufsize")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
532 self.assertIs(mode, result.mode, "mode")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
533 self.assertIs(openhook, result.openhook, "openhook")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
534
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
535 class Test_fileinput_close(BaseFileInputGlobalMethodsTest):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
536 """Unit tests for fileinput.close()"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
537
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
538 def test_state_is_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
539 """Tests that fileinput.close() does nothing if fileinput._state
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
540 is None"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
541 fileinput._state = None
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
542 fileinput.close()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
543 self.assertIsNone(fileinput._state)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
544
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
545 def test_state_is_not_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
546 """Tests that fileinput.close() invokes close() on fileinput._state
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
547 and sets _state=None"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
548 instance = MockFileInput()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
549 fileinput._state = instance
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
550 fileinput.close()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
551 self.assertExactlyOneInvocation(instance, "close")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
552 self.assertIsNone(fileinput._state)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
553
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
554 class Test_fileinput_nextfile(BaseFileInputGlobalMethodsTest):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
555 """Unit tests for fileinput.nextfile()"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
556
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
557 def test_state_is_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
558 """Tests fileinput.nextfile() when fileinput._state is None.
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
559 Ensure that it raises RuntimeError with a meaningful error message
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
560 and does not modify fileinput._state"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
561 fileinput._state = None
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
562 with self.assertRaises(RuntimeError) as cm:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
563 fileinput.nextfile()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
564 self.assertEqual(("no active input()",), cm.exception.args)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
565 self.assertIsNone(fileinput._state)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
566
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
567 def test_state_is_not_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
568 """Tests fileinput.nextfile() when fileinput._state is not None.
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
569 Ensure that it invokes fileinput._state.nextfile() exactly once,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
570 returns whatever it returns, and does not modify fileinput._state
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
571 to point to a different object."""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
572 nextfile_retval = object()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
573 instance = MockFileInput()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
574 instance.return_values["nextfile"] = nextfile_retval
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
575 fileinput._state = instance
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
576 retval = fileinput.nextfile()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
577 self.assertExactlyOneInvocation(instance, "nextfile")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
578 self.assertIs(retval, nextfile_retval)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
579 self.assertIs(fileinput._state, instance)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
580
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
581 class Test_fileinput_filename(BaseFileInputGlobalMethodsTest):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
582 """Unit tests for fileinput.filename()"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
583
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
584 def test_state_is_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
585 """Tests fileinput.filename() when fileinput._state is None.
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
586 Ensure that it raises RuntimeError with a meaningful error message
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
587 and does not modify fileinput._state"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
588 fileinput._state = None
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
589 with self.assertRaises(RuntimeError) as cm:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
590 fileinput.filename()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
591 self.assertEqual(("no active input()",), cm.exception.args)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
592 self.assertIsNone(fileinput._state)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
593
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
594 def test_state_is_not_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
595 """Tests fileinput.filename() when fileinput._state is not None.
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
596 Ensure that it invokes fileinput._state.filename() exactly once,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
597 returns whatever it returns, and does not modify fileinput._state
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
598 to point to a different object."""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
599 filename_retval = object()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
600 instance = MockFileInput()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
601 instance.return_values["filename"] = filename_retval
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
602 fileinput._state = instance
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
603 retval = fileinput.filename()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
604 self.assertExactlyOneInvocation(instance, "filename")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
605 self.assertIs(retval, filename_retval)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
606 self.assertIs(fileinput._state, instance)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
607
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
608 class Test_fileinput_lineno(BaseFileInputGlobalMethodsTest):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
609 """Unit tests for fileinput.lineno()"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
610
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
611 def test_state_is_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
612 """Tests fileinput.lineno() when fileinput._state is None.
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
613 Ensure that it raises RuntimeError with a meaningful error message
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
614 and does not modify fileinput._state"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
615 fileinput._state = None
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
616 with self.assertRaises(RuntimeError) as cm:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
617 fileinput.lineno()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
618 self.assertEqual(("no active input()",), cm.exception.args)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
619 self.assertIsNone(fileinput._state)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
620
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
621 def test_state_is_not_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
622 """Tests fileinput.lineno() when fileinput._state is not None.
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
623 Ensure that it invokes fileinput._state.lineno() exactly once,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
624 returns whatever it returns, and does not modify fileinput._state
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
625 to point to a different object."""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
626 lineno_retval = object()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
627 instance = MockFileInput()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
628 instance.return_values["lineno"] = lineno_retval
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
629 fileinput._state = instance
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
630 retval = fileinput.lineno()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
631 self.assertExactlyOneInvocation(instance, "lineno")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
632 self.assertIs(retval, lineno_retval)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
633 self.assertIs(fileinput._state, instance)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
634
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
635 class Test_fileinput_filelineno(BaseFileInputGlobalMethodsTest):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
636 """Unit tests for fileinput.filelineno()"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
637
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
638 def test_state_is_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
639 """Tests fileinput.filelineno() when fileinput._state is None.
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
640 Ensure that it raises RuntimeError with a meaningful error message
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
641 and does not modify fileinput._state"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
642 fileinput._state = None
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
643 with self.assertRaises(RuntimeError) as cm:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
644 fileinput.filelineno()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
645 self.assertEqual(("no active input()",), cm.exception.args)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
646 self.assertIsNone(fileinput._state)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
647
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
648 def test_state_is_not_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
649 """Tests fileinput.filelineno() when fileinput._state is not None.
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
650 Ensure that it invokes fileinput._state.filelineno() exactly once,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
651 returns whatever it returns, and does not modify fileinput._state
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
652 to point to a different object."""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
653 filelineno_retval = object()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
654 instance = MockFileInput()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
655 instance.return_values["filelineno"] = filelineno_retval
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
656 fileinput._state = instance
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
657 retval = fileinput.filelineno()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
658 self.assertExactlyOneInvocation(instance, "filelineno")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
659 self.assertIs(retval, filelineno_retval)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
660 self.assertIs(fileinput._state, instance)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
661
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
662 class Test_fileinput_fileno(BaseFileInputGlobalMethodsTest):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
663 """Unit tests for fileinput.fileno()"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
664
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
665 def test_state_is_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
666 """Tests fileinput.fileno() when fileinput._state is None.
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
667 Ensure that it raises RuntimeError with a meaningful error message
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
668 and does not modify fileinput._state"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
669 fileinput._state = None
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
670 with self.assertRaises(RuntimeError) as cm:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
671 fileinput.fileno()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
672 self.assertEqual(("no active input()",), cm.exception.args)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
673 self.assertIsNone(fileinput._state)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
674
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
675 def test_state_is_not_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
676 """Tests fileinput.fileno() when fileinput._state is not None.
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
677 Ensure that it invokes fileinput._state.fileno() exactly once,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
678 returns whatever it returns, and does not modify fileinput._state
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
679 to point to a different object."""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
680 fileno_retval = object()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
681 instance = MockFileInput()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
682 instance.return_values["fileno"] = fileno_retval
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
683 instance.fileno_retval = fileno_retval
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
684 fileinput._state = instance
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
685 retval = fileinput.fileno()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
686 self.assertExactlyOneInvocation(instance, "fileno")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
687 self.assertIs(retval, fileno_retval)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
688 self.assertIs(fileinput._state, instance)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
689
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
690 class Test_fileinput_isfirstline(BaseFileInputGlobalMethodsTest):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
691 """Unit tests for fileinput.isfirstline()"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
692
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
693 def test_state_is_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
694 """Tests fileinput.isfirstline() when fileinput._state is None.
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
695 Ensure that it raises RuntimeError with a meaningful error message
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
696 and does not modify fileinput._state"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
697 fileinput._state = None
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
698 with self.assertRaises(RuntimeError) as cm:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
699 fileinput.isfirstline()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
700 self.assertEqual(("no active input()",), cm.exception.args)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
701 self.assertIsNone(fileinput._state)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
702
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
703 def test_state_is_not_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
704 """Tests fileinput.isfirstline() when fileinput._state is not None.
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
705 Ensure that it invokes fileinput._state.isfirstline() exactly once,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
706 returns whatever it returns, and does not modify fileinput._state
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
707 to point to a different object."""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
708 isfirstline_retval = object()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
709 instance = MockFileInput()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
710 instance.return_values["isfirstline"] = isfirstline_retval
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
711 fileinput._state = instance
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
712 retval = fileinput.isfirstline()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
713 self.assertExactlyOneInvocation(instance, "isfirstline")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
714 self.assertIs(retval, isfirstline_retval)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
715 self.assertIs(fileinput._state, instance)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
716
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
717 class Test_fileinput_isstdin(BaseFileInputGlobalMethodsTest):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
718 """Unit tests for fileinput.isstdin()"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
719
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
720 def test_state_is_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
721 """Tests fileinput.isstdin() when fileinput._state is None.
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
722 Ensure that it raises RuntimeError with a meaningful error message
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
723 and does not modify fileinput._state"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
724 fileinput._state = None
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
725 with self.assertRaises(RuntimeError) as cm:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
726 fileinput.isstdin()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
727 self.assertEqual(("no active input()",), cm.exception.args)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
728 self.assertIsNone(fileinput._state)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
729
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
730 def test_state_is_not_None(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
731 """Tests fileinput.isstdin() when fileinput._state is not None.
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
732 Ensure that it invokes fileinput._state.isstdin() exactly once,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
733 returns whatever it returns, and does not modify fileinput._state
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
734 to point to a different object."""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
735 isstdin_retval = object()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
736 instance = MockFileInput()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
737 instance.return_values["isstdin"] = isstdin_retval
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
738 fileinput._state = instance
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
739 retval = fileinput.isstdin()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
740 self.assertExactlyOneInvocation(instance, "isstdin")
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
741 self.assertIs(retval, isstdin_retval)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
742 self.assertIs(fileinput._state, instance)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
743
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
744 class InvocationRecorder:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
745 def __init__(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
746 self.invocation_count = 0
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
747 def __call__(self, *args, **kwargs):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
748 self.invocation_count += 1
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
749 self.last_invocation = (args, kwargs)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
750
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
751 class Test_hook_compressed(unittest.TestCase):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
752 """Unit tests for fileinput.hook_compressed()"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
753
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
754 def setUp(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
755 self.fake_open = InvocationRecorder()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
756
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
757 def test_empty_string(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
758 self.do_test_use_builtin_open("", 1)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
759
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
760 def test_no_ext(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
761 self.do_test_use_builtin_open("abcd", 2)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
762
70096
e0b4ed75b8e7 Skip test in test_fileinput when zlib is missing.
Ezio Melotti <ezio.melotti@gmail.com>
parents: 68669
diff changeset
763 @unittest.skipUnless(gzip, "Requires gzip and zlib")
68497
d1acf761a911 Adjust test names to be more descriptive instead of depending on capitalization.
briancurtin <brian.curtin@gmail.com>
parents: 68496
diff changeset
764 def test_gz_ext_fake(self):
68495
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
765 original_open = gzip.open
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
766 gzip.open = self.fake_open
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
767 try:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
768 result = fileinput.hook_compressed("test.gz", 3)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
769 finally:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
770 gzip.open = original_open
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
771
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
772 self.assertEqual(self.fake_open.invocation_count, 1)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
773 self.assertEqual(self.fake_open.last_invocation, (("test.gz", 3), {}))
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
774
68669
bb645cc39e60 Fix #11596. When bz2 isn't available, skip test_bz2_ext_fake.
briancurtin <brian.curtin@gmail.com>
parents: 68530
diff changeset
775 @unittest.skipUnless(bz2, "Requires bz2")
68497
d1acf761a911 Adjust test names to be more descriptive instead of depending on capitalization.
briancurtin <brian.curtin@gmail.com>
parents: 68496
diff changeset
776 def test_bz2_ext_fake(self):
68495
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
777 original_open = bz2.BZ2File
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
778 bz2.BZ2File = self.fake_open
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
779 try:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
780 result = fileinput.hook_compressed("test.bz2", 4)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
781 finally:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
782 bz2.BZ2File = original_open
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
783
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
784 self.assertEqual(self.fake_open.invocation_count, 1)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
785 self.assertEqual(self.fake_open.last_invocation, (("test.bz2", 4), {}))
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
786
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
787 def test_blah_ext(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
788 self.do_test_use_builtin_open("abcd.blah", 5)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
789
68497
d1acf761a911 Adjust test names to be more descriptive instead of depending on capitalization.
briancurtin <brian.curtin@gmail.com>
parents: 68496
diff changeset
790 def test_gz_ext_builtin(self):
68495
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
791 self.do_test_use_builtin_open("abcd.Gz", 6)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
792
68497
d1acf761a911 Adjust test names to be more descriptive instead of depending on capitalization.
briancurtin <brian.curtin@gmail.com>
parents: 68496
diff changeset
793 def test_bz2_ext_builtin(self):
68495
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
794 self.do_test_use_builtin_open("abcd.Bz2", 7)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
795
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
796 def do_test_use_builtin_open(self, filename, mode):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
797 original_open = self.replace_builtin_open(self.fake_open)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
798 try:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
799 result = fileinput.hook_compressed(filename, mode)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
800 finally:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
801 self.replace_builtin_open(original_open)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
802
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
803 self.assertEqual(self.fake_open.invocation_count, 1)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
804 self.assertEqual(self.fake_open.last_invocation,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
805 ((filename, mode), {}))
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
806
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
807 @staticmethod
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
808 def replace_builtin_open(new_open_func):
73434
c2b3e8b437a4 fileinput: replace this last occurence of codecs.open with builtins.open.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 70096
diff changeset
809 original_open = builtins.open
c2b3e8b437a4 fileinput: replace this last occurence of codecs.open with builtins.open.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 70096
diff changeset
810 builtins.open = new_open_func
68495
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
811 return original_open
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
812
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
813 class Test_hook_encoded(unittest.TestCase):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
814 """Unit tests for fileinput.hook_encoded()"""
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
815
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
816 def test(self):
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
817 encoding = object()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
818 result = fileinput.hook_encoded(encoding)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
819
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
820 fake_open = InvocationRecorder()
73434
c2b3e8b437a4 fileinput: replace this last occurence of codecs.open with builtins.open.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 70096
diff changeset
821 original_open = builtins.open
c2b3e8b437a4 fileinput: replace this last occurence of codecs.open with builtins.open.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 70096
diff changeset
822 builtins.open = fake_open
68495
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
823 try:
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
824 filename = object()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
825 mode = object()
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
826 open_result = result(filename, mode)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
827 finally:
73434
c2b3e8b437a4 fileinput: replace this last occurence of codecs.open with builtins.open.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 70096
diff changeset
828 builtins.open = original_open
68495
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
829
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
830 self.assertEqual(fake_open.invocation_count, 1)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
831
73434
c2b3e8b437a4 fileinput: replace this last occurence of codecs.open with builtins.open.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 70096
diff changeset
832 args, kwargs = fake_open.last_invocation
68495
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
833 self.assertIs(args[0], filename)
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
834 self.assertIs(args[1], mode)
73434
c2b3e8b437a4 fileinput: replace this last occurence of codecs.open with builtins.open.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 70096
diff changeset
835 self.assertIs(kwargs.pop('encoding'), encoding)
c2b3e8b437a4 fileinput: replace this last occurence of codecs.open with builtins.open.
Florent Xicluna <florent.xicluna@gmail.com>
parents: 70096
diff changeset
836 self.assertFalse(kwargs)
63379
ab89d0f52a7f #1286: allow using fileinput.FileInput as context manager.
Georg Brandl <georg@python.org>
parents: 55593
diff changeset
837
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
838 def test_main():
68495
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
839 run_unittest(
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
840 BufferSizesTests,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
841 FileInputTests,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
842 Test_fileinput_input,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
843 Test_fileinput_close,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
844 Test_fileinput_nextfile,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
845 Test_fileinput_filename,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
846 Test_fileinput_lineno,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
847 Test_fileinput_filelineno,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
848 Test_fileinput_fileno,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
849 Test_fileinput_isfirstline,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
850 Test_fileinput_isstdin,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
851 Test_hook_compressed,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
852 Test_hook_encoded,
f2e04ce75f0b Fix #11509. Significantly increase test coverage for fileinput.
briancurtin <brian.curtin@gmail.com>
parents: 63379
diff changeset
853 )
41554
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
854
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
855 if __name__ == "__main__":
06145fbc7ab9 Merged revisions 53952-54987 via svnmerge from
Guido van Rossum <guido@python.org>
parents: 41501
diff changeset
856 test_main()