1 files changed, 72 insertions, 0 deletions
diff --git a/tests/unittests/test_packets.c b/tests/unittests/test_packets.c
new file mode 100644
index 0000000..be2e570
--- /dev/null
+++ b/tests/unittests/test_packets.c
@@ -0,0 +1,72 @@
+/*
+ * This file is part of RTRlib.
+ *
+ * This file is subject to the terms and conditions of the MIT license.
+ * See the file LICENSE in the top level directory for more details.
+ *
+ * Website: http://rtrlib.realmv6.org/
+ */
+
+#include "rtrlib_unittests.h"
+#include "test_packets.h"
+
+#include "rtrlib/rtr/packets_private.h"
+
+int __wrap_tr_send_all(const struct tr_socket *socket, const void *pdu, const size_t len, const time_t timeout)
+{
+	UNUSED(socket);
+	UNUSED(pdu);
+	UNUSED(len);
+	UNUSED(timeout);
+	return (int)mock();
+}
+
+void __wrap_rtr_change_socket_state(struct rtr_socket *rtr_socket, const enum rtr_socket_state new_state)
+{
+	UNUSED(rtr_socket);
+	UNUSED(new_state);
+}
+
+static void test_rtr_send_reset_query(void **state)
+{
+	struct rtr_socket socket;
+
+	UNUSED(state);
+
+	socket.connection_state_fp = NULL;
+
+	will_return(__wrap_tr_send_all, 0);
+	assert_int_equal(rtr_send_reset_query(&socket), RTR_ERROR);
+
+	will_return(__wrap_tr_send_all, 10);
+	assert_int_equal(rtr_send_reset_query(&socket), RTR_SUCCESS);
+}
+
+static void test_rtr_change_socket_state(void **state)
+{
+	struct rtr_socket socket;
+
+	UNUSED(state);
+
+	socket.connection_state_fp = NULL;
+	socket.state = RTR_RESET;
+
+	__real_rtr_change_socket_state(&socket, RTR_SYNC);
+	assert_int_equal(socket.state, RTR_SYNC);
+
+	__real_rtr_change_socket_state(&socket, RTR_SYNC);
+	assert_int_equal(socket.state, RTR_SYNC);
+
+	__real_rtr_change_socket_state(&socket, RTR_SHUTDOWN);
+	assert_int_equal(socket.state, RTR_SHUTDOWN);
+
+	__real_rtr_change_socket_state(&socket, RTR_ESTABLISHED);
+	assert_int_equal(socket.state, RTR_SHUTDOWN);
+}
+
+int main(void)
+{
+	const struct CMUnitTest tests[] = {cmocka_unit_test(test_rtr_send_reset_query),
+					   cmocka_unit_test(test_rtr_change_socket_state)};
+	return cmocka_run_group_tests(tests, NULL, NULL);
+}