test_compliance.py

# -*- coding: utf-8 -*-
"""Testcase for the magcreator module."""

import datetime
import sys
import os
import unittest
import pep8


class TestCaseCompliance(unittest.TestCase):
    """
    Class for checking compliance of pyjurassic.
    """
    
    def setUp(self):
        pass
    
    def tearDown(self):
        pass
    
    
    def get_files_to_check(self, rootdir):
        filepaths = []
        for root, dirs, files in os.walk(rootdir):
            for filename in files:
                if filename.endswith('.py'):
                    filepaths.append(os.path.join(root,filename))
        return filepaths
    
    def test_pep8(self):
        # TODO: Docstring
        files = self.get_files_to_check('..')  # search in pyramid package   
        ignores = ('E226',)# 'E201', 'E203')        
        pep8.MAX_LINE_LENGTH = 99
        pep8style = pep8.StyleGuide(quiet=False)
        pep8style.options.ignore = ignores
        
        stdout_buffer = sys.stdout
        with open(os.path.join('..','..','output','pep8_log.txt'), 'w') as sys.stdout:
            print '<<< PEP8 LOGFILE >>>' 
            print 'RUN:', datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
            print 'IGNORED RULES:', ignores
            print 'MAX LINE LENGTH:', pep8.MAX_LINE_LENGTH          
            print '\nERRORS AND WARNINGS:'
            result= pep8style.check_files(files)
            if result.total_errors == 0:
                print 'No Warnings or Errors detected!'
        sys.stdout = stdout_buffer
        
        error_message = 'Found %s Errors and Warnings!' % result.total_errors
        self.assertEqual(result.total_errors, 0, error_message)
    

if __name__ == '__main__':
    suite = unittest.TestLoader().loadTestsFromTestCase(TestCaseCompliance)
    unittest.TextTestRunner(verbosity=2).run(suite)