class TestFileExtensionIs(unittest.TestCase): def setUp(self): self.rule = FileExtensionIs('.log .txt') def test_extensions_are_split(self): self.assertEqual(self.rule.extensions, ['.log', '.txt']) def test_case_sensitive_match(self): self.rule.case_sensitive = True self.assertTrue(self.rule.test('/tmp/test_file.log')) self.assertTrue(self.rule.test('/tmp/test_file.txt')) self.assertFalse(self.rule.test('/tmp/TEST_FILE.LOG')) self.assertFalse(self.rule.test('/tmp/TEST_FILE.TxT')) def test_case_insensitive_match(self): self.rule.case_sensitive = False self.assertTrue(self.rule.test('/tmp/test_file.log')) self.assertTrue(self.rule.test('/tmp/test_file.txt')) self.assertTrue(self.rule.test('/tmp/TEST_FILE.LOG')) self.assertTrue(self.rule.test('/tmp/TEST_FILE.TXT'))
def setUp(self): self.rule = FileExtensionIs('.log .txt')