def test_06_new_existing_chain_fails(self): """ Chain 06: create an exsiting chain should fail """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) with self.assertRaises(ConverterError): filter.put_into_fgr("-N INPUT")
def test_08_insert_rule_fail(self): """ Chain 08: insert a rule into a non_existing chain fails """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) with self.assertRaises(ConverterError): filter.put_into_fgr("-I PUT -j ACCEPT")
def test_13_illegal_command(self): """ Chain 13: try an ilegal command """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) with self.assertRaises(ConverterError): filter.put_into_fgr("-Y USERCHAIN")
def test_06_new_existing_chain_fails(self): """ Chain 06: create an exsiting chain should fail """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) with self.assertRaises(ConverterError): filter.put_into_fgr("-N INPUT")
def test_11_remove_predef_chain(self): """ Chain 11: try to remove a prefined chain """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) with self.assertRaises(ConverterError): filter.put_into_fgr("-X INPUT")
def test_02_prove_policies(self): """ Chain 02: check 3 valid policies, 1 exception """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) filter.put_into_fgr("-P INPUT DROP") self.assertEqual( { 'FORWARD': 'ACCEPT', 'INPUT': 'DROP', 'OUTPUT': 'ACCEPT' }, filter.poli) filter.put_into_fgr("-P FORWARD REJECT") self.assertEqual( { 'FORWARD': 'REJECT', 'INPUT': 'DROP', 'OUTPUT': 'ACCEPT' }, filter.poli) filter.put_into_fgr("-P OUTPUT DROP") self.assertEqual( { 'FORWARD': 'REJECT', 'INPUT': 'DROP', 'OUTPUT': 'DROP' }, filter.poli) with self.assertRaises(ConverterError): filter.put_into_fgr("-P OUTPUT FAIL")
def test_13_illegal_command(self): """ Chain 13: try an ilegal command """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) with self.assertRaises(ConverterError): filter.put_into_fgr("-Y USERCHAIN")
def test_11_remove_predef_chain(self): """ Chain 11: try to remove a prefined chain """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) with self.assertRaises(ConverterError): filter.put_into_fgr("-X INPUT")
def test_08_insert_rule_fail(self): """ Chain 08: insert a rule into a non_existing chain fails """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) with self.assertRaises(ConverterError): filter.put_into_fgr("-I PUT -j ACCEPT")
def test_05_new_chain(self): """ Chain 05: create a new chain in filtergroup, """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) filter.put_into_fgr("-N NEWCHAIN") self.assertEqual( {'FORWARD': [], 'INPUT': [], 'NEWCHAIN': [], 'OUTPUT': []}, filter.data)
def test_09_insert_rule_works(self): """ Chain 09: insert a rule into a nonempty chain works at start """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) filter.put_into_fgr("-A INPUT -p tcp -j ACCEPT") filter.put_into_fgr("-I INPUT -p udp -j ACCEPT") filter.put_into_fgr("-I INPUT -p esp -j ACCEPT") expect = ['-I INPUT -p esp -j ACCEPT', '-I INPUT -p udp -j ACCEPT', '-A INPUT -p tcp -j ACCEPT'] self.assertEqual(expect, filter.data["INPUT"])
def test_12_remove_chain(self): """ Chain 12: try to remove an existing chain """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) filter.put_into_fgr("-N NEWCHAIN") self.assertEqual( {'FORWARD': [], 'INPUT': [], 'NEWCHAIN': [], 'OUTPUT': []}, filter.data) filter.put_into_fgr("-X NEWCHAIN") self.assertEqual( {'FORWARD': [], 'INPUT': [], 'OUTPUT': []}, filter.data)
def test_05_new_chain(self): """ Chain 05: create a new chain in filtergroup, """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) filter.put_into_fgr("-N NEWCHAIN") self.assertEqual( { 'FORWARD': [], 'INPUT': [], 'NEWCHAIN': [], 'OUTPUT': [] }, filter.data)
def test_10_append_rule(self): """ Chain 10: append a rule to a chain """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) filter.put_into_fgr("-A INPUT -p tcp -j ACCEPT") self.assertEqual(['-A INPUT -p tcp -j ACCEPT'], filter.data["INPUT"]) filter.put_into_fgr("-A INPUT -p udp -j ACCEPT") filter.put_into_fgr("-A INPUT -p esp -j ACCEPT") expect = ['-A INPUT -p tcp -j ACCEPT', '-A INPUT -p udp -j ACCEPT', '-A INPUT -p esp -j ACCEPT'] self.assertEqual(expect, filter.data["INPUT"])
def test_04_flush(self): """ Chain 04: flush filter group, 2 rules and an invalid chain """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) filter.put_into_fgr("-t filter -A INPUT -i sl0 -j ACCEPT") self.assertEqual(['-A INPUT -i sl0 -j ACCEPT '], filter.data["INPUT"]) filter.put_into_fgr("-A OUTPUT -o sl1 -j ACCEPT") self.assertEqual(['-A OUTPUT -o sl1 -j ACCEPT'], filter.data["OUTPUT"]) filter.put_into_fgr("-F") self.assertEqual([], filter.data["INPUT"]) self.assertEqual([], filter.data["OUTPUT"]) with self.assertRaises(ConverterError): filter.put_into_fgr("-t inval -F")
def test_03_tables_names(self): """ Chain 03: 3 cases OK, 1 Exception """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) filter.put_into_fgr("-t filter -A INPUT -i sl0 -j ACCEPT") self.assertEqual(['-A INPUT -i sl0 -j ACCEPT '], filter.data["INPUT"]) filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) filter.put_into_fgr("-t nat -A OUTPUT -j ACCEPT") self.assertEqual(['-A OUTPUT -j ACCEPT '], filter.data["OUTPUT"]) filter.put_into_fgr("-t nat -A FORWARD -j ACCEPT") self.assertEqual(['-A FORWARD -j ACCEPT '], filter.data["FORWARD"]) with self.assertRaises(ConverterError): filter.put_into_fgr("-t na -A INPUT")
def test_01_create_a_chain_object(self): """ Chain 01: create a Filter group, f.e. filter """ self.assertIsInstance(Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]), Chains) self.assertEqual({}, Chains("filter", [])) filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) self.assertEqual("filter", filter.name) self.assertEqual(['INPUT', 'FORWARD', 'OUTPUT'], filter.tables) self.assertEqual("-", filter.policy) self.assertEqual(0, filter.length) self.assertEqual( { 'FORWARD': 'ACCEPT', 'INPUT': 'ACCEPT', 'OUTPUT': 'ACCEPT' }, filter.poli)
def test_04_flush(self): """ Chain 04: flush filter group, 2 rules and an invalid chain """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) filter.put_into_fgr("-t filter -A INPUT -i sl0 -j ACCEPT") self.assertEqual(['-A INPUT -i sl0 -j ACCEPT '], filter.data["INPUT"]) filter.put_into_fgr("-A OUTPUT -o sl1 -j ACCEPT") self.assertEqual(['-A OUTPUT -o sl1 -j ACCEPT'], filter.data["OUTPUT"]) filter.put_into_fgr("-F") self.assertEqual([], filter.data["INPUT"]) self.assertEqual([], filter.data["OUTPUT"]) with self.assertRaises(ConverterError): filter.put_into_fgr("-t inval -F")
def test_02_prove_policies(self): """ Chain 02: check 3 valid policies, 1 exception """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) filter.put_into_fgr("-P INPUT DROP") self.assertEqual( {'FORWARD': 'ACCEPT', 'INPUT': 'DROP', 'OUTPUT': 'ACCEPT'}, filter.poli) filter.put_into_fgr("-P FORWARD REJECT") self.assertEqual( {'FORWARD': 'REJECT', 'INPUT': 'DROP', 'OUTPUT': 'ACCEPT'}, filter.poli) filter.put_into_fgr("-P OUTPUT DROP") self.assertEqual( {'FORWARD': 'REJECT', 'INPUT': 'DROP', 'OUTPUT': 'DROP'}, filter.poli) with self.assertRaises(ConverterError): filter.put_into_fgr("-P OUTPUT FAIL")
def test_09_insert_rule_works(self): """ Chain 09: insert a rule into a nonempty chain works at start """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) filter.put_into_fgr("-A INPUT -p tcp -j ACCEPT") filter.put_into_fgr("-I INPUT -p udp -j ACCEPT") filter.put_into_fgr("-I INPUT -p esp -j ACCEPT") expect = [ '-I INPUT -p esp -j ACCEPT', '-I INPUT -p udp -j ACCEPT', '-A INPUT -p tcp -j ACCEPT' ] self.assertEqual(expect, filter.data["INPUT"])
def test_10_append_rule(self): """ Chain 10: append a rule to a chain """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) filter.put_into_fgr("-A INPUT -p tcp -j ACCEPT") self.assertEqual(['-A INPUT -p tcp -j ACCEPT'], filter.data["INPUT"]) filter.put_into_fgr("-A INPUT -p udp -j ACCEPT") filter.put_into_fgr("-A INPUT -p esp -j ACCEPT") expect = [ '-A INPUT -p tcp -j ACCEPT', '-A INPUT -p udp -j ACCEPT', '-A INPUT -p esp -j ACCEPT' ] self.assertEqual(expect, filter.data["INPUT"])
def test_12_remove_chain(self): """ Chain 12: try to remove an existing chain """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) filter.put_into_fgr("-N NEWCHAIN") self.assertEqual( { 'FORWARD': [], 'INPUT': [], 'NEWCHAIN': [], 'OUTPUT': [] }, filter.data) filter.put_into_fgr("-X NEWCHAIN") self.assertEqual({ 'FORWARD': [], 'INPUT': [], 'OUTPUT': [] }, filter.data)
def test_03_tables_names(self): """ Chain 03: 3 cases OK, 1 Exception """ filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) filter.put_into_fgr("-t filter -A INPUT -i sl0 -j ACCEPT") self.assertEqual(['-A INPUT -i sl0 -j ACCEPT '], filter.data["INPUT"]) filter = Chains("filter", ["INPUT", "FORWARD", "OUTPUT"]) filter.put_into_fgr("-t nat -A OUTPUT -j ACCEPT") self.assertEqual(['-A OUTPUT -j ACCEPT '], filter.data["OUTPUT"]) filter.put_into_fgr("-t nat -A FORWARD -j ACCEPT") self.assertEqual(['-A FORWARD -j ACCEPT '], filter.data["FORWARD"]) with self.assertRaises(ConverterError): filter.put_into_fgr("-t na -A INPUT")