Added type to control

microsoft · Jan 22, 2024 · ba33231 · ba33231
1 parent 123b84e
commit ba33231
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 9 deletions.
diff --git a/src/PAModel/Model/Control.cs b/src/PAModel/Model/Control.cs
@@ -2,8 +2,6 @@
 // Licensed under the MIT License.
 
 using System.Diagnostics;
-using System.Text.Json.Serialization;
-using Microsoft.PowerPlatform.Formulas.Tools.JsonConverters;
 using YamlDotNet.Serialization;
 
 namespace Microsoft.PowerPlatform.Formulas.Tools.Model;
@@ -20,6 +18,7 @@ public Control(ControlEditorState editorState)
     {
         EditorState = editorState ?? throw new ArgumentNullException(nameof(editorState));
         Name = editorState.Name;
+        Type = editorState.Type;
 
         if (editorState.Children != null)
         {
@@ -38,10 +37,8 @@ public Control(ControlEditorState editorState)
 
     public string Name { get; init; }
 
-    public string Type { get; set; }
-
-    [JsonConverter(typeof(JsonDoubleToIntConverter))]
-    public int Index { get; set; }
+    [YamlMember(Alias = "Control")]
+    public string Type { get; init; }
 
     public IList<Control> Controls { get; init; }
 

diff --git a/src/PAModel/Model/ControlEditorState.cs b/src/PAModel/Model/ControlEditorState.cs
@@ -1,12 +1,20 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
 
+using System.Text.Json.Serialization;
+using Microsoft.PowerPlatform.Formulas.Tools.JsonConverters;
+
 namespace Microsoft.PowerPlatform.Formulas.Tools.Model;
 
 public record ControlEditorState
 {
     public string Name { get; init; }
 
+    public string Type { get; set; }
+
+    [JsonConverter(typeof(JsonDoubleToIntConverter))]
+    public int Index { get; set; }
+
     public ControlEditorState[] Children { get; set; }
 
     public IList<RuleEditorState> Rules { get; init; }

diff --git a/src/PAModelTests/MsApp/MsappArchiveTests.cs b/src/PAModelTests/MsApp/MsappArchiveTests.cs
@@ -74,9 +74,11 @@ public void AddEntryTests(string[] entries)
     }
 
     [TestMethod]
-    [DataRow(@"Apps/WithYaml/HelloWorld.msapp", 14, 2, "HelloScreen", 8)]
-    [DataRow(@"Apps/AppWithLabel.msapp", 11, 2, "Screen1", 8)]
-    public void GetTopLevelControlsTests(string testFile, int allEntriesCount, int controlsCount, string topLevelControlName, int topLevelRulesCount)
+    [DataRow(@"Apps/WithYaml/HelloWorld.msapp", 14, 2, "HelloScreen", "screen", 8)]
+    [DataRow(@"Apps/AppWithLabel.msapp", 11, 2, "Screen1", "ControlInfo", 8)]
+    public void GetTopLevelControlsTests(string testFile, int allEntriesCount, int controlsCount,
+        string topLevelControlName, string topLevelControlType,
+        int topLevelRulesCount)
     {
         // Arrange: Create new ZipArchive in memory
         using var msappArchive = new MsappArchive(testFile);
@@ -88,5 +90,6 @@ public void GetTopLevelControlsTests(string testFile, int allEntriesCount, int c
 
         var topLevelControl = msappArchive.TopLevelControls.Single(c => c.Name == topLevelControlName);
         topLevelControl.EditorState.Rules.Count.Should().Be(topLevelRulesCount);
+        topLevelControl.Type.Should().Be(topLevelControlType);
     }
 }