1 files changed, 97 insertions, 0 deletions
diff --git a/ctrl/libhicnctrl/src/test/test_data.cc b/ctrl/libhicnctrl/src/test/test_data.cc
new file mode 100644
index 000000000..46debb0e7
--- /dev/null
+++ b/ctrl/libhicnctrl/src/test/test_data.cc
@@ -0,0 +1,97 @@
+/*
+ * 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 <sstream>
+
+extern "C" {
+#include <hicn/ctrl/objects.h>
+#include <hicn/ctrl/data.h>
+#include <hicn/ctrl/object.h>
+}
+
+#include "common.h"
+
+namespace {
+
+TEST_F(TestHicnLight, TestHicnLightData) {
+  hc_object_t object;
+  memset(&object, 0, sizeof(hc_object_t));
+
+  int rc;
+  hc_data_t* data = hc_data_create(OBJECT_TYPE_FACE);
+  hc_data_set_max_size(data, 2);
+
+  ASSERT_EQ(hc_data_get_size(data), 0) << "Initial data size should be zero";
+
+  /* Try to allocate more than max */
+  rc = hc_data_allocate(data, 5);
+  ASSERT_EQ(rc, -1) << "Allocating above max_size should fail";
+
+  /* Allocate room for two objects */
+  rc = hc_data_allocate(data, 2);
+  ASSERT_EQ(rc, 0) << "Allocating data the first time should succeed";
+
+  ASSERT_EQ(hc_data_get_size(data), 0)
+      << "Initial size should be 0 after allocation";
+
+  /* Try to allocate twice */
+  rc = hc_data_allocate(data, 2);
+  ASSERT_EQ(rc, -1) << "Allocating data multiple times should fail";
+
+  ASSERT_EQ(hc_data_get_size(data), 0)
+      << "Size after failed push should remain unchanged";
+
+  /* Push a first object */
+  rc = hc_data_push(data, &object);
+  ASSERT_EQ(rc, 0) << "First push should succeed";
+
+  ASSERT_EQ(hc_data_get_size(data), 1)
+      << "Data size first successful push should be 1";
+
+  /* Push a second object */
+  rc = hc_data_push(data, &object);
+  ASSERT_EQ(rc, 0) << "Second push should succeed";
+
+  ASSERT_EQ(hc_data_get_size(data), 2)
+      << "Data size after second successful push should be 2";
+
+  /* Push a third object, exceeding the allocated size */
+  rc = hc_data_push(data, &object);
+  ASSERT_EQ(rc, -1) << "Third push on full data of size 2 should fail";
+
+  /* Clear */
+  rc = hc_data_clear(data);
+  ASSERT_EQ(rc, 0) << "Clear should always succeed";
+
+  rc = hc_data_push(data, &object);
+  ASSERT_EQ(rc, 0) << "Pushing element after reallocation should succeed";
+
+  ASSERT_EQ(hc_data_get_size(data), 1) << "Size after first push should be one";
+  // XXX
+
+  /* Try to push an invalid object */
+  // XXX so far NULL
+  rc = hc_data_push(data, NULL);
+  ASSERT_EQ(rc, -1) << "Pushing invalid element should fail";
+
+  ASSERT_EQ(hc_data_get_size(data), 1)
+      << "Size after push failure should remain unchanged";
+  // XXX
+
+  hc_data_free(data);
+}
+
+}  // namespace