add some tests for records

Weingartner · May 10, 2023 · e0c3fca · e0c3fca
1 parent 4533daf
commit e0c3fca
Show file tree

Hide file tree

Showing 2 changed files with 63 additions and 0 deletions.
diff --git a/Weingartner.Json.Migration.Roslyn.Spec/DataContractAnalyzerSpec.cs b/Weingartner.Json.Migration.Roslyn.Spec/DataContractAnalyzerSpec.cs
@@ -10,6 +10,7 @@ namespace Weingartner.Json.Migration.Roslyn.Spec
 {
     public class DataContractAnalyzerSpec : CodeFixVerifier
     {
+
         [Fact]
         public void ShouldNotCreateDiagnosticIfTypeIsNotMigratable()
         {
@@ -117,5 +118,32 @@ protected override DiagnosticAnalyzer GetCSharpDiagnosticAnalyzer()
         {
             return new DataContractAnalyzer();
         }
+
+        [Fact]
+        public void ShouldCreateDiagnosticIfRecord()
+        {
+            var source = @"
+using Weingartner.Json.Migration;
+using System.Runtime.Serialization;
+
+[Migratable("""")]
+record TypeName
+{
+    [DataMember]
+    public int A { get; set; }
+}";
+            var expected = new DiagnosticResult
+            {
+                Id = DataContractAnalyzer.DiagnosticId,
+                Message = string.Format(DataContractAnalyzer.MessageFormat.ToString(CultureInfo.InvariantCulture), "TypeName"),
+                Severity = DiagnosticSeverity.Error,
+                Locations =
+                    new[] {
+                            new DiagnosticResultLocation("Test0.cs", 5, 7)
+                        }
+            };
+
+            VerifyCSharpDiagnostic(source, expected);
+        }
     }
 }
diff --git a/Weingartner.Json.Migration.Roslyn.Spec/MigrationHashAnalyzerSpec.cs b/Weingartner.Json.Migration.Roslyn.Spec/MigrationHashAnalyzerSpec.cs
@@ -163,6 +163,41 @@ class TypeName
             VerifyCSharpFix(source, expected);
         }
 
+        [Fact]
+        public void ShouldFixRecordIfInCorrectIsSpecified()
+        {
+            var source = @"
+using Weingartner.Json.Migration;
+using System.Runtime.Serialization;
+
+[Migratable(""758832573"")]
+[DataContract]
+record TypeName
+{
+    [DataMember]
+    public int A { get; set; }
+    [DataMember]
+    public double B { get; set; }
+}";
+
+            var expected = @"
+using Weingartner.Json.Migration;
+using System.Runtime.Serialization;
+
+[Migratable(""687340935"")]
+[DataContract]
+record TypeName
+{
+    [DataMember]
+    public int A { get; set; }
+    [DataMember]
+    public double B { get; set; }
+}";
+
+            VerifyCSharpFix(source, expected);
+        }
+
+
         protected override CodeFixProvider GetCSharpCodeFixProvider()
         {
             return new MigrationHashAnalyzerCodeFixProvider();