Add Test.

Author: shogo4405

Examples/macOS/PublishViewModel.swift

@@ -98,13 +98,25 @@ final class PublishViewModel: ObservableObject {
             await makeSession(preference)
         }
         Task { @ScreenActor in
+            await mixer.screen.size = .init(width: 1280, height: 720)
+            await mixer.screen.backgroundColor = NSColor.black.cgColor
+
             let assetScreenObject = AssetScreenObject()
             assetScreenObject.size = .init(width: 180, height: 180)
             assetScreenObject.layoutMargin = .init(top: 16, left: 16, bottom: 0, right: 0)
             try? assetScreenObject.startReading(AVAsset(url: URL(fileURLWithPath: Bundle.main.path(forResource: "SampleVideo_360x240_5mb", ofType: "mp4") ?? "")))
             try? await mixer.screen.addChild(assetScreenObject)
-            await mixer.screen.size = .init(width: 1280, height: 720)
-            await mixer.screen.backgroundColor = NSColor.black.cgColor
+
+            let image = ImageScreenObject()
+            image.size = .init(width: 120, height: 120)
+            image.horizontalAlignment = .right
+            image.verticalAlignment = .bottom
+            image.layoutMargin = .init(top: 0, left: 0, bottom: 16, right: 16)
+            let nsImage = await NSApplication.shared.applicationIconImage
+            if let cgImage = nsImage?.cgImage(forProposedRect: nil, context: nil, hints: nil) {
+                image.ciImage = CIImage(cgImage: cgImage)
+            }
+            try? await mixer.screen.addChild(image)
         }
     }
 

HaishinKit/Sources/Screen/ImageScreenObject.swift

@@ -127,7 +127,7 @@ public final class ImageScreenObject: ScreenObject {
         return super.makeBounds(size == .zero ? ciImage.extent.size : size)
     }
 
-    func setSource(_ source: String?) throws {
+    public func setSource(_ source: String?) throws {
         self.source = source
         let imageSource = try ImageSourceFactory.parse(URL(string: source ?? ""))
         ciImage = try imageSource.toImage()

HaishinKit/Sources/Screen/Screen.swift

@@ -91,7 +91,7 @@ public final class Screen: ScreenObjectContainerConvertible {
     private(set) var targetTimestamp: TimeInterval = 0.0
     private(set) var videoTrackScreenObject = VideoScreenObject()
     private var videoCaptureLatency: TimeInterval = 0.0
-    private var root: ScreenObjectContainer = .init()
+    private(set) var root: ScreenObjectContainer = .init()
     private var outputFormat: CMFormatDescription?
     private var pixelBufferPool: CVPixelBufferPool? {
         didSet {

HaishinKit/Sources/Screen/ScreenObjectSnapshotFactory.swift

@@ -41,4 +41,8 @@ public struct ScreenObjectSnapshotFactory {
             )
         }
     }
+
+    public func make(_ screen: Screen)-> ScreenObjectSnapshot {
+        return make(screen.root)
+    }
 }

HaishinKit/Tests/Screen/ScreenObjectContainerTests.swift

@@ -15,15 +15,15 @@ import Testing
         try? container1.addChild(videoTrack1)
         try? container1.addChild(videoTrack2)
 
-        let videoTracks1 = container1.getScreenObjects() as [VideoTrackScreenObject]
+        let videoTracks1 = container1.getScreenObjects() as [VideoScreenObject]
         #expect(videoTracks1.count == 2)
 
         let container2 = ScreenObjectContainer()
         let videoTrack3 = VideoScreenObject()
         try? container2.addChild(videoTrack3)
         try? container1.addChild(container2)
 
-        let videoTracks2 = container1.getScreenObjects() as [VideoTrackScreenObject]
+        let videoTracks2 = container1.getScreenObjects() as [VideoScreenObject]
         #expect(videoTracks2.count == 3)
     }
 }

HaishinKit/Tests/Screen/ScreenObjectFactoryTests.swift

@@ -0,0 +1,21 @@
+import AVFoundation
+import Foundation
+import Testing
+
+@testable import HaishinKit
+
+@ScreenActor
+@Suite struct ScreenObjectFactoryTests {
+    @Test func videoSnapshot() throws {
+        let string = """
+{"type":"video","id":"1771162233189358-1238207813","size":{"width":90,"height":160},"isVisible":true,"layoutMargin":{"top":16,"left":0,"bottom":0,"right":16},"horizontalAlignment":2,"verticalAlignment":0,"elements":{"track":"1"},"children":[]}
+""".data(using: .utf8)!
+        let snapshot = try JSONDecoder().decode(ScreenObjectSnapshot.self, from: string)
+        let factory = ScreenObjectFactory()
+        let videoScreenObject = factory.make(snapshot) as? VideoScreenObject
+        #expect(videoScreenObject?.track == 1)
+        #expect(videoScreenObject?.horizontalAlignment == .right)
+        #expect(videoScreenObject?.verticalAlignment == .top)
+        #expect(videoScreenObject?.size == .init(width: 90, height: 160))
+    }
+}

HaishinKit/Tests/Screen/ScreenObjectSnapshotTests.swift

@@ -0,0 +1,18 @@
+import AVFoundation
+import Foundation
+import Testing
+
+@testable import HaishinKit
+
+@ScreenActor
+@Suite struct ScreenObjectSnapshotTests {
+    @Test func videoSnapshot() throws {
+        let string = """
+{"type":"video","id":"1771162233189358-1238207813","size":{"width":90,"height":160},"isVisible":true,"layoutMargin":{"top":16,"left":0,"bottom":0,"right":16},"horizontalAlignment":2,"verticalAlignment":0,"elements":{"track":"1"},"children":[]}
+""".data(using: .utf8)!
+        let snapshot = try JSONDecoder().decode(ScreenObjectSnapshot.self, from: string)
+        #expect(snapshot.type == "video")
+        #expect(snapshot.horizontalAlignment == 2)
+        #expect(snapshot.verticalAlignment == 0)
+    }
+}