def test_init(self): f = FilteredJavaFile('file') self.assertEqual('file', f.file_name) self.assertEqual('utf-8', f.encoding) self.assertEqual([], f.contents) self.assertEqual(FilteredJavaFile.ST_START, f._state) f = FilteredJavaFile('file', 'iso-8859-1') self.assertEqual('file', f.file_name) self.assertEqual('iso-8859-1', f.encoding) self.assertEqual([], f.contents) self.assertEqual(FilteredJavaFile.ST_START, f._state)
def test_load_doxygen(self): f = FilteredJavaFile( os.path.join('sample', 'VersionInfoBeanDoxygen1.java')) f.load() self.assertEqual('/** Doxygen block', f.contents[0].strip()) f = FilteredJavaFile( os.path.join('sample', 'VersionInfoBeanDoxygen2.java')) f.load() self.assertEqual('/*! Doxygen block', f.contents[0].strip())
def test_add_license(self): f = FilteredJavaFile( os.path.join('sample', 'VersionInfoBeanLicensed.java')) f.load() self.assertEqual('package it.is.just.a.test;', f.contents[0].strip()) first_line = f.contents[0] license = ['this', 'is', 'it'] f.add_license(license) for i in range(0, len(license)): self.assertEqual(license[i], f.contents[i]) self.assertEqual(first_line, f.contents[3])
if len(sys.argv) != 2: print('Usage: {} <java file>'.format(sys.argv[0])) sys.exit(1) src_file = sys.argv[1] # Check the license file license_file = FilteredTextFile(LICENSE_FILE) try: license_file.load() except: print('License file {} could not be loaded.'.format( license_file.file_name)) sys.exit(2) # Load the java file java_file = FilteredJavaFile(src_file) try: java_file.load() except: print('Java file {} could not be loaded.'.format(java_file.file_name)) sys.exit(3) if len(java_file.contents) == 0: print('The Java file {} is empty or broken.'.format(java_file.file_name)) sys.exit(4) # Save the backup file try: java_file.save(java_file.file_name + '.bak') except: print('Unable to create the backup for the file {}.'.format( java_file.file_name))
def test_load_no_license(self): f = FilteredJavaFile(os.path.join('sample', 'VersionInfoBean.java')) f.load() self.assertEqual('package it.is.just.a.test;', f.contents[0].strip())
def test_is_multi_line_comment_start(self): f = FilteredJavaFile( os.path.join('sample', 'VersionInfoBeanLicensed.java')) self.assertTrue(f.is_multi_line_comment_start('/*')) self.assertTrue(f.is_multi_line_comment_start(' /*')) self.assertTrue(f.is_multi_line_comment_start('\t/*')) self.assertTrue(f.is_multi_line_comment_start('/* something else...')) self.assertFalse(f.is_multi_line_comment_start('\r\n')) self.assertFalse(f.is_multi_line_comment_start('\n')) self.assertFalse(f.is_multi_line_comment_start('\t\r\n')) self.assertFalse(f.is_multi_line_comment_start('\t\n')) self.assertFalse(f.is_multi_line_comment_start('package')) self.assertFalse(f.is_multi_line_comment_start(' package')) self.assertFalse(f.is_multi_line_comment_start('\tpackage')) self.assertFalse(f.is_multi_line_comment_start('/* */')) self.assertFalse(f.is_multi_line_comment_start('*/')) # Doxygen self.assertFalse(f.is_multi_line_comment_start('/**')) self.assertFalse(f.is_multi_line_comment_start(' /**')) self.assertFalse(f.is_multi_line_comment_start('\t/**')) self.assertFalse(f.is_multi_line_comment_start('/*!')) self.assertFalse(f.is_multi_line_comment_start(' /*!')) self.assertFalse(f.is_multi_line_comment_start('\t/*!'))
def test_load_with_comment3_and_license(self): f = FilteredJavaFile( os.path.join('sample', 'VersionInfoBeanWithComment3Licensed.java')) f.load() self.assertEqual('/* This should not be ignored */', f.contents[0].strip())