Преглед изворни кода

Fix bug v-add-firewall-rule

+ Add ipset tests
Jaap Marcus пре 4 година
родитељ
комит
d11c0971b5
2 измењених фајлова са 32 додато и 1 уклоњено
  1. 1 1
      bin/v-add-firewall-rule
  2. 31 0
      test/test.bats

+ 1 - 1
bin/v-add-firewall-rule

@@ -61,7 +61,7 @@ if [ -n "$comment" ]; then
 fi
 fi
 if [[ "$ip" =~ ^ipset: ]]; then
 if [[ "$ip" =~ ^ipset: ]]; then
     ipset_name="${ip#ipset:}"
     ipset_name="${ip#ipset:}"
-    v-list-firewall-ipset plain | grep "^$ipset_name\s" >/dev/null
+    $BIN/v-list-firewall-ipset plain | grep "^$ipset_name\s" >/dev/null
     check_result $? 'ipset object not found' "$E_NOTEXIST"
     check_result $? 'ipset object not found' "$E_NOTEXIST"
 else
 else
     is_format_valid 'ip'
     is_format_valid 'ip'

+ 31 - 0
test/test.bats

@@ -1586,6 +1586,37 @@ echo   "1.2.3.4" >> $HESTIA/data/firewall/excludes.conf
   check_ip_not_banned '1.2.3.4' 'HESTIA'
   check_ip_not_banned '1.2.3.4' 'HESTIA'
 }
 }
 
 
+@test "Test create ipset" {
+  run v-add-firewall-ipset "blacklist" "script:/usr/local/hestia/install/deb/firewall/ipset/blacklist.sh" v4 yes
+  assert_success
+  refute_output
+}
+
+@test "Create firewall with Ipset" {
+  run v-add-firewall-rule 'DROP' 'ipset:blacklist' '8083,22' 'TCP' 'Test'
+  assert_success
+  refute_output
+}
+
+@test "List firewall rules" {
+  run v-list-firewall csv
+  assert_success
+  assert_line --partial '11,DROP,TCP,8083,22,ipset:blacklist'
+  
+}
+
+@test "Delete firewall with Ipset" {
+  run v-delete-firewall-rule '11'
+  assert_success
+  refute_output
+}
+
+@test "Test delete ipset" {
+  run v-delete-firewall-ipset "blacklist" 
+  assert_success
+  refute_output
+}
+
 #----------------------------------------------------------#
 #----------------------------------------------------------#
 #                         PACKAGE                          #
 #                         PACKAGE                          #
 #----------------------------------------------------------#
 #----------------------------------------------------------#