From 6d2b97fc49c024a2da4543756a77cb1752caa89c Mon Sep 17 00:00:00 2001
From: eeff <33566201+eeff@users.noreply.github.com>
Date: Mon, 14 Mar 2022 22:15:17 +0800
Subject: [PATCH] Update datatag table unit tests

---
 tests/datatag_table_test.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/datatag_table_test.cc b/tests/datatag_table_test.cc
index eb4f28b8..7e86b9e9 100644
--- a/tests/datatag_table_test.cc
+++ b/tests/datatag_table_test.cc
@@ -82,6 +82,7 @@ TEST(DatatageTableTest, DatatagTableAdd)
     EXPECT_EQ(0, neu_datatag_tbl_add(datatag_table, datatag1));
     EXPECT_EQ(0, neu_datatag_tbl_add(datatag_table, datatag2));
     EXPECT_EQ(0, neu_datatag_tbl_add(datatag_table, datatag3));
+    EXPECT_EQ(3, neu_datatag_tbl_size(datatag_table));
 
     neu_datatag_tbl_destroy(datatag_table);
 }
@@ -101,9 +102,11 @@ TEST(DatatageTableTest, DatatagTableRemove)
     neu_datatag_t *datatag_id_return3 = NULL;
 
     datatag_id1 = neu_datatag_tbl_add(datatag_table, datatag1);
+    EXPECT_EQ(1, neu_datatag_tbl_size(datatag_table));
     EXPECT_EQ(0, neu_datatag_tbl_remove(datatag_table, datatag_id1));
     datatag_id_return1 = neu_datatag_tbl_get(datatag_table, datatag_id1);
     EXPECT_EQ(NULL, datatag_id_return1);
+    EXPECT_EQ(0, neu_datatag_tbl_size(datatag_table));
 
     datatag1 = rand_tag(addr_str, "tag1");
 
@@ -161,6 +164,7 @@ TEST(DatatageTableTest, DatatagTableUpdate)
     EXPECT_EQ(datatag_id_return->type, datatag4->type);
     EXPECT_EQ(datatag_id_return->id, datatag4->id);
     EXPECT_EQ(datatag_id_return->addr_str, datatag4->addr_str);
+    EXPECT_EQ(2, neu_datatag_tbl_size(datatag_table));
 
     EXPECT_EQ(0, neu_datatag_tbl_remove(datatag_table, datatag_id2));
     EXPECT_EQ(0, neu_datatag_tbl_update(datatag_table, datatag_id1, datatag3));