add unit tests for preserveMediaQueries

garbidge · garbidge · commit 22ad181d62de · 2021-11-02T13:15:35.000+10:30
diff --git a/PreMailer.Net/PreMailer.Net.Tests/PreMailerTests.cs b/PreMailer.Net/PreMailer.Net.Tests/PreMailerTests.cs
@@ -137,6 +137,26 @@ public void MoveCssInline_KeepStyleElementsIgnoreElementsMatchesStyleElement_Doe
 			Assert.Contains("<style id=\"ignore\" type=\"text/css\">", premailedOutput.Html);
 		}
 
+		[Fact]
+		public void MoveCssInline_PreserveMediaQueries_RemovesStyleElementsWithoutMediaQueries()
+		{
+			string input = "<html><head><style>div { width: 42px; }</style></head><body><div>test</div></body></html>";
+
+			var premailedOutput = PreMailer.MoveCssInline(input, removeStyleElements: true, preserveMediaQueries: true);
+
+			Assert.DoesNotContain("<style>", premailedOutput.Html);
+		}
+
+		[Fact]
+		public void MoveCssInline_PreserveMediaQueries_PreservesStyleElementsWithMediaQueries()
+		{
+			string input = "<html><head><style>div { width: 42px; }  @media (max-width: 250px) { div { width: 20px; } }</style></head><body><div>test</div></body></html>";
+
+			var premailedOutput = PreMailer.MoveCssInline(input, removeStyleElements: true, preserveMediaQueries: true);
+
+			Assert.Contains("<style>@media (max-width: 250px) { div { width: 20px; } }</style>", premailedOutput.Html);
+		}
+
 		[Fact]
 		public void MoveCssInline_MultipleSelectors_HonorsIndividualSpecificity()
 		{
