[ksqlDB.RestApi.Client]: enabled specifying of pseudocolumns using the Fluent API's model builder unit tests #96

Author: tomasfabian

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

@@ -367,9 +367,29 @@ public void AsStruct()
       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");
+      var metadata = entityMetadata!.FieldsMetadata.First(c => c is {Path: nameof(Record.Headers)});
       metadata.IsStruct.Should().BeTrue();
     }
+
+    [Test]
+    public void AsPseudoColumn()
+    {
+      //Arrange
+
+      //Act
+      var fieldTypeBuilder = builder.Entity<Record>()
+        .Property(b => b.Headers)
+        .AsPseudoColumn();
+
+      //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 is {Path: nameof(Record.Headers)});
+      metadata.IsPseudoColumn.Should().BeTrue();
+    }
   }
 
   internal record Payment

Tests/ksqlDB.RestApi.Client.Tests/FluentAPI/FieldTypeBuilderTests.cs

@@ -32,6 +32,7 @@ public void InitState()
 
       fieldMetadata.HasHeaders.Should().BeFalse();
       fieldMetadata.IsStruct.Should().BeFalse();
+      fieldMetadata.IsPseudoColumn.Should().BeFalse();
       fieldMetadata.ColumnName.Should().BeNull();
     }
 
@@ -93,6 +94,19 @@ public void AsStruct()
       fieldMetadata.IsStruct.Should().BeTrue();
     }
 
+    [Test]
+    public void AsPseudoColumn()
+    {
+      //Arrange
+
+      //Act
+      var fieldTypeBuilder = builder.AsPseudoColumn();
+
+      //Assert
+      fieldTypeBuilder.Should().NotBeNull();
+      fieldMetadata.IsPseudoColumn.Should().BeTrue();
+    }
+
     [Test]
     public void HasColumnName()
     {

Tests/ksqlDB.RestApi.Client.Tests/KSql/RestApi/Parsers/IdentifierUtilTests.cs

@@ -1,6 +1,9 @@
+using System.Linq.Expressions;
+using ksqlDb.RestApi.Client.FluentAPI.Builders;
 using ksqlDB.RestApi.Client.KSql.RestApi.Enums;
 using ksqlDb.RestApi.Client.KSql.RestApi.Parsers;
 using NUnit.Framework;
+using FluentAssertions;
 
 namespace ksqlDb.RestApi.Client.Tests.KSql.RestApi.Parsers
 {
@@ -31,5 +34,56 @@ public class IdentifierUtilTests
     [TestCase(SystemColumns.WINDOWEND, IdentifierEscaping.Always, ExpectedResult = $"`{SystemColumns.WINDOWEND}`")]
     public string ShouldBeFormatted(string identifier, IdentifierEscaping escaping) =>
       IdentifierUtil.Format(identifier, escaping);
+
+    private ModelBuilder builder = null!;
+
+    [SetUp]
+    public void TestInitialize()
+    {
+      builder = new();
+    }
+
+    private class Record
+    {
+      public long RowTime { get; }
+    }
+
+    [TestCase(IdentifierEscaping.Keywords)]
+    [TestCase(IdentifierEscaping.Always)]
+    public void Format_MemberExpression_AsPseudoColumn(IdentifierEscaping escaping)
+    {
+      //Arrange
+      builder.Entity<Record>()
+        .Property(c => c.RowTime)
+        .AsPseudoColumn();
+
+      Expression<Func<Record, long>> expression = c => c.RowTime;
+      var memberExpression = (MemberExpression) expression.Body;
+
+      //Act
+      var formattedIdentifier = IdentifierUtil.Format(memberExpression, escaping, builder);
+
+      //Assert
+      formattedIdentifier.Should().Be(nameof(Record.RowTime));
+    }
+
+    [TestCase(IdentifierEscaping.Keywords)]
+    [TestCase(IdentifierEscaping.Always)]
+    public void Format_MemberInfo_AsPseudoColumn(IdentifierEscaping escaping)
+    {
+      //Arrange
+      builder.Entity<Record>()
+        .Property(c => c.RowTime)
+        .AsPseudoColumn();
+
+      Expression<Func<Record, long>> expression = c => c.RowTime;
+      var memberInfo = ((MemberExpression)expression.Body).Member;
+
+      //Act
+      var formattedIdentifier = IdentifierUtil.Format(memberInfo, escaping, builder);
+
+      //Assert
+      formattedIdentifier.Should().Be(nameof(Record.RowTime));
+    }
   }
 }