From 554734659aa115fd9ddf24fbe44ce233ff124694 Mon Sep 17 00:00:00 2001
From: Babicaa <andrej.babic@psi.ch>
Date: Mon, 19 Nov 2018 18:10:38 +0100
Subject: [PATCH] Add tests for MetadataBuffer

---
 lib/test/test_MetadataBuffer.cpp | 43 ++++++++++++++++++++++++++++++++
 lib/test/test_main.cpp           |  3 ++-
 2 files changed, 45 insertions(+), 1 deletion(-)
 create mode 100644 lib/test/test_MetadataBuffer.cpp

diff --git a/lib/test/test_MetadataBuffer.cpp b/lib/test/test_MetadataBuffer.cpp
new file mode 100644
index 0000000..054f395
--- /dev/null
+++ b/lib/test/test_MetadataBuffer.cpp
@@ -0,0 +1,43 @@
+
+#include "../src/MetadataBuffer.hpp"
+using namespace std;
+
+TEST(MetadataBuffer, basic_operations)
+{
+
+    int n_frames = 10;
+    int n_modules = 3;
+
+    auto header_values = shared_ptr<unordered_map<string, HeaderDataType>>(new unordered_map<string, HeaderDataType> {
+        {"frame", HeaderDataType("uint64")},
+        {"module_number", HeaderDataType("uint64", n_modules)}
+    });
+
+    MetadataBuffer metadata_buffer(10, header_values);
+
+    uint64_t base_frame_data = 12345678;
+    uint64_t base_module_number = 0;
+
+    for (int i = 0; i<n_frames; i++) {
+	uint64_t frame_index = base_frame_data + i;
+	metadata_buffer.add_metadata_to_buffer("frame", i, (char*)&frame_index);
+
+	uint64_t module_number = base_module_number + i;
+	uint64_t module_number_data[n_modules];
+
+	for (int module_index=0; module_index<n_modules; module_index++) {
+	    module_number_data[module_index] = module_number;
+	}
+	
+	metadata_buffer.add_metadata_to_buffer("module_number", i, (char*)&module_number_data);
+    }
+
+    uint64_t* frame_data = (uint64_t*) metadata_buffer.get_metadata_values("frame").get();
+    uint64_t* modulo_number_data = (uint64_t*) metadata_buffer.get_metadata_values("modulo_number").get();
+
+    for (int i = 0; i<n_frames; i++) {
+	EXPECT_EQ(frame_data[i], base_frame_data+i);
+    }
+
+}
+
diff --git a/lib/test/test_main.cpp b/lib/test/test_main.cpp
index 20b340e..7dfcfd9 100644
--- a/lib/test/test_main.cpp
+++ b/lib/test/test_main.cpp
@@ -1,10 +1,11 @@
 #include "gtest/gtest.h"
 #include "test_ZmqReceiver.cpp"
 #include "test_H5Writer.cpp"
+#include "test_MetadataBuffer.cpp"
 
 using namespace std;
 
 int main(int argc, char **argv) {
   ::testing::InitGoogleTest(&argc, argv);
   return RUN_ALL_TESTS();
-}
\ No newline at end of file
+}