Add some tests for EnumIndexedVector.

2 files changed, 76 insertions, 0 deletions

diff --git a/test/enum_indexed_vector_test.cc b/test/enum_indexed_vector_test.cc
new file mode 100644
index 000000000..d3f83805e
--- /dev/null
+++ b/test/enum_indexed_vector_test.cc
@@ -0,0 +1,75 @@
+/*
+    Copyright (C) 2022 Carl Hetherington <cth@carlh.net>
+
+    This file is part of DCP-o-matic.
+
+    DCP-o-matic is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 2 of the License, or
+    (at your option) any later version.
+
+    DCP-o-matic is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
+
+*/
+
+
+#include "lib/enum_indexed_vector.h"
+#include "test.h"
+#include <boost/test/unit_test.hpp>
+
+
+BOOST_AUTO_TEST_CASE(enum_indexed_vector_referencing_test)
+{
+	enum class Foo {
+		ONE,
+		TWO,
+		THREE,
+		COUNT
+	};
+
+	EnumIndexedVector<int, Foo> vector;
+	vector[Foo::ONE] = 1;
+	vector[Foo::TWO] = 2;
+	vector[Foo::THREE] = 3;
+
+	for (auto i: vector) {
+		++i;
+	}
+
+	BOOST_CHECK_EQUAL(vector[Foo::ONE], 1);
+	BOOST_CHECK_EQUAL(vector[Foo::TWO], 2);
+	BOOST_CHECK_EQUAL(vector[Foo::THREE], 3);
+
+	for (auto& i: vector) {
+		++i;
+	}
+
+	BOOST_CHECK_EQUAL(vector[Foo::ONE], 2);
+	BOOST_CHECK_EQUAL(vector[Foo::TWO], 3);
+	BOOST_CHECK_EQUAL(vector[Foo::THREE], 4);
+}
+
+
+BOOST_AUTO_TEST_CASE(enum_indexed_vector_indices_test)
+{
+	enum class Foo {
+		ONE,
+		TWO,
+		THREE,
+		COUNT
+	};
+
+	EnumIndexedVector<int, Foo> vector;
+
+	auto const ref = std::vector<Foo>{Foo::ONE, Foo::TWO, Foo::THREE};
+	BOOST_CHECK(vector.indices() == ref);
+
+	// Make sure nothing funny happens on the second call to indices
+	BOOST_CHECK(vector.indices() == ref);
+}
diff --git a/test/wscript b/test/wscript
index e9f24f0fe..0d39934ba 100644
--- a/test/wscript
+++ b/test/wscript
@@ -83,6 +83,7 @@ def build(bld):
                  empty_caption_test.cc
                  empty_test.cc
                  encryption_test.cc
+                 enum_indexed_vector_test.cc
                  file_extension_test.cc
                  ffmpeg_audio_only_test.cc
                  ffmpeg_audio_test.cc