github_mirrors/openappsec
3
0
Fork 1
mirror of https://github.com/openappsec/openappsec.git synced 2025-09-29 19:24:26 +03:00
Code Issues Packages Projects Releases Wiki Activity

First release of open-appsec source code

Browse Source
This commit is contained in:
roybarda
2022-10-26 19:33:19 +03:00
parent 3883109caf
commit a883352f79
1353 changed files with 276290 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
components/gradual_deployment/gradual_deployment_ut/CMakeLists.txt Normal file
View File

@@ -0,0 +1,8 @@
include_directories(${Boost_INCLUDE_DIRS})
include_directories(${CMAKE_SOURCE_DIR}/components/include)
add_unit_test(
gradual_deployment_ut
"gradual_deployment_ut.cc"
"singleton;rest;connkey;${RT_LIBRARY};gradual_deployment;"
)

127
components/gradual_deployment/gradual_deployment_ut/gradual_deployment_ut.cc Normal file
View File

@@ -0,0 +1,127 @@
#include "http_manager.h"
#include <string>
#include <fstream>
#include <streambuf>
#include "cptest.h"
#include "config.h"
#include "singleton.h"
#include "environment.h"
#include "rest_server.h"
#include "table.h"
#include "time_proxy.h"
#include "mainloop.h"
#include "mock/mock_rest_api.h"
#include "i_http_manager.h"
#include "gradual_deployment.h"
using namespace std;
using namespace testing;
class GradualDeploymentTest : public Test
{
public:
GradualDeploymentTest()
{
EXPECT_CALL(rest, mockRestCall(RestAction::SET, "gradual-deployment-policy", _)).WillOnce(
WithArg<2>(Invoke(this, &GradualDeploymentTest::setGDPolicy))
);
gradual_deployment.init();
i_gradual_deployment = Singleton::Consume<I_GradualDeployment>::from(gradual_deployment);
}
bool
setGDPolicy(const unique_ptr<RestInit> &p)
{
gradual_rest_listener = p->getRest();
return true;
}
unique_ptr<ServerRest> gradual_rest_listener;
I_GradualDeployment *i_gradual_deployment;
private:
StrictMock<MockRestApi> rest;
GradualDeployment gradual_deployment;
};
TEST_F(GradualDeploymentTest, getPolicyTest)
{
stringstream is;
is << "{"
<< "\"attachment_type\":\"HTTP-Manager\","
<< "\"ip_ranges\":[\"8.8.8.8\",\"9.9.9.9-10.10.10.10\","
<< "\"0:0:0:0:0:0:0:1-0:0:0:0:0:0:0:4\""
<< "]}";
Maybe<string> rest_call_result = gradual_rest_listener->performRestCall(is);
EXPECT_TRUE(rest_call_result.ok());
vector<string> expected = {"8.8.8.8-8.8.8.8", "9.9.9.9-10.10.10.10", "::1-::4"};
vector<string> curr_policy = i_gradual_deployment->getPolicy(I_GradualDeployment::AttachmentType::NGINX);
EXPECT_EQ(curr_policy, expected);
}
TEST_F(GradualDeploymentTest, MissingAttachmentType)
{
stringstream is("{\"ip_ranges\":[\"8.8\"]}");
Maybe<string> rest_call_result = gradual_rest_listener->performRestCall(is);
EXPECT_FALSE(rest_call_result.ok());
EXPECT_THAT(
rest_call_result.getErr(),
HasSubstr("Couldn't get variable attachment_type")
);
vector<string> expected = {};
vector<string> curr_policy = i_gradual_deployment->getPolicy(I_GradualDeployment::AttachmentType::NGINX);
EXPECT_EQ(curr_policy, expected);
}
TEST_F(GradualDeploymentTest, InvalidAttachmentType)
{
stringstream is;
is << "{"
<< "\"attachment_type\":\"unsupported-attachment-type\","
<< "\"ip_ranges\":[\"8.8.8.8\",\"9.9.9.9-10.10.10.10\","
<< "\"0:0:0:0:0:0:0:1-0:0:0:0:0:0:0:4\""
<< "]}";
Maybe<string> rest_call_result = gradual_rest_listener->performRestCall(is);
EXPECT_FALSE(rest_call_result.ok());
EXPECT_THAT(
rest_call_result.getErr(),
HasSubstr(
"Failed to determine attachment type. "
"Type: unsupported-attachment-type, error: unknown attachment type"
)
);
vector<string> expected = {};
vector<string> curr_policy = i_gradual_deployment->getPolicy(I_GradualDeployment::AttachmentType::NGINX);
EXPECT_EQ(curr_policy, expected);
}
TEST_F(GradualDeploymentTest, InvalidIPRanges)
{
stringstream is;
is << "{"
<< "\"attachment_type\":\"HTTP-Manager\","
<< "\"ip_ranges\":[\"8.8\"]"
<< "}";
Maybe<string> rest_call_result = gradual_rest_listener->performRestCall(is);
EXPECT_FALSE(rest_call_result.ok());
EXPECT_THAT(
rest_call_result.getErr(),
HasSubstr(
"Failed to set gradual deployment policy. "
"Error: Failed to parse gradual deployment IP range: "
"Could not create IP address, String '8.8' is not a valid IPv4/IPv6 address"
)
);
vector<string> expected = {};
vector<string> curr_policy = i_gradual_deployment->getPolicy(I_GradualDeployment::AttachmentType::NGINX);
EXPECT_EQ(curr_policy, expected);
}