Merge branch 'master' into feature/dotnet_core_8

Puchaczov · Jun 8, 2024 · 3c76447 · 3c76447
2 parents 5a0c78c + 1af7548
commit 3c76447
Show file tree

Hide file tree

Showing 4 changed files with 65 additions and 6 deletions.
diff --git a/Musoq.Parser.Tests/NodesTests.cs b/Musoq.Parser.Tests/NodesTests.cs
@@ -0,0 +1,50 @@
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using Musoq.Parser.Nodes;
+
+namespace Musoq.Parser.Tests;
+
+[TestClass]
+public class NodesTests
+{
+    [TestMethod]
+    public void WhenOrderByNode_ShouldReturnString()
+    {
+        var node = new OrderByNode(new []{
+            new FieldOrderedNode(new AccessColumnNode("col1", string.Empty, TextSpan.Empty), 0, null, Order.Ascending)
+        });
+
+        Assert.AreEqual("order by col1 ascending", node.ToString());
+    } 
+
+    [TestMethod]
+    public void WhenOrderByDescendingNode_ShouldReturnString()
+    {
+        var node = new OrderByNode(new []{
+            new FieldOrderedNode(new AccessColumnNode("col1", string.Empty, TextSpan.Empty), 0, null, Order.Descending)
+        });
+
+        Assert.AreEqual("order by col1 descending", node.ToString());
+    }
+
+    [TestMethod]
+    public void WhenOrderByMultipleNodes_ShouldReturnString()
+    {
+        var node = new OrderByNode(new[]{
+            new FieldOrderedNode(new AccessColumnNode("col1", string.Empty, TextSpan.Empty), 0, null, Order.Ascending),
+            new FieldOrderedNode(new AccessColumnNode("col2", string.Empty, TextSpan.Empty), 1, null, Order.Ascending)
+        });
+
+        Assert.AreEqual("order by col1 ascending, col2 ascending", node.ToString());
+    }
+
+    [TestMethod]
+    public void WhenOrderByMultipleNodesWithDifferentOrder_ShouldReturnString()
+    {
+        var node = new OrderByNode(new []{
+            new FieldOrderedNode(new AccessColumnNode("col1", string.Empty, TextSpan.Empty), 0, null, Order.Ascending),
+            new FieldOrderedNode(new AccessColumnNode("col2", string.Empty, TextSpan.Empty), 1, null, Order.Descending)
+        });
+
+        Assert.AreEqual("order by col1 ascending, col2 descending", node.ToString());
+    }
+}
diff --git a/Musoq.Parser/Nodes/FieldNode.cs b/Musoq.Parser/Nodes/FieldNode.cs
@@ -8,9 +8,9 @@ public class FieldNode : Node
 
         public FieldNode(Node expression, int fieldOrder, string fieldName)
         {
+            _fieldName = fieldName;
             Expression = expression;
             FieldOrder = fieldOrder;
-            _fieldName = fieldName;
             Id = $"{nameof(FieldNode)}{expression.Id}";
         }
 

diff --git a/Musoq.Parser/Nodes/FieldOrderedNode.cs b/Musoq.Parser/Nodes/FieldOrderedNode.cs
@@ -2,14 +2,23 @@
 {
     public class FieldOrderedNode : FieldNode
     {
-        public FieldOrderedNode(Node expression, int fieldOrder, string fieldName, Order order)
-            : base(expression, fieldOrder, fieldName) => Order = order;
+        public FieldOrderedNode(Node expression, int fieldOrder, string fieldName, Order order) : base(expression, fieldOrder, fieldName)
+        {
+            Order = order;
+        }
 
         public Order Order { get; }
 
         public override void Accept(IExpressionVisitor visitor)
         {
             visitor.Visit(this);
         }
+
+        public override string Id => $"{nameof(FieldOrderedNode)}{Expression.Id}{Order}";
+
+        public override string ToString()
+        {
+            return $"{Expression.ToString()} {Order.ToString().ToLowerInvariant()}";
+        }
     }
 }
diff --git a/Musoq.Parser/Nodes/OrderByNode.cs b/Musoq.Parser/Nodes/OrderByNode.cs
@@ -14,7 +14,7 @@ public OrderByNode(FieldOrderedNode[] fields)
 
         public FieldOrderedNode[] Fields { get; }
 
-        public override Type ReturnType { get; }
+        public override Type ReturnType { get; } = typeof(void);
 
         public override string Id { get; }
 
@@ -31,9 +31,9 @@ public override string ToString()
             return $"order by {fieldsTxt}";
         }
 
-        private string FieldToString(FieldNode node)
+        private static string FieldToString(FieldOrderedNode node)
         {
-            return string.IsNullOrEmpty(node.FieldName) ? node.Expression.ToString() : node.FieldName;
+            return node.ToString();
         }
     }
 }