blob: 2dd76ad9ff676167e5f84c34cf890b4a51d906cf [file] [log] [blame]
R. David Murray59beec32009-03-30 19:04:00 +00001from test.test_support import TESTFN, run_unittest, import_module
Georg Brandl31631792006-10-29 19:13:40 +00002import unittest
Fred Drake62787992001-05-11 14:29:21 +00003import os, re
Andrew M. Kuchlinge81b9cf2000-03-30 21:15:29 +00004
R. David Murray59beec32009-03-30 19:04:00 +00005mmap = import_module('mmap')
6
Andrew M. Kuchlinge81b9cf2000-03-30 21:15:29 +00007PAGESIZE = mmap.PAGESIZE
8
Georg Brandl31631792006-10-29 19:13:40 +00009class MmapTests(unittest.TestCase):
Fred Drake004d5e62000-10-23 17:22:08 +000010
Georg Brandl31631792006-10-29 19:13:40 +000011 def setUp(self):
12 if os.path.exists(TESTFN):
13 os.unlink(TESTFN)
Fred Drake004d5e62000-10-23 17:22:08 +000014
Georg Brandl31631792006-10-29 19:13:40 +000015 def tearDown(self):
Tim Petersfd692082001-05-10 20:03:04 +000016 try:
Fred Drake62787992001-05-11 14:29:21 +000017 os.unlink(TESTFN)
Tim Petersfd692082001-05-10 20:03:04 +000018 except OSError:
19 pass
20
Georg Brandl31631792006-10-29 19:13:40 +000021 def test_basic(self):
22 # Test mmap module on Unix systems and Windows
23
24 # Create a file to be mmap'ed.
25 f = open(TESTFN, 'w+')
26 try:
27 # Write 2 pages worth of data to the file
28 f.write('\0'* PAGESIZE)
29 f.write('foo')
30 f.write('\0'* (PAGESIZE-3) )
31 f.flush()
32 m = mmap.mmap(f.fileno(), 2 * PAGESIZE)
33 f.close()
34
35 # Simple sanity checks
36
37 tp = str(type(m)) # SF bug 128713: segfaulted on Linux
38 self.assertEqual(m.find('foo'), PAGESIZE)
39
40 self.assertEqual(len(m), 2*PAGESIZE)
41
42 self.assertEqual(m[0], '\0')
43 self.assertEqual(m[0:3], '\0\0\0')
44
Hirokazu Yamamotof6bbd0e2009-02-17 10:12:10 +000045 # Shouldn't crash on boundary (Issue #5292)
46 self.assertRaises(IndexError, m.__getitem__, len(m))
47 self.assertRaises(IndexError, m.__setitem__, len(m), '\0')
48
Georg Brandl31631792006-10-29 19:13:40 +000049 # Modify the file's content
50 m[0] = '3'
51 m[PAGESIZE +3: PAGESIZE +3+3] = 'bar'
52
53 # Check that the modification worked
54 self.assertEqual(m[0], '3')
55 self.assertEqual(m[0:3], '3\0\0')
56 self.assertEqual(m[PAGESIZE-1 : PAGESIZE + 7], '\0foobar\0')
57
58 m.flush()
59
60 # Test doing a regular expression match in an mmap'ed file
61 match = re.search('[A-Za-z]+', m)
62 if match is None:
63 self.fail('regex match on mmap failed!')
64 else:
65 start, end = match.span(0)
66 length = end - start
67
68 self.assertEqual(start, PAGESIZE)
69 self.assertEqual(end, PAGESIZE + 6)
70
71 # test seeking around (try to overflow the seek implementation)
72 m.seek(0,0)
73 self.assertEqual(m.tell(), 0)
74 m.seek(42,1)
75 self.assertEqual(m.tell(), 42)
76 m.seek(0,2)
77 self.assertEqual(m.tell(), len(m))
78
79 # Try to seek to negative position...
80 self.assertRaises(ValueError, m.seek, -1)
81
82 # Try to seek beyond end of mmap...
83 self.assertRaises(ValueError, m.seek, 1, 2)
84
85 # Try to seek to negative position...
86 self.assertRaises(ValueError, m.seek, -len(m)-1, 2)
87
88 # Try resizing map
89 try:
90 m.resize(512)
91 except SystemError:
92 # resize() not supported
93 # No messages are printed, since the output of this test suite
94 # would then be different across platforms.
95 pass
96 else:
97 # resize() is supported
98 self.assertEqual(len(m), 512)
99 # Check that we can no longer seek beyond the new size.
100 self.assertRaises(ValueError, m.seek, 513, 0)
101
102 # Check that the underlying file is truncated too
103 # (bug #728515)
104 f = open(TESTFN)
105 f.seek(0, 2)
106 self.assertEqual(f.tell(), 512)
107 f.close()
108 self.assertEqual(m.size(), 512)
109
110 m.close()
111
112 finally:
113 try:
114 f.close()
115 except OSError:
116 pass
117
118 def test_access_parameter(self):
119 # Test for "access" keyword parameter
Tim Peters5ebfd362001-11-13 23:11:19 +0000120 mapsize = 10
Tim Peters5ebfd362001-11-13 23:11:19 +0000121 open(TESTFN, "wb").write("a"*mapsize)
Tim Peters5ebfd362001-11-13 23:11:19 +0000122 f = open(TESTFN, "rb")
123 m = mmap.mmap(f.fileno(), mapsize, access=mmap.ACCESS_READ)
Georg Brandl31631792006-10-29 19:13:40 +0000124 self.assertEqual(m[:], 'a'*mapsize, "Readonly memory map data incorrect.")
Tim Peters5ebfd362001-11-13 23:11:19 +0000125
Georg Brandl31631792006-10-29 19:13:40 +0000126 # Ensuring that readonly mmap can't be slice assigned
Tim Peters5ebfd362001-11-13 23:11:19 +0000127 try:
128 m[:] = 'b'*mapsize
129 except TypeError:
130 pass
131 else:
Georg Brandl31631792006-10-29 19:13:40 +0000132 self.fail("Able to write to readonly memory map")
Tim Peters5ebfd362001-11-13 23:11:19 +0000133
Georg Brandl31631792006-10-29 19:13:40 +0000134 # Ensuring that readonly mmap can't be item assigned
Tim Peters5ebfd362001-11-13 23:11:19 +0000135 try:
136 m[0] = 'b'
137 except TypeError:
138 pass
139 else:
Georg Brandl31631792006-10-29 19:13:40 +0000140 self.fail("Able to write to readonly memory map")
Tim Peters5ebfd362001-11-13 23:11:19 +0000141
Georg Brandl31631792006-10-29 19:13:40 +0000142 # Ensuring that readonly mmap can't be write() to
Tim Peters5ebfd362001-11-13 23:11:19 +0000143 try:
144 m.seek(0,0)
145 m.write('abc')
146 except TypeError:
147 pass
148 else:
Georg Brandl31631792006-10-29 19:13:40 +0000149 self.fail("Able to write to readonly memory map")
Tim Peters5ebfd362001-11-13 23:11:19 +0000150
Georg Brandl31631792006-10-29 19:13:40 +0000151 # Ensuring that readonly mmap can't be write_byte() to
Tim Peters5ebfd362001-11-13 23:11:19 +0000152 try:
153 m.seek(0,0)
154 m.write_byte('d')
155 except TypeError:
156 pass
157 else:
Georg Brandl31631792006-10-29 19:13:40 +0000158 self.fail("Able to write to readonly memory map")
Tim Peters5ebfd362001-11-13 23:11:19 +0000159
Georg Brandl31631792006-10-29 19:13:40 +0000160 # Ensuring that readonly mmap can't be resized
Tim Peters5ebfd362001-11-13 23:11:19 +0000161 try:
162 m.resize(2*mapsize)
163 except SystemError: # resize is not universally supported
164 pass
165 except TypeError:
166 pass
167 else:
Georg Brandl31631792006-10-29 19:13:40 +0000168 self.fail("Able to resize readonly memory map")
Neal Norwitzd48a2f72008-04-01 05:40:43 +0000169 f.close()
Tim Peters5ebfd362001-11-13 23:11:19 +0000170 del m, f
Georg Brandl31631792006-10-29 19:13:40 +0000171 self.assertEqual(open(TESTFN, "rb").read(), 'a'*mapsize,
Tim Peters5ebfd362001-11-13 23:11:19 +0000172 "Readonly memory map data file was modified")
173
Georg Brandl31631792006-10-29 19:13:40 +0000174 # Opening mmap with size too big
Neal Norwitzb5673922002-09-05 21:48:07 +0000175 import sys
176 f = open(TESTFN, "r+b")
177 try:
178 m = mmap.mmap(f.fileno(), mapsize+1)
179 except ValueError:
180 # we do not expect a ValueError on Windows
Tim Peters4f4f4d72002-09-10 20:49:15 +0000181 # CAUTION: This also changes the size of the file on disk, and
182 # later tests assume that the length hasn't changed. We need to
183 # repair that.
Neal Norwitzb5673922002-09-05 21:48:07 +0000184 if sys.platform.startswith('win'):
Georg Brandl31631792006-10-29 19:13:40 +0000185 self.fail("Opening mmap with size+1 should work on Windows.")
Neal Norwitzb5673922002-09-05 21:48:07 +0000186 else:
187 # we expect a ValueError on Unix, but not on Windows
188 if not sys.platform.startswith('win'):
Georg Brandl31631792006-10-29 19:13:40 +0000189 self.fail("Opening mmap with size+1 should raise ValueError.")
Barry Warsawccd9e752002-09-11 02:56:42 +0000190 m.close()
Tim Peters4f4f4d72002-09-10 20:49:15 +0000191 f.close()
192 if sys.platform.startswith('win'):
193 # Repair damage from the resizing test.
194 f = open(TESTFN, 'r+b')
195 f.truncate(mapsize)
196 f.close()
Neal Norwitzb5673922002-09-05 21:48:07 +0000197
Georg Brandl31631792006-10-29 19:13:40 +0000198 # Opening mmap with access=ACCESS_WRITE
Tim Peters5ebfd362001-11-13 23:11:19 +0000199 f = open(TESTFN, "r+b")
200 m = mmap.mmap(f.fileno(), mapsize, access=mmap.ACCESS_WRITE)
Georg Brandl31631792006-10-29 19:13:40 +0000201 # Modifying write-through memory map
Tim Peters5ebfd362001-11-13 23:11:19 +0000202 m[:] = 'c'*mapsize
Georg Brandl31631792006-10-29 19:13:40 +0000203 self.assertEqual(m[:], 'c'*mapsize,
Tim Peters5ebfd362001-11-13 23:11:19 +0000204 "Write-through memory map memory not updated properly.")
205 m.flush()
Tim Peters1b5112a2002-09-10 21:19:55 +0000206 m.close()
207 f.close()
Tim Peters4f4f4d72002-09-10 20:49:15 +0000208 f = open(TESTFN, 'rb')
209 stuff = f.read()
210 f.close()
Georg Brandl31631792006-10-29 19:13:40 +0000211 self.assertEqual(stuff, 'c'*mapsize,
Tim Peters5ebfd362001-11-13 23:11:19 +0000212 "Write-through memory map data file not updated properly.")
213
Georg Brandl31631792006-10-29 19:13:40 +0000214 # Opening mmap with access=ACCESS_COPY
Tim Peters5ebfd362001-11-13 23:11:19 +0000215 f = open(TESTFN, "r+b")
216 m = mmap.mmap(f.fileno(), mapsize, access=mmap.ACCESS_COPY)
Georg Brandl31631792006-10-29 19:13:40 +0000217 # Modifying copy-on-write memory map
Tim Peters5ebfd362001-11-13 23:11:19 +0000218 m[:] = 'd'*mapsize
Georg Brandl31631792006-10-29 19:13:40 +0000219 self.assertEqual(m[:], 'd' * mapsize,
Tim Peters5ebfd362001-11-13 23:11:19 +0000220 "Copy-on-write memory map data not written correctly.")
221 m.flush()
Georg Brandl31631792006-10-29 19:13:40 +0000222 self.assertEqual(open(TESTFN, "rb").read(), 'c'*mapsize,
Tim Peters5ebfd362001-11-13 23:11:19 +0000223 "Copy-on-write test data file should not be modified.")
Georg Brandl31631792006-10-29 19:13:40 +0000224 # Ensuring copy-on-write maps cannot be resized
225 self.assertRaises(TypeError, m.resize, 2*mapsize)
Neal Norwitzd48a2f72008-04-01 05:40:43 +0000226 f.close()
Tim Peters5ebfd362001-11-13 23:11:19 +0000227 del m, f
Tim Petersabd8a332006-11-03 02:32:46 +0000228
Georg Brandl31631792006-10-29 19:13:40 +0000229 # Ensuring invalid access parameter raises exception
230 f = open(TESTFN, "r+b")
231 self.assertRaises(ValueError, mmap.mmap, f.fileno(), mapsize, access=4)
232 f.close()
Tim Peters5ebfd362001-11-13 23:11:19 +0000233
234 if os.name == "posix":
Tim Peters00cafa02001-11-13 23:39:47 +0000235 # Try incompatible flags, prot and access parameters.
236 f = open(TESTFN, "r+b")
Georg Brandl31631792006-10-29 19:13:40 +0000237 self.assertRaises(ValueError, mmap.mmap, f.fileno(), mapsize,
238 flags=mmap.MAP_PRIVATE,
Tim Peters5ebfd362001-11-13 23:11:19 +0000239 prot=mmap.PROT_READ, access=mmap.ACCESS_WRITE)
Tim Peters5379dea2002-04-18 04:30:18 +0000240 f.close()
Tim Peters5ebfd362001-11-13 23:11:19 +0000241
Georg Brandl31631792006-10-29 19:13:40 +0000242 def test_bad_file_desc(self):
243 # Try opening a bad file descriptor...
244 self.assertRaises(mmap.error, mmap.mmap, -2, 4096)
Neal Norwitz3b4fff82006-01-11 08:54:45 +0000245
Georg Brandl31631792006-10-29 19:13:40 +0000246 def test_tougher_find(self):
247 # Do a tougher .find() test. SF bug 515943 pointed out that, in 2.2,
248 # searching for data with embedded \0 bytes didn't work.
249 f = open(TESTFN, 'w+')
Tim Petersc9ffa062002-03-08 05:43:32 +0000250
Tim Petersc9ffa062002-03-08 05:43:32 +0000251 data = 'aabaac\x00deef\x00\x00aa\x00'
252 n = len(data)
253 f.write(data)
Tim Peters5379dea2002-04-18 04:30:18 +0000254 f.flush()
Tim Petersc9ffa062002-03-08 05:43:32 +0000255 m = mmap.mmap(f.fileno(), n)
256 f.close()
257
258 for start in range(n+1):
259 for finish in range(start, n+1):
260 slice = data[start : finish]
Georg Brandl31631792006-10-29 19:13:40 +0000261 self.assertEqual(m.find(slice), data.find(slice))
262 self.assertEqual(m.find(slice + 'x'), -1)
Tim Petersddc82ea2003-01-13 21:38:45 +0000263 m.close()
Tim Petersc9ffa062002-03-08 05:43:32 +0000264
Andrew M. Kuchling5c60bfc2008-01-19 18:18:41 +0000265 def test_find_end(self):
266 # test the new 'end' parameter works as expected
267 f = open(TESTFN, 'w+')
268 data = 'one two ones'
269 n = len(data)
270 f.write(data)
271 f.flush()
272 m = mmap.mmap(f.fileno(), n)
273 f.close()
274
275 self.assertEqual(m.find('one'), 0)
276 self.assertEqual(m.find('ones'), 8)
277 self.assertEqual(m.find('one', 0, -1), 0)
278 self.assertEqual(m.find('one', 1), 8)
279 self.assertEqual(m.find('one', 1, -1), 8)
280 self.assertEqual(m.find('one', 1, -2), -1)
281
282
283 def test_rfind(self):
284 # test the new 'end' parameter works as expected
285 f = open(TESTFN, 'w+')
286 data = 'one two ones'
287 n = len(data)
288 f.write(data)
289 f.flush()
290 m = mmap.mmap(f.fileno(), n)
291 f.close()
292
293 self.assertEqual(m.rfind('one'), 8)
294 self.assertEqual(m.rfind('one '), 0)
295 self.assertEqual(m.rfind('one', 0, -1), 8)
296 self.assertEqual(m.rfind('one', 0, -2), 0)
297 self.assertEqual(m.rfind('one', 1, -1), 8)
298 self.assertEqual(m.rfind('one', 1, -2), -1)
299
300
Georg Brandl31631792006-10-29 19:13:40 +0000301 def test_double_close(self):
302 # make sure a double close doesn't crash on Solaris (Bug# 665913)
303 f = open(TESTFN, 'w+')
Tim Petersc9ffa062002-03-08 05:43:32 +0000304
Tim Petersddc82ea2003-01-13 21:38:45 +0000305 f.write(2**16 * 'a') # Arbitrary character
Neal Norwitze604c022003-01-10 20:52:16 +0000306 f.close()
307
308 f = open(TESTFN)
Tim Petersddc82ea2003-01-13 21:38:45 +0000309 mf = mmap.mmap(f.fileno(), 2**16, access=mmap.ACCESS_READ)
Neal Norwitze604c022003-01-10 20:52:16 +0000310 mf.close()
311 mf.close()
312 f.close()
313
Georg Brandl31631792006-10-29 19:13:40 +0000314 def test_entire_file(self):
315 # test mapping of entire file by passing 0 for map length
316 if hasattr(os, "stat"):
317 f = open(TESTFN, "w+")
Tim Petersc9ffa062002-03-08 05:43:32 +0000318
Martin v. Löwis7fe60c02005-03-03 11:22:44 +0000319 f.write(2**16 * 'm') # Arbitrary character
320 f.close()
321
322 f = open(TESTFN, "rb+")
Tim Peterseba28be2005-03-28 01:08:02 +0000323 mf = mmap.mmap(f.fileno(), 0)
Georg Brandl31631792006-10-29 19:13:40 +0000324 self.assertEqual(len(mf), 2**16, "Map size should equal file size.")
325 self.assertEqual(mf.read(2**16), 2**16 * "m")
Martin v. Löwis7fe60c02005-03-03 11:22:44 +0000326 mf.close()
327 f.close()
328
Georg Brandl31631792006-10-29 19:13:40 +0000329 def test_move(self):
330 # make move works everywhere (64-bit format problem earlier)
331 f = open(TESTFN, 'w+')
Tim Peterseba28be2005-03-28 01:08:02 +0000332
Neal Norwitz8856fb72005-12-18 03:34:22 +0000333 f.write("ABCDEabcde") # Arbitrary character
334 f.flush()
335
336 mf = mmap.mmap(f.fileno(), 10)
337 mf.move(5, 0, 5)
Georg Brandl31631792006-10-29 19:13:40 +0000338 self.assertEqual(mf[:], "ABCDEABCDE", "Map move should have duplicated front 5")
Neal Norwitz8856fb72005-12-18 03:34:22 +0000339 mf.close()
340 f.close()
341
Georg Brandl31631792006-10-29 19:13:40 +0000342 def test_anonymous(self):
343 # anonymous mmap.mmap(-1, PAGE)
344 m = mmap.mmap(-1, PAGESIZE)
345 for x in xrange(PAGESIZE):
346 self.assertEqual(m[x], '\0', "anonymously mmap'ed contents should be zero")
Neal Norwitz8856fb72005-12-18 03:34:22 +0000347
Georg Brandl31631792006-10-29 19:13:40 +0000348 for x in xrange(PAGESIZE):
349 m[x] = ch = chr(x & 255)
350 self.assertEqual(m[x], ch)
Neal Norwitz0e6bc8c2006-02-05 05:45:43 +0000351
Thomas Wouters3ccec682007-08-28 15:28:19 +0000352 def test_extended_getslice(self):
353 # Test extended slicing by comparing with list slicing.
354 s = "".join(chr(c) for c in reversed(range(256)))
355 m = mmap.mmap(-1, len(s))
356 m[:] = s
357 self.assertEqual(m[:], s)
358 indices = (0, None, 1, 3, 19, 300, -1, -2, -31, -300)
359 for start in indices:
360 for stop in indices:
361 # Skip step 0 (invalid)
362 for step in indices[1:]:
363 self.assertEqual(m[start:stop:step],
364 s[start:stop:step])
365
366 def test_extended_set_del_slice(self):
367 # Test extended slicing by comparing with list slicing.
368 s = "".join(chr(c) for c in reversed(range(256)))
369 m = mmap.mmap(-1, len(s))
370 indices = (0, None, 1, 3, 19, 300, -1, -2, -31, -300)
371 for start in indices:
372 for stop in indices:
373 # Skip invalid step 0
374 for step in indices[1:]:
375 m[:] = s
376 self.assertEqual(m[:], s)
377 L = list(s)
378 # Make sure we have a slice of exactly the right length,
379 # but with different data.
380 data = L[start:stop:step]
381 data = "".join(reversed(data))
382 L[start:stop:step] = data
383 m[start:stop:step] = data
384 self.assertEquals(m[:], "".join(L))
385
Travis E. Oliphant8feafab2007-10-23 02:40:56 +0000386 def make_mmap_file (self, f, halfsize):
387 # Write 2 pages worth of data to the file
388 f.write ('\0' * halfsize)
389 f.write ('foo')
390 f.write ('\0' * (halfsize - 3))
391 f.flush ()
392 return mmap.mmap (f.fileno(), 0)
393
394 def test_offset (self):
395 f = open (TESTFN, 'w+b')
396
397 try: # unlink TESTFN no matter what
398 halfsize = mmap.ALLOCATIONGRANULARITY
399 m = self.make_mmap_file (f, halfsize)
400 m.close ()
401 f.close ()
402
403 mapsize = halfsize * 2
404 # Try invalid offset
405 f = open(TESTFN, "r+b")
406 for offset in [-2, -1, None]:
407 try:
408 m = mmap.mmap(f.fileno(), mapsize, offset=offset)
409 self.assertEqual(0, 1)
410 except (ValueError, TypeError, OverflowError):
411 pass
412 else:
413 self.assertEqual(0, 0)
414 f.close()
415
416 # Try valid offset, hopefully 8192 works on all OSes
417 f = open(TESTFN, "r+b")
418 m = mmap.mmap(f.fileno(), mapsize - halfsize, offset=halfsize)
419 self.assertEqual(m[0:3], 'foo')
420 f.close()
Hirokazu Yamamoto17a837e2009-02-17 13:17:26 +0000421
422 # Try resizing map
423 try:
424 m.resize(512)
425 except SystemError:
426 pass
427 else:
428 # resize() is supported
429 self.assertEqual(len(m), 512)
430 # Check that we can no longer seek beyond the new size.
431 self.assertRaises(ValueError, m.seek, 513, 0)
432 # Check that the content is not changed
433 self.assertEqual(m[0:3], 'foo')
434
435 # Check that the underlying file is truncated too
436 f = open(TESTFN)
437 f.seek(0, 2)
438 self.assertEqual(f.tell(), halfsize + 512)
439 f.close()
440 self.assertEqual(m.size(), halfsize + 512)
441
Travis E. Oliphant8feafab2007-10-23 02:40:56 +0000442 m.close()
443
444 finally:
445 f.close()
446 try:
447 os.unlink(TESTFN)
448 except OSError:
449 pass
450
Georg Brandld02fc482008-01-22 19:56:03 +0000451 def test_subclass(self):
452 class anon_mmap(mmap.mmap):
453 def __new__(klass, *args, **kwargs):
454 return mmap.mmap.__new__(klass, -1, *args, **kwargs)
455 anon_mmap(PAGESIZE)
456
Christian Heimes7adfad82008-02-15 08:20:11 +0000457 def test_prot_readonly(self):
Amaury Forgeot d'Arc64d68432008-02-16 00:16:50 +0000458 if not hasattr(mmap, 'PROT_READ'):
459 return
Christian Heimes7adfad82008-02-15 08:20:11 +0000460 mapsize = 10
461 open(TESTFN, "wb").write("a"*mapsize)
462 f = open(TESTFN, "rb")
463 m = mmap.mmap(f.fileno(), mapsize, prot=mmap.PROT_READ)
464 self.assertRaises(TypeError, m.write, "foo")
Neal Norwitzd48a2f72008-04-01 05:40:43 +0000465 f.close()
Georg Brandld02fc482008-01-22 19:56:03 +0000466
Facundo Batistae1396882008-02-17 18:59:29 +0000467 def test_error(self):
468 self.assert_(issubclass(mmap.error, EnvironmentError))
469 self.assert_("mmap.error" in str(mmap.error))
470
Hirokazu Yamamotof2dc8852009-02-28 10:31:54 +0000471 def test_io_methods(self):
472 data = "0123456789"
473 open(TESTFN, "wb").write("x"*len(data))
474 f = open(TESTFN, "r+b")
475 m = mmap.mmap(f.fileno(), len(data))
476 f.close()
477 # Test write_byte()
478 for i in xrange(len(data)):
479 self.assertEquals(m.tell(), i)
480 m.write_byte(data[i:i+1])
481 self.assertEquals(m.tell(), i+1)
482 self.assertRaises(ValueError, m.write_byte, "x")
483 self.assertEquals(m[:], data)
484 # Test read_byte()
485 m.seek(0)
486 for i in xrange(len(data)):
487 self.assertEquals(m.tell(), i)
488 self.assertEquals(m.read_byte(), data[i:i+1])
489 self.assertEquals(m.tell(), i+1)
490 self.assertRaises(ValueError, m.read_byte)
491 # Test read()
492 m.seek(3)
493 self.assertEquals(m.read(3), "345")
494 self.assertEquals(m.tell(), 6)
495 # Test write()
496 m.seek(3)
497 m.write("bar")
498 self.assertEquals(m.tell(), 6)
499 self.assertEquals(m[:], "012bar6789")
500 m.seek(8)
501 self.assertRaises(ValueError, m.write, "bar")
502
Hirokazu Yamamotob0e10c72009-02-28 12:13:07 +0000503 if os.name == 'nt':
504 def test_tagname(self):
505 data1 = "0123456789"
506 data2 = "abcdefghij"
507 assert len(data1) == len(data2)
Hirokazu Yamamoto264fc122009-03-05 14:21:12 +0000508
Hirokazu Yamamotob0e10c72009-02-28 12:13:07 +0000509 # Test same tag
510 m1 = mmap.mmap(-1, len(data1), tagname="foo")
511 m1[:] = data1
512 m2 = mmap.mmap(-1, len(data2), tagname="foo")
513 m2[:] = data2
514 self.assertEquals(m1[:], data2)
515 self.assertEquals(m2[:], data2)
Hirokazu Yamamoto264fc122009-03-05 14:21:12 +0000516 m2.close()
517 m1.close()
518
Hirokazu Yamamotob0e10c72009-02-28 12:13:07 +0000519 # Test differnt tag
520 m1 = mmap.mmap(-1, len(data1), tagname="foo")
521 m1[:] = data1
522 m2 = mmap.mmap(-1, len(data2), tagname="boo")
523 m2[:] = data2
524 self.assertEquals(m1[:], data1)
525 self.assertEquals(m2[:], data2)
Hirokazu Yamamoto264fc122009-03-05 14:21:12 +0000526 m2.close()
527 m1.close()
Hirokazu Yamamotob0e10c72009-02-28 12:13:07 +0000528
Hirokazu Yamamoto264fc122009-03-05 14:21:12 +0000529 def test_crasher_on_windows(self):
Hirokazu Yamamotob0e10c72009-02-28 12:13:07 +0000530 # Should not crash (Issue 1733986)
531 m = mmap.mmap(-1, 1000, tagname="foo")
532 try:
533 mmap.mmap(-1, 5000, tagname="foo")[:] # same tagname, but larger size
534 except:
535 pass
Hirokazu Yamamoto264fc122009-03-05 14:21:12 +0000536 m.close()
537
538 # Should not crash (Issue 5385)
Hirokazu Yamamoto09033062009-03-05 14:52:44 +0000539 open(TESTFN, "wb").write("x"*10)
540 f = open(TESTFN, "r+b")
541 m = mmap.mmap(f.fileno(), 0)
542 f.close()
Hirokazu Yamamoto264fc122009-03-05 14:21:12 +0000543 try:
Hirokazu Yamamoto09033062009-03-05 14:52:44 +0000544 m.resize(0) # will raise WindowsError
Hirokazu Yamamoto264fc122009-03-05 14:21:12 +0000545 except:
546 pass
547 try:
548 m[:]
549 except:
550 pass
551 m.close()
552
Facundo Batistae1396882008-02-17 18:59:29 +0000553
Georg Brandl31631792006-10-29 19:13:40 +0000554def test_main():
555 run_unittest(MmapTests)
Andrew M. Kuchlinge81b9cf2000-03-30 21:15:29 +0000556
Georg Brandl31631792006-10-29 19:13:40 +0000557if __name__ == '__main__':
558 test_main()