libcryfs/test/cryfs/testutils/MockConsole.h

34 lines
1.1 KiB
C
Raw Normal View History

2015-10-22 18:48:04 +02:00
#pragma once
#ifndef MESSMER_CRYFS_TEST_TESTUTILS_MOCKCONSOLE_H
#define MESSMER_CRYFS_TEST_TESTUTILS_MOCKCONSOLE_H
2016-02-11 16:39:42 +01:00
#include <cpp-utils/io/Console.h>
#include <gmock/gmock.h>
2015-10-22 18:48:04 +02:00
class MockConsole: public cpputils::Console {
public:
MOCK_METHOD1(print, void(const std::string&));
MOCK_METHOD2(ask, unsigned int(const std::string&, const std::vector<std::string>&));
2016-09-25 02:50:28 +02:00
MOCK_METHOD2(askYesNo, bool(const std::string&, bool));
2018-09-04 01:51:59 +02:00
MOCK_METHOD1(askPassword, std::string(const std::string&));
2015-10-22 18:48:04 +02:00
};
ACTION_P(ChooseCipher, cipherName) {
return std::find(arg1.begin(), arg1.end(), cipherName) - arg1.begin();
}
#define ChooseAnyCipher() ChooseCipher("aes-256-gcm")
class TestWithMockConsole {
public:
// Return a console that chooses a valid cryfs setting
static std::shared_ptr<MockConsole> mockConsole() {
auto console = std::make_shared<::testing::NiceMock<MockConsole>>();
EXPECT_CALL(*console, ask(::testing::_, ::testing::_)).WillRepeatedly(ChooseCipher("aes-256-gcm"));
2016-09-25 02:50:28 +02:00
EXPECT_CALL(*console, askYesNo(::testing::_, ::testing::_)).WillRepeatedly(::testing::Return(true));
return console;
}
};
2015-10-22 18:48:04 +02:00
#endif