rublon-ssh/PAM/ssh/tests/init_test.cpp
Bartosz Wieczorek 9d6790c3cf snapshot
2023-07-19 12:09:02 +02:00

116 lines
3.5 KiB
C++

#include <gmock/gmock.h>
#include <gtest/gtest.h>
#include "../src/rublon.hpp"
#include "core_response_generator.hpp"
using namespace rublon;
using namespace testing;
namespace {
Configuration conf;
}
class CoreHandlerMock : public CoreHandlerInterface< CoreHandlerMock > {
public:
CoreHandlerMock() {}
MOCK_METHOD(( tl::expected< Document, CoreHandlerError > ), request, ( std::string_view, const Document & ), (const));
CoreHandlerMock & statusPending() {
gen.status = "pending";
return *this;
}
CoreHandlerMock & brokenBody() {
gen.generateBrokenData = true;
return *this;
}
operator tl::expected< Document, CoreHandlerError >() {
auto body = gen.generateBody();
rublon::Document doc;
doc.Parse(body.c_str());
return doc;
}
CoreResponseGenerator gen;
};
class PamInfoMock {
public:
PamInfoMock(pam_handle_t *) {}
MOCK_METHOD(rublon::NonOwningPtr< const char >, ip, (), (const));
MOCK_METHOD(rublon::NonOwningPtr< const char >, username, (), (const));
};
template < typename Pam >
class MethodFactoryMock {
public:
MethodFactoryMock(const Pam &) {}
template < typename Array_t >
tl::expected< Method, PamAction > create(const Array_t & methods) const {}
};
class InitTestable : public Init< MethodFactoryMock, PamInfoMock > {
public:
InitTestable(const rublon::Configuration & conf) : Init{nullptr, conf} {}
PamInfoMock & pam() {
return _pamInfo;
}
};
class RublonHttpInitTest : public testing::Test {
public:
void expectDefaultPamInfo() {
EXPECT_CALL(pam, ip()).WillOnce(Return("192.168.0.1"));
EXPECT_CALL(pam, username()).WillOnce(Return("bwi"));
}
RublonHttpInitTest() : coreHandler{}, sut{conf}, pam{sut.pam()} {}
CoreHandlerMock coreHandler;
InitTestable sut{conf};
PamInfoMock & pam;
};
using CoreReturn = tl::expected< Document, CoreHandlerError >;
TEST_F(RublonHttpInitTest, initializationSendsRequestOnGoodPath) {
expectDefaultPamInfo();
EXPECT_CALL(coreHandler, request("/api/transaction/init", _))
.WillOnce(Return(tl::unexpected{CoreHandlerError{CoreHandlerError::BadSigature}}));
sut.handle(coreHandler);
}
MATCHER_P(HoldsPamAction, action, "") {
return not arg.has_value() && arg.error() == action;
}
TEST_F(RublonHttpInitTest, rublon_Accept_pamLoginWhenThereIsNoConnection) {
expectDefaultPamInfo();
EXPECT_CALL(coreHandler, request(_, _)).WillOnce(Return(tl::unexpected{CoreHandlerError{CoreHandlerError::ConnectionError}}));
EXPECT_THAT(sut.handle(coreHandler), HoldsPamAction(PamAction::decline));
}
TEST_F(RublonHttpInitTest, rublon_Decline_pamLoginWhenServerHasBadSignature) {
expectDefaultPamInfo();
EXPECT_CALL(coreHandler, request(_, _)).WillOnce(Return(tl::unexpected{CoreHandlerError{CoreHandlerError::BadSigature}}));
EXPECT_THAT(sut.handle(coreHandler), HoldsPamAction(PamAction::decline));
}
TEST_F(RublonHttpInitTest, rublon_Decline_pamLoginWhenServerReturnsBrokenData) {
expectDefaultPamInfo();
EXPECT_CALL(coreHandler, request(_, _)).WillOnce(Return(tl::unexpected{CoreHandlerError{CoreHandlerError::BrokenData}}));
EXPECT_THAT(sut.handle(coreHandler), HoldsPamAction(PamAction::decline));
}
TEST_F(RublonHttpInitTest, rublon_Decline_pamLoginWhenServerReturnsCoreException) {
expectDefaultPamInfo();
EXPECT_CALL(coreHandler, request(_, _)).WillOnce(Return(tl::unexpected{CoreHandlerError{CoreHandlerError::CoreException}}));
EXPECT_THAT(sut.handle(coreHandler), HoldsPamAction(PamAction::decline));
}