-
Notifications
You must be signed in to change notification settings - Fork 0
/
TestSuite.hpp
86 lines (73 loc) · 2.15 KB
/
TestSuite.hpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
#ifndef TESTSUITE_HPP
#define TESTSUITE_HPP
#include <functional>
#include <iostream>
#include <utility>
#include <string>
#include <vector>
#include "ColorFormat.hpp"
class TestSuite {
private:
std::vector<std::pair<std::string, std::function<bool()>>> tests;
std::function<bool()> setup, before;
std::string suiteName;
ColorFormat cf;
public:
TestSuite( std::string suiteName) : suiteName(suiteName) {
setup = []() { return true; };
before = []() { return true; };
}
void beforeAll(std::function<bool()> setupFunction) {
this->setup = setupFunction;
}
void beforeEach(std::function<bool()> beforeFunction) {
this->before = beforeFunction;
}
void addTest(std::string testName, std::function<bool()> test) {
tests.push_back(std::make_pair(testName, test));
}
void runTests() {
cf.setFormat(Style::BOLD);
cf.setFormat(Style::UNDERLINE);
cf.addText("Test Suite: " + suiteName);
std::cout << cf << "\n";
if (!setup()) {
cf.setFormat(Style::BOLD, Color::RED);
cf.addText("Before All Failed");
cf.setFormat(Style::NORMAL);
cf.addText(" - ");
cf.setFormat(Style::ITALIC);
cf.setFormat(Style::UNDERLINE, Color::YELLOW);
cf.addText("ALL TESTS SKIPPED");
std::cout << cf << "\n\n\n";
return;
}
size_t numPassed = 0;
for (auto test : tests) {
if (!before()) {
cf.setFormat(Style::BOLD, Color::RED);
cf.addText("Before Each Failed");
std::cout << cf << " - Test: " << test.first << " ... ";
cf.setFormat(Style::ITALIC);
cf.setFormat(Style::UNDERLINE, Color::YELLOW);
cf.addText("SKIPPED");
std::cout << cf << "\n";
continue;
}
if (test.second()) {
cf.setFormat(Style::ITALIC);
cf.setFormat(Style::UNDERLINE, Color::GREEN);
cf.addText("PASSED");
std::cout << "Test: " << test.first << " ... " << cf << "\n";
numPassed++;
} else {
cf.setFormat(Style::ITALIC);
cf.setFormat(Style::UNDERLINE, Color::RED);
cf.addText("FAILED");
std::cout << "Test: " << test.first << " ... " << cf << "\n";
}
}
std::cout << numPassed << "/" << tests.size() << " passed\n\n\n";
}
};
#endif // TESTSUITE_HPP