[ntuple] add serialization test of descriptor with attribute sets

Author: silverweed

tree/ntuple/test/ntuple_serialize.cxx

@@ -2082,3 +2082,93 @@ TEST(RNTuple, SerializeAttrSets)
 
    ASSERT_EQ(attrSetDesc, deserializedDesc);
 }
+
+TEST(RNTuple, SerializeDescriptorWithAttrSets)
+{
+   RNTupleDescriptorBuilder builder;
+   builder.SetVersionForWriting();
+   builder.SetNTuple("ntpl", "");
+
+   builder.AddField(RFieldDescriptorBuilder()
+                       .FieldId(0)
+                       .FieldName("")
+                       .Structure(ROOT::ENTupleStructure::kRecord)
+                       .MakeDescriptor()
+                       .Unwrap());
+
+   builder.AddField(RFieldDescriptorBuilder()
+                       .FieldId(5)
+                       .FieldName("pt")
+                       .TypeName("float")
+                       .Structure(ROOT::ENTupleStructure::kPlain)
+                       .MakeDescriptor()
+                       .Unwrap());
+   builder.AddFieldLink(0, 5);
+   builder.AddColumn(RColumnDescriptorBuilder()
+                        .LogicalColumnId(0)
+                        .PhysicalColumnId(0)
+                        .FieldId(5)
+                        .BitsOnStorage(32)
+                        .Type(ROOT::ENTupleColumnType::kReal32)
+                        .FirstElementIndex(1)
+                        .MakeDescriptor()
+                        .Unwrap());
+   builder.AddColumn(RColumnDescriptorBuilder()
+                        .LogicalColumnId(1)
+                        .PhysicalColumnId(1)
+                        .FieldId(5)
+                        .BitsOnStorage(16)
+                        .Type(ROOT::ENTupleColumnType::kReal16)
+                        .RepresentationIndex(1)
+                        .FirstElementIndex(1)
+                        .SetSuppressedDeferred()
+                        .MakeDescriptor()
+                        .Unwrap());
+
+   RNTupleLocator locator;
+   locator.SetType(ROOT::RNTupleLocator::kTypeFile);
+   locator.SetPosition(128ul);
+   auto attrSetDesc = ROOT::Experimental::Internal::RNTupleAttrSetDescriptor{1,    // schema version major
+                                                                             0,    // schema version minor
+                                                                             1024, // uncomp len
+                                                                             locator, "AttrSetName"};
+   // First cluster
+   RClusterDescriptorBuilder clusterBuilder;
+   clusterBuilder.ClusterId(0).FirstEntryIndex(0).NEntries(1);
+   builder.AddCluster(clusterBuilder.MoveDescriptor().Unwrap());
+
+   RClusterGroupDescriptorBuilder cgBuilder;
+   cgBuilder.ClusterGroupId(0).NClusters(1).EntrySpan(3);
+   cgBuilder.AddSortedClusters({0});
+   builder.AddClusterGroup(cgBuilder.MoveDescriptor().Unwrap());
+
+   builder.AddAttributeSet(std::move(attrSetDesc));
+   auto desc = builder.MoveDescriptor();
+
+   auto context = RNTupleSerializer::SerializeHeader(nullptr, desc).Unwrap();
+   auto bufHeader = MakeUninitArray<unsigned char>(context.GetHeaderSize());
+   context = RNTupleSerializer::SerializeHeader(bufHeader.get(), desc).Unwrap();
+
+   std::vector<ROOT::DescriptorId_t> physClusterIDs{context.MapClusterId(0)};
+   context.MapClusterGroupId(0);
+   auto sizePageList = RNTupleSerializer::SerializePageList(nullptr, desc, physClusterIDs, context).Unwrap();
+   EXPECT_GT(sizePageList, 0);
+   auto bufPageList = MakeUninitArray<unsigned char>(sizePageList);
+   EXPECT_EQ(sizePageList,
+             RNTupleSerializer::SerializePageList(bufPageList.get(), desc, physClusterIDs, context).Unwrap());
+
+   auto sizeFooter = RNTupleSerializer::SerializeFooter(nullptr, desc, context).Unwrap();
+   EXPECT_GT(sizeFooter, 0);
+   auto bufFooter = MakeUninitArray<unsigned char>(sizeFooter);
+   EXPECT_EQ(sizeFooter, RNTupleSerializer::SerializeFooter(bufFooter.get(), desc, context).Unwrap());
+
+   builder.SetVersionForWriting();
+   RNTupleSerializer::DeserializeHeader(bufHeader.get(), context.GetHeaderSize(), builder).ThrowOnError();
+   RNTupleSerializer::DeserializeFooter(bufFooter.get(), sizeFooter, builder);
+   auto deserializedDesc = builder.MoveDescriptor();
+   RNTupleSerializer::DeserializePageList(bufPageList.get(), sizePageList, 0, deserializedDesc,
+                                          EDeserializeMode::kForReading);
+
+   ASSERT_EQ(ROOT::Experimental::Internal::GetAttributeSets(desc),
+             ROOT::Experimental::Internal::GetAttributeSets(deserializedDesc));
+}