Unit tests, small refactors

1 files changed, 73 insertions, 0 deletions

diff --git a/Tests/LibMatrix.Tests/Abstractions/HomeserverAbstraction.cs b/Tests/LibMatrix.Tests/Abstractions/HomeserverAbstraction.cs
new file mode 100644
index 0000000..abd3e99
--- /dev/null
+++ b/Tests/LibMatrix.Tests/Abstractions/HomeserverAbstraction.cs
@@ -0,0 +1,73 @@
+using ArcaneLibs.Extensions;
+using LibMatrix.Homeservers;
+using LibMatrix.Responses;
+
+namespace LibMatrix.Tests.Abstractions;
+
+public static class HomeserverAbstraction {
+    public static async Task<AuthenticatedHomeserverGeneric> GetHomeserver() {
+        var rhs = new RemoteHomeServer("https://matrixunittests.rory.gay");
+        // string username = Guid.NewGuid().ToString();
+        // string password = Guid.NewGuid().ToString();
+        string username = "@f1a2d2d6-1924-421b-91d0-893b347b2a49:matrixunittests.rory.gay";
+        string password = "d6d782d6-8bc9-4fac-9cd8-78e101b4298b";
+        LoginResponse reg;
+        try {
+            reg = await rhs.LoginAsync(username, password);
+        }
+        catch (MatrixException e) {
+            if (e.ErrorCode == "M_FORBIDDEN") {
+                await rhs.RegisterAsync(Guid.NewGuid().ToString(), Guid.NewGuid().ToString(), "Unit tests!");
+                reg = await rhs.RegisterAsync(username, password, "Unit tests!");
+            }
+            else throw new Exception("Failed to register", e);
+        }
+
+        var hs = await reg.GetAuthenticatedHomeserver("https://matrixunittests.rory.gay");
+
+        var rooms = await hs.GetJoinedRooms();
+
+        var disbandRoomTasks = rooms.Select(async room => {
+            // await room.DisbandRoomAsync();
+            await room.LeaveAsync();
+            await room.ForgetAsync();
+            return room;
+        }).ToList();
+        await Task.WhenAll(disbandRoomTasks);
+
+        // foreach (var room in rooms) {
+        //     // await room.DisbandRoomAsync();
+        //     await room.LeaveAsync();
+        //     await room.ForgetAsync();
+        // }
+
+        return hs;
+    }
+
+    public static async Task<AuthenticatedHomeserverGeneric> GetRandomHomeserver() {
+        var rhs = new RemoteHomeServer("https://matrixunittests.rory.gay");
+        LoginResponse reg = await rhs.RegisterAsync(Guid.NewGuid().ToString(), Guid.NewGuid().ToString(), "Unit tests!");
+        var hs = await reg.GetAuthenticatedHomeserver("https://matrixunittests.rory.gay");
+
+        var rooms = await hs.GetJoinedRooms();
+
+        var disbandRoomTasks = rooms.Select(async room => {
+            // await room.DisbandRoomAsync();
+            await room.LeaveAsync();
+            await room.ForgetAsync();
+            return room;
+        }).ToList();
+        await Task.WhenAll(disbandRoomTasks);
+
+        return hs;
+    }
+
+    public static async IAsyncEnumerable<AuthenticatedHomeserverGeneric> GetRandomHomeservers(int count = 1) {
+        var createSpaceTasks = Enumerable
+            .Range(0, count)
+            .Select(_ => GetRandomHomeserver()).ToAsyncEnumerable();
+        await foreach (var hs in createSpaceTasks) {
+            yield return hs;
+        }
+    }
+}