Add EventCounter class

4 files changed, 98 insertions, 0 deletions

diff --git a/test/Jamfile.jam b/test/Jamfile.jam
index 1f0dac3..16b7244 100644
--- a/test/Jamfile.jam
+++ b/test/Jamfile.jam
@@ -17,6 +17,7 @@ run test-rawDataReader.cpp ;
 run test-bitset.cpp ;
 run test-streams.cpp ;
 run test-misc.cpp ;
+run test-counters.cpp : : : <library>testdb ;
 run test-mysql.cpp : : : <library>testdb ;
 run test-postgresql.cpp : -- : ../db/schema.sql : <library>testdb ;
 run test-e2e.cpp : -- : ../db/schema.sql : <library>testdb <implicit-dependency>testdb ;
diff --git a/test/helpers.cpp b/test/helpers.cpp
index f4fc030..6cb6fdc 100644
--- a/test/helpers.cpp
+++ b/test/helpers.cpp
@@ -13,3 +13,10 @@ MemStream::flush()
 {
 	fflush(s);
 }
+
+EventCounterTarget &
+EventCounterTarget::add(mariadb_rpl_event e, unsigned long n)
+{
+	counters[e] += n;
+	return *this;
+}
diff --git a/test/helpers.h b/test/helpers.h
index 87ef31d..9596e12 100644
--- a/test/helpers.h
+++ b/test/helpers.h
@@ -5,6 +5,7 @@
 #include <cstddef>
 #include <cstdio>
 #include <ctime>
+#include <eventCounter.h>
 #include <iostream>
 #include <memory>
 #include <optional>
@@ -68,4 +69,9 @@ struct MemStream {
 	FILE * s;
 };
 
+class EventCounterTarget : public MyGrate::EventCounter {
+public:
+	EventCounterTarget & add(mariadb_rpl_event, unsigned long);
+};
+
 #endif
diff --git a/test/test-counters.cpp b/test/test-counters.cpp
new file mode 100644
index 0000000..1bc30dc
--- /dev/null
+++ b/test/test-counters.cpp
@@ -0,0 +1,84 @@
+#define BOOST_TEST_MODULE Counters
+#include <boost/test/unit_test.hpp>
+
+#include "helpers.h"
+
+BOOST_FIXTURE_TEST_SUITE(ec, EventCounterTarget);
+
+BOOST_AUTO_TEST_CASE(zero)
+{
+	std::array<unsigned long, ENUM_END_EVENT> z {};
+	BOOST_CHECK_EQUAL_COLLECTIONS(counters.begin(), counters.end(), z.begin(), z.end());
+}
+
+BOOST_AUTO_TEST_CASE(tick_getAll)
+{
+	BOOST_CHECK_EQUAL(&counters, &getAll());
+}
+
+BOOST_AUTO_TEST_CASE(tick_one)
+{
+	tick(ROTATE_EVENT);
+	BOOST_CHECK_EQUAL(get(ROTATE_EVENT), 1);
+}
+
+BOOST_AUTO_TEST_CASE(tick_three)
+{
+	add(ROTATE_EVENT, 3);
+	BOOST_CHECK_EQUAL(get(ROTATE_EVENT), 3);
+}
+
+BOOST_AUTO_TEST_CASE(tick_different)
+{
+	add(ROTATE_EVENT, 2).add(UPDATE_ROWS_EVENT_V1, 1);
+	BOOST_CHECK_EQUAL(get(ROTATE_EVENT), 2);
+	BOOST_CHECK_EQUAL(get(UPDATE_ROWS_EVENT_V1), 1);
+}
+
+BOOST_AUTO_TEST_CASE(compare_zero)
+{
+	EventCounter other;
+	BOOST_CHECK(other >= *this);
+}
+
+BOOST_AUTO_TEST_CASE(compare_one)
+{
+	EventCounter other;
+	tick(ROTATE_EVENT);
+
+	BOOST_CHECK(!(other >= *this));
+
+	other.tick(ROTATE_EVENT);
+	BOOST_CHECK(other >= *this);
+
+	other.tick(ROTATE_EVENT);
+	BOOST_CHECK(other >= *this);
+}
+
+BOOST_AUTO_TEST_CASE(compare_many)
+{
+	EventCounter other;
+	add(ROTATE_EVENT, 2).add(UPDATE_ROWS_EVENT_V1, 1);
+
+	BOOST_CHECK(!(other >= *this));
+
+	other.tick(UPDATE_ROWS_EVENT_V1);
+	BOOST_CHECK(!(other >= *this));
+
+	other.tick(UPDATE_ROWS_EVENT_V1);
+	BOOST_CHECK(!(other >= *this));
+
+	other.tick(ROTATE_EVENT);
+	BOOST_CHECK(!(other >= *this));
+
+	other.tick(ROTATE_EVENT);
+	BOOST_CHECK(other >= *this);
+
+	other.tick(ROTATE_EVENT);
+	BOOST_CHECK(other >= *this);
+
+	other.tick(UPDATE_ROWS_EVENT_V1);
+	BOOST_CHECK(other >= *this);
+}
+
+BOOST_AUTO_TEST_SUITE_END();