Merge pull request #572 from bfroehle/loader_usage_txt

Use `nose.__loader__` to load 'usage.txt', if available.

1 files changed, 10 insertions, 38 deletions

diff --git a/unit_tests/test_core.py b/unit_tests/test_core.py
index 6fb6cef..d84c085 100644
--- a/unit_tests/test_core.py
+++ b/unit_tests/test_core.py
@@ -33,64 +33,36 @@ class Undefined(object):
     pass
 
 class TestUsage(unittest.TestCase):
-    
+
     def test_from_directory(self):
         usage_txt = nose.core.TestProgram.usage()
         assert usage_txt.startswith('nose collects tests automatically'), (
                 "Unexpected usage: '%s...'" % usage_txt[0:50].replace("\n", '\n'))
-    
+
     def test_from_zip(self):
         requested_data = []
-        
+
         # simulates importing nose from a zip archive
         # with a zipimport.zipimporter instance
         class fake_zipimporter(object):
-            
-            prefix = ''
-            zipfile = '<fake zipfile>'
-            
+
             def get_data(self, path):
                 requested_data.append(path)
-                return "<usage>"
-                    
+                # Return as str in Python 2, bytes in Python 3.
+                return '<usage>'.encode('utf-8')
+
         existing_loader = getattr(nose, '__loader__', Undefined)
         try:
             nose.__loader__ = fake_zipimporter()
             usage_txt = nose.core.TestProgram.usage()
             self.assertEqual(usage_txt, '<usage>')
-            self.assertEqual(requested_data, ['nose%susage.txt' % os.sep])
+            self.assertEqual(requested_data, [os.path.join(
+                os.path.dirname(nose.__file__), 'usage.txt')])
         finally:
             if existing_loader is not Undefined:
                 nose.__loader__ = existing_loader
             else:
                 del nose.__loader__
-    
-    def test_from_zip_with_prefix(self):
-        requested_data = []
-        
-        # simulates importing nose from a zip archive
-        # with a zipimport.zipimporter instance
-        class fake_zipimporter(object):
-            
-            prefix = 'PREFIX'
-            zipfile = '<fake zipfile>'
-            
-            def get_data(self, path):
-                requested_data.append(path)
-                return "<usage>"
-                
-        existing_loader = getattr(nose, '__loader__', Undefined)
-        try:            
-            nose.__loader__ = fake_zipimporter()
-            usage_txt = nose.core.TestProgram.usage()
-            self.assertEqual(usage_txt, '<usage>')
-            self.assertEqual(requested_data, 
-                             ['PREFIX%snose%susage.txt' % (os.sep, os.sep)])
-        finally:
-            if existing_loader is not Undefined:
-                nose.__loader__ = existing_loader
-            else:
-                del nose.__loader__
-        
+
 if __name__ == '__main__':
     unittest.main()