Skip to content

Commit

Permalink
test: apply reviewer suggestions
Browse files Browse the repository at this point in the history
Signed-off-by: Alexandre Rulleau <[email protected]>
  • Loading branch information
Leiyks committed Nov 19, 2024
1 parent 7e3a0f8 commit 1a0b972
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 16 deletions.
21 changes: 10 additions & 11 deletions appsec/tests/helper/client_test.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1764,7 +1764,7 @@ TEST(ClientTest, RequestExecWithAttack)
}
}

TEST(ClientTest, RequestShutdownWithAttackAndFingerprint)
TEST(ClientTest, RequestShutdownWithWithAttackAndFingerprint)
{
auto smanager = std::make_shared<service_manager>();
auto broker = new mock::broker();
Expand All @@ -1776,14 +1776,7 @@ TEST(ClientTest, RequestShutdownWithAttackAndFingerprint)
// Request Init
{
network::request_init::request msg;

auto query = parameter::map();
query.add("query", parameter::string("asdfds"sv));

msg.data = parameter::map();
msg.data.add("server.request.uri.raw", parameter::string("asdfds"sv));
msg.data.add("server.request.method", parameter::string("GET"sv));
msg.data.add("server.request.query", std::move(query));

network::request req(std::move(msg));

Expand All @@ -1804,8 +1797,15 @@ TEST(ClientTest, RequestShutdownWithAttackAndFingerprint)
// Request Shutdown
{
network::request_shutdown::request msg;

auto query = parameter::map();
query.add("query", parameter::string("asdfds"sv));

msg.data = parameter::map();
msg.data.add("http.client_ip", parameter::string("192.168.1.1"sv));
msg.data.add("server.request.uri.raw", parameter::string("asdfds"sv));
msg.data.add("server.request.method", parameter::string("GET"sv));
msg.data.add("server.request.query", std::move(query));

network::request req(std::move(msg));

Expand All @@ -1820,10 +1820,9 @@ TEST(ClientTest, RequestShutdownWithAttackAndFingerprint)
auto msg_res =
dynamic_cast<network::request_shutdown::response *>(res.get());
EXPECT_STREQ(msg_res->actions[0].verdict.c_str(), "block");
EXPECT_FALSE(std::regex_match(
EXPECT_TRUE(std::regex_match(
msg_res->meta["_dd.appsec.fp.http.endpoint"].c_str(),
std::regex(
"http-get-[A-Za-z0-9]{8}-[A-Za-z0-9]{8}-([A-Za-z0-9]{8})?")));
std::regex("\"http-get(-[A-Za-z0-9]*){2,3}\"")));
}
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -222,7 +222,7 @@ trait CommonTests {
assert span.metrics."_dd.appsec.enabled" == 1.0d
assert span.metrics."_dd.appsec.waf.duration" > 0.0d
assert span.meta."_dd.appsec.event_rules.version" != ''
assert span.meta."_dd.appsec.fp.http.endpoint" != ''
assert span.meta."_dd.appsec.fp.http.endpoint" ==~ /^"http-get(-[a-zA-Z0-9]*){3,4}"$/
}

@Test
Expand All @@ -237,7 +237,7 @@ trait CommonTests {
assert span.metrics."_dd.appsec.enabled" == 1.0d
assert span.metrics."_dd.appsec.waf.duration" > 0.0d
assert span.meta."_dd.appsec.event_rules.version" != ''
assert span.meta."_dd.appsec.fp.http.endpoint" != ''
assert span.meta."_dd.appsec.fp.http.endpoint" ==~ /^"http-get(-[a-zA-Z0-9]*){3,4}"$/
}

@Test
Expand All @@ -251,7 +251,7 @@ trait CommonTests {
assert span.metrics."_dd.appsec.enabled" == 1.0d
assert span.metrics."_dd.appsec.waf.duration" > 0.0d
assert span.meta."_dd.appsec.event_rules.version" != ''
assert span.meta."_dd.appsec.fp.http.endpoint" != ''
assert span.meta."_dd.appsec.fp.http.endpoint" ==~ /^"http-get(-[a-zA-Z0-9]*){3,4}"$/
}

@Test
Expand All @@ -265,7 +265,7 @@ trait CommonTests {
assert span.metrics."_dd.appsec.enabled" == 1.0d
assert span.metrics."_dd.appsec.waf.duration" > 0.0d
assert span.meta."_dd.appsec.event_rules.version" != ''
assert span.meta."_dd.appsec.fp.http.endpoint" != ''
assert span.meta."_dd.appsec.fp.http.endpoint" ==~ /^"http-get(-[a-zA-Z0-9]*){3,4}"$/
}

@Test
Expand All @@ -282,7 +282,7 @@ trait CommonTests {
assert span.metrics."_dd.appsec.waf.duration" > 0.0d
assert span.meta."_dd.appsec.event_rules.version" != ''
assert span.meta."appsec.blocked" == "true"
assert span.meta."_dd.appsec.fp.http.endpoint" != ''
assert span.meta."_dd.appsec.fp.http.endpoint" ==~ /^"http-get(-[a-zA-Z0-9]*){3,4}"$/
}

@Test
Expand Down

0 comments on commit 1a0b972

Please sign in to comment.