Tests

Author: nvborisenko

dotnet/src/webdriver/BiDi/BrowsingContext/DownloadEndEventArgs.cs

@@ -18,18 +18,18 @@
 // </copyright>
 
 using System;
-using System.Text.Json.Serialization;
 
 namespace OpenQA.Selenium.BiDi.BrowsingContext;
 
-[JsonPolymorphic(TypeDiscriminatorPropertyName = "status")]
-[JsonDerivedType(typeof(DownloadCanceledEventArgs), "canceled")]
-[JsonDerivedType(typeof(DownloadCompleteEventArgs), "complete")]
+// https://github.com/dotnet/runtime/issues/72604
+//[JsonPolymorphic(TypeDiscriminatorPropertyName = "status")]
+//[JsonDerivedType(typeof(DownloadCanceledEventArgs), "canceled")]
+//[JsonDerivedType(typeof(DownloadCompleteEventArgs), "complete")]
 public abstract record DownloadEndEventArgs(BiDi BiDi, BrowsingContext Context)
     : BrowsingContextEventArgs(BiDi, Context);
 
-public abstract record DownloadCanceledEventArgs(BiDi BiDi, BrowsingContext Context, Navigation? Navigation, DateTimeOffset Timestamp, string Url)
+public sealed record DownloadCanceledEventArgs(BiDi BiDi, BrowsingContext Context, Navigation? Navigation, DateTimeOffset Timestamp, string Url)
     : DownloadEndEventArgs(BiDi, Context), IBaseNavigationInfo;
 
-public abstract record DownloadCompleteEventArgs(BiDi BiDi, string? Filepath, BrowsingContext Context, Navigation? Navigation, DateTimeOffset Timestamp, string Url)
+public sealed record DownloadCompleteEventArgs(BiDi BiDi, string? Filepath, BrowsingContext Context, Navigation? Navigation, DateTimeOffset Timestamp, string Url)
     : DownloadEndEventArgs(BiDi, Context), IBaseNavigationInfo;

dotnet/src/webdriver/BiDi/Communication/Broker.cs

@@ -97,6 +97,7 @@ internal Broker(BiDi bidi, Uri url)
                 new Json.Converters.Polymorphic.RemoteValueConverter(),
                 new Json.Converters.Polymorphic.RealmInfoConverter(),
                 new Json.Converters.Polymorphic.LogEntryConverter(),
+                new Json.Converters.Polymorphic.DownloadEndEventArgsConverter(),
                 //
 
                 // Enumerable

dotnet/src/webdriver/BiDi/Communication/Json/BiDiJsonSerializerContext.cs

@@ -114,6 +114,8 @@ namespace OpenQA.Selenium.BiDi.Communication.Json;
 [JsonSerializable(typeof(BrowsingContext.BrowsingContextInfo))]
 [JsonSerializable(typeof(BrowsingContext.DownloadWillBeginEventArgs))]
 [JsonSerializable(typeof(BrowsingContext.DownloadEndEventArgs))]
+[JsonSerializable(typeof(BrowsingContext.DownloadCanceledEventArgs))]
+[JsonSerializable(typeof(BrowsingContext.DownloadCompleteEventArgs))]
 [JsonSerializable(typeof(BrowsingContext.HistoryUpdatedEventArgs))]
 [JsonSerializable(typeof(BrowsingContext.NavigationInfo))]
 [JsonSerializable(typeof(BrowsingContext.UserPromptOpenedEventArgs))]

dotnet/src/webdriver/BiDi/Communication/Json/Converters/Polymorphic/DownloadEndEventArgsConverter.cs

@@ -0,0 +1,45 @@
+// <copyright file="DownloadEndEventArgsConverter.cs" company="Selenium Committers">
+// Licensed to the Software Freedom Conservancy (SFC) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The SFC licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+// </copyright>
+
+using OpenQA.Selenium.BiDi.BrowsingContext;
+using OpenQA.Selenium.BiDi.Communication.Json.Internal;
+using System;
+using System.Text.Json;
+using System.Text.Json.Serialization;
+
+namespace OpenQA.Selenium.BiDi.Communication.Json.Converters.Polymorphic;
+
+// https://github.com/dotnet/runtime/issues/72604
+internal class DownloadEndEventArgsConverter : JsonConverter<DownloadEndEventArgs>
+{
+    public override DownloadEndEventArgs? Read(ref Utf8JsonReader reader, Type typeToConvert, JsonSerializerOptions options)
+    {
+        return reader.GetDiscriminator("status") switch
+        {
+            "canceled" => JsonSerializer.Deserialize(ref reader, options.GetTypeInfo<DownloadCanceledEventArgs>()),
+            "complete" => JsonSerializer.Deserialize(ref reader, options.GetTypeInfo<DownloadCompleteEventArgs>()),
+            _ => null,
+        };
+    }
+
+    public override void Write(Utf8JsonWriter writer, DownloadEndEventArgs value, JsonSerializerOptions options)
+    {
+        throw new NotImplementedException();
+    }
+}

dotnet/test/common/BiDi/BrowsingContext/BrowsingContextEventsTest.cs

@@ -0,0 +1,65 @@
+// <copyright file="BrowsingContextEventsTest.cs" company="Selenium Committers">
+// Licensed to the Software Freedom Conservancy (SFC) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The SFC licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+// </copyright>
+
+using NUnit.Framework;
+using System;
+using System.Threading.Tasks;
+
+namespace OpenQA.Selenium.BiDi.BrowsingContext;
+
+class BrowsingContextEventsTest : BiDiTestFixture
+{
+    [Test]
+    public async Task CanListenDownloadWillBeginEvent()
+    {
+        await context.NavigateAsync(UrlBuilder.WhereIs("downloads/download.html"), new() { Wait = ReadinessState.Complete });
+
+        TaskCompletionSource<DownloadWillBeginEventArgs> tcs = new();
+
+        await using var subscription = await context.OnDownloadWillBeginAsync(tcs.SetResult);
+
+        driver.FindElement(By.Id("file-1")).Click();
+
+        var eventArgs = await tcs.Task.WaitAsync(TimeSpan.FromSeconds(5));
+
+        Assert.That(eventArgs, Is.Not.Null);
+        Assert.That(eventArgs.Context, Is.EqualTo(context));
+        Assert.That(eventArgs.Url, Does.EndWith("downloads/file_1.txt"));
+        Assert.That(eventArgs.SuggestedFilename, Is.EqualTo("file_1.txt"));
+    }
+
+    [Test]
+    public async Task CanListenDownloadEndEvent()
+    {
+        await context.NavigateAsync(UrlBuilder.WhereIs("downloads/download.html"), new() { Wait = ReadinessState.Complete });
+
+        TaskCompletionSource<DownloadEndEventArgs> tcs = new();
+
+        await using var subscription = await context.OnDownloadEndAsync(tcs.SetResult);
+
+        driver.FindElement(By.Id("file-1")).Click();
+
+        var eventArgs = await tcs.Task.WaitAsync(TimeSpan.FromSeconds(5));
+
+        Assert.That(eventArgs, Is.Not.Null);
+        Assert.That(eventArgs.Context, Is.EqualTo(context));
+        Assert.That(eventArgs, Is.TypeOf<DownloadCompleteEventArgs>());
+        Assert.That(((DownloadCompleteEventArgs)eventArgs).Filepath, Is.Not.Empty);
+    }
+}