""" Tests for the linecache module """

import linecache
import unittest
import os.path
from test import test_support as support


FILENAME = linecache.__file__
INVALID_NAME = '!@$)(!@#_1'
EMPTY = ''
TESTS = 'cjkencodings_test inspect_fodder inspect_fodder2 mapping_tests'
TESTS = TESTS.split()
TEST_PATH = os.path.dirname(support.__file__)
MODULES = "linecache abc".split()
MODULE_PATH = os.path.dirname(FILENAME)

SOURCE_1 = '''
" Docstring "

def function():
    return result

'''

SOURCE_2 = '''
def f():
    return 1 + 1

a = f()

'''

class LineCacheTests(unittest.TestCase):

    def test_getline(self):
        getline = linecache.getline

        # Bad values for line number should return an empty string
        self.assertEquals(getline(FILENAME, 2**15), EMPTY)
        self.assertEquals(getline(FILENAME, -1), EMPTY)

        # Float values currently raise TypeError, should it?
        self.assertRaises(TypeError, getline, FILENAME, 1.1)

        # Bad filenames should return an empty string
        self.assertEquals(getline(EMPTY, 1), EMPTY)
        self.assertEquals(getline(INVALID_NAME, 1), EMPTY)

        # Check whether lines correspond to those from file iteration
        for entry in TESTS:
            filename = os.path.join(TEST_PATH, entry) + '.py'
            for index, line in enumerate(open(filename)):
                self.assertEquals(line, getline(filename, index + 1))

        # Check module loading
        for entry in MODULES:
            filename = os.path.join(MODULE_PATH, entry) + '.py'
            for index, line in enumerate(open(filename)):
                self.assertEquals(line, getline(filename, index + 1))

        # Check that bogus data isn't returned (issue #1309567)
        empty = linecache.getlines('a/b/c/__init__.py')
        self.assertEquals(empty, [])

    def test_clearcache(self):
        cached = []
        for entry in TESTS:
            filename = os.path.join(TEST_PATH, entry) + '.py'
            cached.append(filename)
            linecache.getline(filename, 1)

        # Are all files cached?
        cached_empty = [fn for fn in cached if fn not in linecache.cache]
        self.assertEquals(cached_empty, [])

        # Can we clear the cache?
        linecache.clearcache()
        cached_empty = [fn for fn in cached if fn in linecache.cache]
        self.assertEquals(cached_empty, [])

    def test_checkcache(self):
        getline = linecache.getline
        try:
            # Create a source file and cache its contents
            source_name = support.TESTFN + '.py'
            with open(source_name, 'w') as source:
                source.write(SOURCE_1)
            getline(source_name, 1)

            # Keep a copy of the old contents
            source_list = []
            with open(source_name) as source:
                for index, line in enumerate(source):
                    self.assertEquals(line, getline(source_name, index + 1))
                    source_list.append(line)

            with open(source_name, 'w') as source:
                source.write(SOURCE_2)

            # Try to update a bogus cache entry
            linecache.checkcache('dummy')

            # Check that the cache matches the old contents
            for index, line in enumerate(source_list):
                self.assertEquals(line, getline(source_name, index + 1))

            # Update the cache and check whether it matches the new source file
            linecache.checkcache(source_name)
            with open(source_name) as source:
                for index, line in enumerate(source):
                    self.assertEquals(line, getline(source_name, index + 1))
                    source_list.append(line)

        finally:
            support.unlink(source_name)

def test_main():
    support.run_unittest(LineCacheTests)

if __name__ == "__main__":
    test_main()

@@ -2438,25 +2438,8 @@
 ine%0A
-        try:%0A    
     
@@ -2492,28 +2492,24 @@
 nts%0A        
-    
 source_name 
@@ -2529,32 +2529,86 @@
 TFN + '.py'%0A    
+    self.addCleanup(test_support.unlink, source_name)%0A
         with ope
@@ -2642,36 +2642,32 @@
 ce:%0A            
-    
 source.write(SOU
@@ -2677,28 +2677,24 @@
 _1)%0A        
-   
- 
 getline(sour
@@ -2702,28 +2702,24 @@
 e_name, 1)%0A%0A
-    
         # Ke
@@ -2748,36 +2748,32 @@
 ontents%0A        
-  
-  
 source_list = %5B%5D
@@ -2769,28 +2769,24 @@
 e_list = %5B%5D%0A
-    
         with
@@ -2819,36 +2819,32 @@
 ce:%0A            
- 
-   
 for index, line 
@@ -2857,36 +2857,32 @@
 merate(source):%0A
-    
                 
@@ -2946,36 +2946,32 @@
                 
-    
 source_list.appe
@@ -2980,36 +2980,32 @@
 (line)%0A%0A        
-  
-  
 with open(source
@@ -3031,36 +3031,32 @@
 ce:%0A            
-    
 source.write(SOU
@@ -3071,20 +3071,16 @@
         
- 
-   
 # Try to
@@ -3107,20 +3107,16 @@
 e entry%0A
-    
         
@@ -3150,28 +3150,24 @@
 ')%0A%0A        
-    
 # Check that
@@ -3202,36 +3202,32 @@
 ontents%0A        
-   
- 
 for index, line 
@@ -3253,20 +3253,16 @@
 _list):%0A
-    
         
@@ -3331,20 +3331,16 @@
         
- 
-   
 # Update
@@ -3399,20 +3399,16 @@
 ce file%0A
-    
         
@@ -3445,28 +3445,24 @@
 me)%0A        
-    
 with open(so
@@ -3487,36 +3487,32 @@
 ce:%0A            
-   
- 
 for index, line 
@@ -3525,36 +3525,32 @@
 merate(source):%0A
-    
                 
@@ -3614,36 +3614,32 @@
                 
-    
 source_list.appe
@@ -3652,66 +3652,8 @@
 e)%0A%0A
-        finally:%0A            support.unlink(source_name)%0A%0A
 def 

use addCleanup