[ksqlDb.RestApi.Client]: added AsStruct to Fluent API unit tests #89

Author: tomasfabian

Tests/ksqlDB.RestApi.Client.Tests/FluentAPI/Builders/ModelBuilderTests.cs

@@ -234,6 +234,37 @@ public void Headers()
 
       metadata.HasHeaders.Should().BeTrue();
     }
+
+    private record KeyValuePair
+    {
+      public string Key { get; set; } = null!;
+      public byte[] Value { get; set; } = null!;
+    }
+
+    private record Record
+    {
+      public KeyValuePair[] Headers { get; init; } = null!;
+    }
+
+    [Test]
+    public void AsStruct()
+    {
+      //Arrange
+
+      //Act
+      var fieldTypeBuilder = builder.Entity<Record>()
+        .Property(b => b.Headers)
+        .AsStruct();
+
+      //Assert
+      fieldTypeBuilder.Should().NotBeNull();
+
+      var entityMetadata = ((IMetadataProvider)builder).GetEntities().FirstOrDefault(c => c.Type == typeof(Record));
+      entityMetadata.Should().NotBeNull();
+
+      var metadata = entityMetadata!.FieldsMetadata.First(c => c.IsStruct && c.Path == "Headers");
+      metadata.IsStruct.Should().BeTrue();
+    }
   }
 
   internal record Payment

Tests/ksqlDB.RestApi.Client.Tests/KSql/RestApi/Generators/StatementGeneratorTests.cs

@@ -1,4 +1,5 @@
 using FluentAssertions;
+using ksqlDb.RestApi.Client.FluentAPI.Builders;
 using ksqlDB.RestApi.Client.KSql.RestApi.Enums;
 using ksqlDB.RestApi.Client.KSql.RestApi.Generators;
 using ksqlDB.RestApi.Client.KSql.RestApi.Serialization;
@@ -162,6 +163,36 @@ public void CreateOrReplaceTableWithEnumProperty()
     //Assert
     statement.Should().Contain($"{nameof(PortType)} VARCHAR");
   }
+  private record KeyValuePair
+  {
+    public string Key { get; set; } = null!;
+    public byte[] Value { get; set; } = null!;
+  }
+
+  private record Record
+  {
+    public KeyValuePair[] Headers { get; init; } = null!;
+  }
+
+  [Test]
+  public void CreateTable_UseModelBuilder_WithFieldAsStruct()
+  {
+    //Arrange
+    var modelBuilder = new ModelBuilder();
+    modelBuilder.Entity<Record>()
+      .Property(b => b.Headers)
+      .AsStruct();
+
+    var creationMetadata = new EntityCreationMetadata("my_topic", partitions: 3);
+
+    //Act
+    var statement = new StatementGenerator(modelBuilder).CreateTable<Record>(creationMetadata, ifNotExists: true);
+
+    //Assert
+    statement.Should().Be(@"CREATE TABLE IF NOT EXISTS Records (
+	Headers ARRAY<STRUCT<Key VARCHAR, Value BYTES>>
+) WITH ( KAFKA_TOPIC='my_topic', VALUE_FORMAT='Json', PARTITIONS='3' );".ReplaceLineEndings());
+  }
 }
 
 internal class Port

Tests/ksqlDB.RestApi.Client.Tests/KSql/RestApi/Statements/KSqlTypeTranslatorTests.cs

@@ -12,7 +12,7 @@ namespace ksqlDb.RestApi.Client.Tests.KSql.RestApi.Statements
   public class KSqlTypeTranslatorTests
   {
     private ModelBuilder modelBuilder = null!;
-    private KSqlTypeTranslator kSqlTypeTranslator = null!;
+    private KSqlTypeTranslator<Poco> kSqlTypeTranslator = null!;
 
     [SetUp]
     public void Init()
@@ -396,7 +396,7 @@ public void Translate_UseModelBuilderConfiguration()
         .Decimal(10, 2);
 
       //Act
-      string ksqlType = kSqlTypeTranslator.Translate(type);
+      string ksqlType = kSqlTypeTranslator.Translate(type, type.GetMember(nameof(Poco.Amount))[0]);
 
       //Assert
       ksqlType.Should().Be($"{KSqlTypes.Struct}<{nameof(Poco.Amount)} {KSqlTypes.Decimal}(10,2)>");