AST-1637 Support model data in BdaBuffer

base/test/unit/tBdaBuffer.cc

@@ -472,6 +472,35 @@ BOOST_AUTO_TEST_CASE(add_data_after_adding_rows) {
   // initialize values.
 }
 
+BOOST_AUTO_TEST_CASE(remove_main_data) {
+  BdaBuffer buffer(kDataSize);
+  AddBasicRow(buffer);
+  BOOST_CHECK(buffer.HasData(""));
+  BOOST_CHECK(buffer.GetData());
+
+  buffer.RemoveData("");
+  BOOST_CHECK(!buffer.HasData(""));
+  BOOST_CHECK(!buffer.GetData());
+
+  BOOST_CHECK_NO_THROW(buffer.RemoveData(""));
+}
+
+BOOST_AUTO_TEST_CASE(remove_named_data) {
+  BdaBuffer buffer(kDataSize, BdaBuffer::Fields(false));
+  buffer.AddData(kDataName);
+  AddBasicRow(buffer);
+  BOOST_CHECK(!buffer.HasData(""));
+  BOOST_CHECK(buffer.HasData(kDataName));
+  BOOST_CHECK_EQUAL(buffer.GetNumberOfElements(), kDataSize);
+
+  buffer.RemoveData(kDataName);
+  BOOST_CHECK(!buffer.HasData(""));
+  BOOST_CHECK(!buffer.HasData(kDataName));
+  BOOST_CHECK_EQUAL(buffer.GetNumberOfElements(), kDataSize);
+
+  BOOST_CHECK_NO_THROW(buffer.RemoveData(kDataName));
+}
+
 BOOST_AUTO_TEST_CASE(clear) {
   BdaBuffer buffer(3 * kDataSize);
   buffer.AddData(kDataName);