1 files changed, 154 insertions, 0 deletions
diff --git a/hicn-light/src/hicn/test/test-probe_generator.cc b/hicn-light/src/hicn/test/test-probe_generator.cc
new file mode 100644
index 000000000..4d6cfa8f7
--- /dev/null
+++ b/hicn-light/src/hicn/test/test-probe_generator.cc
@@ -0,0 +1,154 @@
+/*
+ * Copyright (c) 2021 Cisco and/or its affiliates.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include <gtest/gtest.h>
+
+#include <netinet/in.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/socket.h>
+#include <sys/un.h>
+#include <unistd.h>
+#include <arpa/inet.h>
+
+extern "C" {
+#define WITH_TESTS
+#include <hicn/strategies/probe_generator.h>
+}
+
+class ProbeGeneratorTest : public ::testing::Test {
+ protected:
+  ProbeGeneratorTest() {}
+
+  virtual ~ProbeGeneratorTest() {}
+};
+
+TEST_F(ProbeGeneratorTest, ProbeGeneratorRegisterProbe) {
+  probe_generator_t *pg = create_probe_generator();
+  EXPECT_FALSE(pg == nullptr);
+
+  register_probe(pg, 1);
+  register_probe(pg, 2);
+  register_probe(pg, 3);
+  register_probe(pg, 4);
+
+  Ticks t = get_probe_send_time(pg, 1);
+  EXPECT_TRUE(t != 0);
+
+  t = get_probe_send_time(pg, 2);
+  EXPECT_TRUE(t != 0);
+
+  t = get_probe_send_time(pg, 3);
+  EXPECT_TRUE(t != 0);
+
+  t = get_probe_send_time(pg, 4);
+  EXPECT_TRUE(t != 0);
+
+  t = get_probe_send_time(pg, 5);
+  EXPECT_FALSE(t != 0);
+
+  destroy_probe_generator(pg);
+}
+
+TEST_F(ProbeGeneratorTest, ProbeGeneratorTime) {
+  probe_generator_t *pg = create_probe_generator();
+  EXPECT_FALSE(pg == nullptr);
+
+  Ticks t1 = register_probe(pg, 1);
+  Ticks t2 = get_probe_send_time(pg, 1);
+
+  EXPECT_TRUE(t2 != 0);
+  EXPECT_TRUE(t1 == t2);
+
+  destroy_probe_generator(pg);
+}
+
+TEST_F(ProbeGeneratorTest, ProbeGeneratorDeleteProbe) {
+  probe_generator_t *pg = create_probe_generator();
+  EXPECT_FALSE(pg == nullptr);
+
+  register_probe(pg, 1);
+  register_probe(pg, 2);
+  register_probe(pg, 3);
+  register_probe(pg, 4);
+
+  Ticks t = get_probe_send_time(pg, 1);
+  EXPECT_TRUE(t != 0);
+
+  t = get_probe_send_time(pg, 2);
+  EXPECT_TRUE(t != 0);
+
+  t = get_probe_send_time(pg, 3);
+  EXPECT_TRUE(t != 0);
+
+  t = get_probe_send_time(pg, 4);
+  EXPECT_TRUE(t != 0);
+
+  t = get_probe_send_time(pg, 5);
+  EXPECT_FALSE(t != 0);
+
+  delete_probe(pg, 1);
+  delete_probe(pg, 3);
+
+  t = get_probe_send_time(pg, 1);
+  EXPECT_FALSE(t != 0);
+
+  t = get_probe_send_time(pg, 3);
+  EXPECT_FALSE(t != 0);
+
+  destroy_probe_generator(pg);
+}
+
+TEST_F(ProbeGeneratorTest, ProbeGeneratorDeleteAll) {
+  probe_generator_t *pg = create_probe_generator();
+  EXPECT_FALSE(pg == nullptr);
+
+  register_probe(pg, 1);
+  register_probe(pg, 2);
+  register_probe(pg, 3);
+  register_probe(pg, 4);
+
+  Ticks t = get_probe_send_time(pg, 1);
+  EXPECT_TRUE(t != 0);
+
+  t = get_probe_send_time(pg, 2);
+  EXPECT_TRUE(t != 0);
+
+  t = get_probe_send_time(pg, 3);
+  EXPECT_TRUE(t != 0);
+
+  t = get_probe_send_time(pg, 4);
+  EXPECT_TRUE(t != 0);
+
+  t = get_probe_send_time(pg, 5);
+  EXPECT_FALSE(t != 0);
+
+  delete_all_probes(pg);
+
+  t = get_probe_send_time(pg, 1);
+  EXPECT_FALSE(t != 0);
+
+  t = get_probe_send_time(pg, 2);
+  EXPECT_FALSE(t != 0);
+
+  t = get_probe_send_time(pg, 3);
+  EXPECT_FALSE(t != 0);
+
+  t = get_probe_send_time(pg, 4);
+  EXPECT_FALSE(t != 0);
+
+  destroy_probe_generator(pg);
+}