mirror of
https://github.com/potats0/lua-resty-coraza.git
synced 2025-11-16 01:12:22 +03:00
refactor: update code
This commit is contained in:
potatso
284
t/sanity.t
Normal file
284
t/sanity.t
Normal file
@@ -0,0 +1,284 @@
|
||||
use Test::Nginx::Socket 'no_plan';
|
||||
|
||||
our $HttpConfig = <<'_EOC_';
|
||||
lua_package_path "lib/?.lua;/usr/local/share/lua/5.1/?.lua;;";
|
||||
lua_socket_log_errors off;
|
||||
lua_need_request_body on;
|
||||
_EOC_
|
||||
|
||||
our $LocationConfig = <<'_EOC_';
|
||||
location /t {
|
||||
content_by_lua_block {
|
||||
local coraza = require "resty.coraza"
|
||||
local waf = coraza.create_waf()
|
||||
if waf then
|
||||
ngx.say("done")
|
||||
end
|
||||
|
||||
}
|
||||
}
|
||||
_EOC_
|
||||
|
||||
add_block_preprocessor(sub {
|
||||
my ($block) = @_;
|
||||
|
||||
if (!defined $block->request) {
|
||||
$block->set_value("request", "GET /t");
|
||||
}
|
||||
});
|
||||
|
||||
no_shuffle();
|
||||
no_long_string();
|
||||
no_root_location();
|
||||
run_tests();
|
||||
|
||||
__DATA__
|
||||
|
||||
=== TEST 1: sanity
|
||||
--- http_config eval: $::HttpConfig
|
||||
--- config
|
||||
location /t {
|
||||
content_by_lua_block {
|
||||
local coraza = require "resty.coraza.coraza"
|
||||
local waf = coraza.new_waf()
|
||||
if waf then
|
||||
ngx.say("done")
|
||||
end
|
||||
}
|
||||
}
|
||||
--- error_code: 200
|
||||
--- response_body_like eval
|
||||
"done"
|
||||
--- error_log eval
|
||||
["Success to creat new waf"]
|
||||
|
||||
=== TEST 2: test to release waf pointer
|
||||
--- http_config eval: $::HttpConfig
|
||||
--- config
|
||||
location /t {
|
||||
content_by_lua_block {
|
||||
local coraza = require "resty.coraza.coraza"
|
||||
local waf = coraza.new_waf()
|
||||
coraza.free_waf(waf)
|
||||
}
|
||||
}
|
||||
--- error_code: 200
|
||||
--- error_log eval
|
||||
["Success to creat new waf", "Success to free new waf"]
|
||||
|
||||
=== TEST 3: test rules_add_file
|
||||
--- http_config eval: $::HttpConfig
|
||||
--- config
|
||||
location /t {
|
||||
content_by_lua_block {
|
||||
local coraza = require "resty.coraza.coraza"
|
||||
local waf = coraza.new_waf()
|
||||
local ok, msg = coraza.rules_add_file(waf, "non-exist")
|
||||
ngx.say(msg)
|
||||
}
|
||||
}
|
||||
--- error_code: 200
|
||||
--- response_body_like eval
|
||||
"failed to readfile: open non-exist: no such file or directory"
|
||||
|
||||
=== TEST 4: test rules_add with sec ssss
|
||||
--- http_config eval: $::HttpConfig
|
||||
--- config
|
||||
location /t {
|
||||
content_by_lua_block {
|
||||
local coraza = require "resty.coraza.coraza"
|
||||
local waf = coraza.new_waf()
|
||||
local ok, msg = coraza.rules_add(waf, "sec ssss")
|
||||
ngx.say(msg)
|
||||
}
|
||||
}
|
||||
--- error_code: 200
|
||||
--- response_body_like eval
|
||||
"unknown directive \"sec\""
|
||||
|
||||
=== TEST 5: test rules_add with SecRule REQUEST_HEADERS:User-Agent "Mozilla" "phase:1, id:3,drop,status:452,log,msg:'Blocked User-Agent'"
|
||||
--- http_config eval: $::HttpConfig
|
||||
--- config
|
||||
location /t {
|
||||
content_by_lua_block {
|
||||
local coraza = require "resty.coraza.coraza"
|
||||
local waf = coraza.new_waf()
|
||||
local ok, msg = coraza.rules_add(waf, [[SecRule REQUEST_HEADERS:User-Agent "Mozilla" "phase:1, id:3,drop,status:452,log,msg:'Blocked User-Agent'"]])
|
||||
ngx.say(msg)
|
||||
}
|
||||
}
|
||||
--- error_code: 200
|
||||
--- error_log eval
|
||||
["Success to load rule with"]
|
||||
|
||||
=== TEST 6: test new_transaction
|
||||
--- http_config eval: $::HttpConfig
|
||||
--- config
|
||||
location /t {
|
||||
content_by_lua_block {
|
||||
local coraza = require "resty.coraza.coraza"
|
||||
local waf = coraza.new_waf()
|
||||
local ok, msg = coraza.new_transaction(waf)
|
||||
}
|
||||
}
|
||||
--- error_code: 200
|
||||
--- error_log eval
|
||||
["Success to creat new transaction id"]
|
||||
|
||||
=== TEST 7: test new_transaction with nil waf pointer
|
||||
--- http_config eval: $::HttpConfig
|
||||
--- config
|
||||
location /t {
|
||||
content_by_lua_block {
|
||||
local coraza = require "resty.coraza.coraza"
|
||||
local waf = coraza.new_waf()
|
||||
local ok, msg = coraza.new_transaction()
|
||||
}
|
||||
}
|
||||
--- error_code: 200
|
||||
--- error_log eval
|
||||
["Failed to creat new transaction id"]
|
||||
|
||||
=== TEST 8: test process_connection
|
||||
--- http_config eval: $::HttpConfig
|
||||
--- config
|
||||
location /t {
|
||||
content_by_lua_block {
|
||||
local coraza = require "resty.coraza.coraza"
|
||||
local waf = coraza.new_waf()
|
||||
local tran = coraza.new_transaction(waf)
|
||||
coraza.process_connection(tran, ngx.var.remote_addr, ngx.var.remote_port,
|
||||
ngx.var.server_addr, ngx.var.server_port)
|
||||
}
|
||||
}
|
||||
--- error_code: 200
|
||||
--- error_log eval
|
||||
["success to invoke coraza_process_connection with"]
|
||||
|
||||
=== TEST 9: test process_connection with error
|
||||
--- http_config eval: $::HttpConfig
|
||||
--- config
|
||||
location /t {
|
||||
content_by_lua_block {
|
||||
local coraza = require "resty.coraza.coraza"
|
||||
local waf = coraza.new_waf()
|
||||
local tran = coraza.new_transaction(waf)
|
||||
coraza.process_connection(nil, ngx.var.remote_addr, ngx.var.remote_port,
|
||||
ngx.var.server_addr, ngx.var.server_port)
|
||||
}
|
||||
}
|
||||
--- error_code: 200
|
||||
--- error_log eval
|
||||
["failed to invoke coraza_process_connection with"]
|
||||
|
||||
=== TEST 10: test process_uri
|
||||
--- http_config eval: $::HttpConfig
|
||||
--- config
|
||||
location /t {
|
||||
content_by_lua_block {
|
||||
local coraza = require "resty.coraza.coraza"
|
||||
local waf = coraza.new_waf()
|
||||
local tran = coraza.new_transaction(waf)
|
||||
coraza.process_connection(tran, ngx.var.remote_addr, ngx.var.remote_port,
|
||||
ngx.var.server_addr, ngx.var.server_port)
|
||||
coraza.process_uri(tran, "GET", "/index.php?a=1", "http/1.1")
|
||||
coraza.process_uri(tran, "aa", "/index.php?a=1", "http/1.1")
|
||||
coraza.process_uri(nil, "aa", "/index.php?a=1", "http/1.1")
|
||||
}
|
||||
}
|
||||
--- error_code: 200
|
||||
--- error_log eval
|
||||
["success to invoke coraza_process_uri with", "failed to invoke coraza_process_uri with"]
|
||||
|
||||
=== TEST 11: test add_request_header
|
||||
--- http_config eval: $::HttpConfig
|
||||
--- config
|
||||
location /t {
|
||||
content_by_lua_block {
|
||||
local coraza = require "resty.coraza.coraza"
|
||||
local waf = coraza.new_waf()
|
||||
local tran = coraza.new_transaction(waf)
|
||||
coraza.process_connection(tran, ngx.var.remote_addr, ngx.var.remote_port,
|
||||
ngx.var.server_addr, ngx.var.server_port)
|
||||
coraza.process_uri(tran, "GET", "/index.php?a=1", "http/1.1")
|
||||
coraza.add_request_header(tran, "key", "value")
|
||||
}
|
||||
}
|
||||
--- error_code: 200
|
||||
--- error_log eval
|
||||
["success to invoke coraza_add_request_header with key:value"]
|
||||
|
||||
=== TEST 12: test add_request_header with error
|
||||
--- http_config eval: $::HttpConfig
|
||||
--- config
|
||||
location /t {
|
||||
content_by_lua_block {
|
||||
local coraza = require "resty.coraza.coraza"
|
||||
local waf = coraza.new_waf()
|
||||
local tran = coraza.new_transaction(waf)
|
||||
coraza.process_connection(tran, ngx.var.remote_addr, ngx.var.remote_port,
|
||||
ngx.var.server_addr, ngx.var.server_port)
|
||||
coraza.process_uri(tran, "GET", "/index.php?a=1", "http/1.1")
|
||||
coraza.add_request_header(nil, "key", "value")
|
||||
}
|
||||
}
|
||||
--- error_code: 200
|
||||
--- error_log eval
|
||||
["failed to invoke coraza_add_request_header with key:value"]
|
||||
|
||||
=== TEST 12: test add_get_args
|
||||
--- http_config eval: $::HttpConfig
|
||||
--- config
|
||||
location /t {
|
||||
content_by_lua_block {
|
||||
local coraza = require "resty.coraza.coraza"
|
||||
local waf = coraza.new_waf()
|
||||
local tran = coraza.new_transaction(waf)
|
||||
coraza.process_connection(tran, ngx.var.remote_addr, ngx.var.remote_port,
|
||||
ngx.var.server_addr, ngx.var.server_port)
|
||||
coraza.process_uri(tran, "GET", "/index.php?a=1", "http/1.1")
|
||||
coraza.add_request_header(tran, "key", "value")
|
||||
coraza.add_get_args(tran, "key", "value")
|
||||
}
|
||||
}
|
||||
--- error_code: 200
|
||||
--- error_log eval
|
||||
["success to invoke coraza_add_get_args with key:value"]
|
||||
|
||||
=== TEST 13: test add_get_args with error
|
||||
--- http_config eval: $::HttpConfig
|
||||
--- config
|
||||
location /t {
|
||||
content_by_lua_block {
|
||||
local coraza = require "resty.coraza.coraza"
|
||||
local waf = coraza.new_waf()
|
||||
local tran = coraza.new_transaction(waf)
|
||||
coraza.process_connection(tran, ngx.var.remote_addr, ngx.var.remote_port,
|
||||
ngx.var.server_addr, ngx.var.server_port)
|
||||
coraza.process_uri(tran, "GET", "/index.php?a=1", "http/1.1")
|
||||
coraza.add_request_header(tran, "key", "value")
|
||||
coraza.add_get_args(nil, "key", "value")
|
||||
}
|
||||
}
|
||||
--- error_code: 200
|
||||
--- error_log eval
|
||||
["failed to invoke coraza_add_get_args with key:value"]
|
||||
|
||||
=== TEST 13: test process_request_headers
|
||||
--- http_config eval: $::HttpConfig
|
||||
--- config
|
||||
location /t {
|
||||
content_by_lua_block {
|
||||
local coraza = require "resty.coraza.coraza"
|
||||
local waf = coraza.new_waf()
|
||||
local tran = coraza.new_transaction(waf)
|
||||
coraza.process_connection(tran, ngx.var.remote_addr, ngx.var.remote_port,
|
||||
ngx.var.server_addr, ngx.var.server_port)
|
||||
coraza.process_uri(tran, "GET", "/index.php?a=1", "http/1.1")
|
||||
coraza.add_request_header(tran, "key", "value")
|
||||
coraza.process_request_headers(tran)
|
||||
}
|
||||
}
|
||||
--- error_code: 200
|
||||
--- error_log eval
|
||||
["success to invoke coraza_process_request_headers"]
|
||||
Reference in New Issue
Block a user