1 files changed, 101 insertions, 0 deletions
diff --git a/Tests/LibMatrix.Tests/Tests/RoomTests/SpaceTests.cs b/Tests/LibMatrix.Tests/Tests/RoomTests/SpaceTests.cs
new file mode 100644
index 0000000..148b5fe
--- /dev/null
+++ b/Tests/LibMatrix.Tests/Tests/RoomTests/SpaceTests.cs
@@ -0,0 +1,101 @@
+using System.Diagnostics;
+using System.Text;
+using ArcaneLibs.Extensions;
+using LibMatrix.EventTypes.Spec.State;
+using LibMatrix.Homeservers;
+using LibMatrix.Responses;
+using LibMatrix.RoomTypes;
+using LibMatrix.Services;
+using LibMatrix.Tests.Abstractions;
+using LibMatrix.Tests.Fixtures;
+using Xunit.Abstractions;
+using Xunit.Microsoft.DependencyInjection.Abstracts;
+
+namespace LibMatrix.Tests.Tests;
+
+public class SpaceTests : TestBed<TestFixture> {
+    private readonly HomeserverAbstraction _hsAbstraction;
+
+    public SpaceTests(ITestOutputHelper testOutputHelper, TestFixture fixture) : base(testOutputHelper, fixture) {
+        _hsAbstraction = _fixture.GetService<HomeserverAbstraction>(_testOutputHelper) ?? throw new InvalidOperationException($"Failed to get {nameof(HomeserverAbstraction)}");
+    }
+
+    [Fact]
+    public async Task AddChildAsync() {
+        var hs = await _hsAbstraction.GetConfiguredHomeserver();
+        var crq = new CreateRoomRequest() {
+            Name = "Test space"
+        };
+        crq.CreationContent["type"] = SpaceRoom.TypeName;
+        var space = (await hs.CreateRoom(crq)).AsSpace;
+
+        var child = await hs.CreateRoom(new CreateRoomRequest() {
+            Name = "Test child"
+        });
+
+        await space.AddChildAsync(child);
+
+        //validate children
+        var children = space.GetChildrenAsync().ToBlockingEnumerable().ToList();
+        Assert.NotNull(children);
+        Assert.NotEmpty(children);
+        Assert.Single(children, x => x.RoomId == child.RoomId);
+    }
+
+    [Fact]
+    public async Task AddChildByIdAsync() {
+        var hs = await _hsAbstraction.GetConfiguredHomeserver();
+        var crq = new CreateRoomRequest() {
+            Name = "Test space"
+        };
+        crq.CreationContent["type"] = SpaceRoom.TypeName;
+        var space = (await hs.CreateRoom(crq)).AsSpace;
+
+        var child = await hs.CreateRoom(new CreateRoomRequest() {
+            Name = "Test child"
+        });
+
+        await space.AddChildByIdAsync(child.RoomId);
+        
+        //validate children
+        var children = space.GetChildrenAsync().ToBlockingEnumerable().ToList();
+        Assert.NotNull(children);
+        Assert.NotEmpty(children);
+        Assert.Single(children, x => x.RoomId == child.RoomId);
+    }
+    
+    [Fact]
+    public async Task GetChildrenAsync() {
+        var hs = await _hsAbstraction.GetConfiguredHomeserver();
+        var expectedChildren = Enumerable.Range(0, 10).Select(async _ => {
+            var room = await hs.CreateRoom(new CreateRoomRequest() {
+                Name = "Test child"
+            });
+            return room;
+        }).ToAsyncEnumerable().ToBlockingEnumerable().ToList();
+        
+        var crq = new CreateRoomRequest() {
+            Name = "Test space",
+            InitialState = expectedChildren.Select(c => new StateEvent() {
+                Type = "m.space.child",
+                StateKey = c.RoomId,
+                TypedContent = new SpaceChildEventContent() {
+                    Via = new List<string> {
+                        c.RoomId.Split(":")[1]
+                    }
+                }
+            }).ToList()
+        };
+        crq.CreationContent["type"] = SpaceRoom.TypeName;
+        var space = (await hs.CreateRoom(crq)).AsSpace;
+
+        var children = space.GetChildrenAsync().ToBlockingEnumerable().ToList();
+        Assert.NotNull(children);
+        Assert.NotEmpty(children);
+        Assert.Equal(expectedChildren.Count, children.Count);
+        foreach (var expectedChild in expectedChildren)
+        {
+            Assert.Single(children, x => x.RoomId == expectedChild.RoomId);
+        }
+    }
+}
\ No newline at end of file