libcryfs/test/cpp-utils/io/ProgressBarTest.cpp

67 lines
1.5 KiB
C++
Raw Normal View History

2019-01-23 07:02:34 +01:00
#include <cpp-utils/io/ProgressBar.h>
#include <gmock/gmock.h>
using cpputils::ProgressBar;
using std::make_shared;
class MockConsole final: public cpputils::Console {
public:
void EXPECT_OUTPUT(const char* expected) {
EXPECT_EQ(expected, _output);
_output = "";
}
void print(const std::string& text) override {
_output += text;
}
unsigned int ask(const std::string&, const std::vector<std::string>&) override {
EXPECT_TRUE(false);
return 0;
}
bool askYesNo(const std::string&, bool) override {
EXPECT_TRUE(false);
return false;
}
std::string askPassword(const std::string&) override {
EXPECT_TRUE(false);
return "";
}
private:
std::string _output;
};
TEST(ProgressBarTest, testProgressBar) {
auto console = make_shared<MockConsole>();
ProgressBar bar(console, "Preamble", 2000);
console->EXPECT_OUTPUT("\n\rPreamble 0%");
// when updating to 0, doesn't reprint
bar.update(0);
console->EXPECT_OUTPUT("");
// update to half
bar.update(1000);
console->EXPECT_OUTPUT("\rPreamble 50%");
// when updating to same value, doesn't reprint
bar.update(1000);
console->EXPECT_OUTPUT("");
// when updating to value with same percentage, doesn't reprint
bar.update(1001);
console->EXPECT_OUTPUT("");
// update to 0
bar.update(0);
console->EXPECT_OUTPUT("\rPreamble 0%");
// update to full
bar.update(2000);
console->EXPECT_OUTPUT("\rPreamble 100%");
}