add tests

include/td/common/DataBuffer.h

@@ -10,8 +10,8 @@
 #include <cstdint>
 #include <cstring>
 #include <string>
-#include <vector>
 #include <td/common/VarInt.h>
+#include <vector>
 
 namespace td {
 
@@ -76,11 +76,23 @@ class DataBuffer {
 
 	/**
 	 * \brief Append a vector to the buffer by first writing the size
-	 * \param data The buffer to append
+	 * \param data The vector to append
 	 */
 	template <typename T>
 	DataBuffer& operator<<(const std::vector<T>& data) {
-		*this << VarInt {data.size()};
+		*this << VarInt{data.size()};
+		for (const auto& element : data) {
+			*this << element;
+		}
+		return *this;
+	}
+
+	/**
+	 * \brief Append an array to the buffer by first writing the size
+	 * \param data The buffer to append
+	 */
+	template <typename T, std::size_t Size>
+	DataBuffer& operator<<(const std::array<T, Size>& data) {
 		for (const auto& element : data) {
 			*this << element;
 		}
@@ -112,7 +124,7 @@ class DataBuffer {
 	DataBuffer& operator>>(std::string& str);
 
 	/**
-	 * \brief Read a vector (size + data) from the buffer 
+	 * \brief Read a vector (size + data) from the buffer
 	 * \pre The vector is assumed to be empty
 	 */
 	template <typename T>
@@ -127,6 +139,19 @@ class DataBuffer {
 		return *this;
 	}
 
+	/**
+	 * \brief Read an array from the buffer
+	 */
+	template <std::size_t Size, typename T>
+	DataBuffer& operator>>(std::array<T, Size>& data) {
+		for (std::size_t i = 0; i < Size; i++) {
+			T newElement;
+			*this >> newElement;
+			data[i] = newElement;
+		}
+		return *this;
+	}
+
 	/**
 	 * \brief Write some data to the buffer
 	 * \param buffer The buffer to write

include/td/common/VarInt.h

@@ -2,7 +2,7 @@
 
 /**
  * \file VarInt.h
- * \brief File containing the blitz::VarInt class
+ * \brief File containing the td::VarInt class
  */
 
 #include <cstddef>

include/td/misc/Test.h

@@ -0,0 +1,70 @@
+#pragma once
+
+/**
+ * \file Test.h
+ * \brief File containing unit testing utilities
+ */
+
+#include <cstdlib>
+#include <td/misc/Log.h>
+
+namespace td {
+namespace test {
+
+/**
+ * \def TD_TEST_SUCCESSFUL
+ * \brief Used in tests to indicate that a test was successful
+ */
+#define TD_TEST_SUCCESSFUL 0
+
+/**
+ * \def BLITZ_TEST_FAILED
+ * \brief Used in tests to indicate that a test failed
+ */
+#define BLITZ_TEST_FAILED 1
+
+#ifndef __FUNCTION_NAME__
+#ifdef _WIN32
+#define __FUNCTION_NAME__ __FUNCTION__
+#else
+#define __FUNCTION_NAME__ __PRETTY_FUNCTION__
+#endif
+#endif
+
+/**
+ * \def blitz_test_assert
+ * \param ... The expression to evaluate
+ * \brief Evaluates the expression and exits the program if not valid.
+ * \note It works like a basic assert() but also in release mode
+ */
+#define td_test_assert(...)                                                                                                           \
+	if (!static_cast<bool>(__VA_ARGS__)) {                                                                                            \
+		td::test::LogAssert(#__VA_ARGS__, __FILE__, __LINE__, __FUNCTION_NAME__);                                                     \
+		std::exit(BLITZ_TEST_FAILED);                                                                                                 \
+	}
+
+
+/**
+ * \def blitz_debug_assert
+ * \param ... The expression to execute
+ * \brief Assertion without checks in release mode
+ * \note The expression is always executed. However, in release, no checks are made !
+ */
+#ifdef NDEBUG
+#define td_debug_assert(...) __VA_ARGS__
+#else
+#define td_debug_assert td_test_assert
+#endif
+
+
+/**
+ * \brief Prints an error message associated with a failed assertion
+ * \param expression The expression that was tested
+ * \param file The file in which the assertion failed
+ * \param line The line in the file in which the assertion failed
+ * \param function The function in which the assertion failed
+ */
+void LogAssert(const char* expression, const char* file, int line, const char* function);
+
+}  // namespace test
+}  // namespace td

include/td/protocol/packet/PacketSender.h

@@ -1,59 +0,0 @@
-#pragma once
-
-#include <td/protocol/packet/PacketVisitor.h>
-
-namespace td {
-
-class NetworkInterface;
-
-namespace protocol {
-
-#define DeclarePacket(PacketName, Reliability, ...) void Visit(const protocol::packets::PacketName& a_Packet) override;
-
-
-
-
-
-///////////////////////
-/* PacketBroadcaster */
-///////////////////////
-class PacketBroadcaster : public protocol::PacketVisitor {
-  private:
-	NetworkInterface& m_NetworkInterface;
-
-  public:
-	PacketBroadcaster(NetworkInterface& a_NetworkInterface) : m_NetworkInterface(a_NetworkInterface) {}
-
-	void BroadcastPacket(const protocol::Packet& a_Packet) {
-		Check(a_Packet);
-	}
-
-	DeclareAllPacket()
-};
-
-
-
-
-
-//////////////////
-/* PacketSender */
-//////////////////
-class PacketSender : public protocol::PacketVisitor {
-  private:
-	NetworkInterface& m_NetworkInterface;
-	PeerID m_PeerId;
-
-  public:
-	PacketSender(PeerID a_PeerId, NetworkInterface& a_NetworkInterface) : m_PeerId(a_PeerId), m_NetworkInterface(a_NetworkInterface) {}
-
-	void SendPacket(const protocol::Packet& a_Packet) {
-		Check(a_Packet);
-	}
-
-	DeclareAllPacket()
-};
-
-#undef DeclarePacket
-
-}  // namespace protocol
-}  // namespace td