Add setup (#49)

Author: noahsmartin

DemoApp/DemoApp.xcodeproj/project.pbxproj

@@ -36,6 +36,7 @@
 		FA4F5ADA2A7A3E7700B268FF /* Snapshotting in Embed Frameworks */ = {isa = PBXBuildFile; productRef = FA4F5AD82A7A3E7700B268FF /* Snapshotting */; settings = {ATTRIBUTES = (CodeSignOnCopy, ); }; };
 		FA5E82972A96448A008DE3F0 /* StateView.swift in Sources */ = {isa = PBXBuildFile; fileRef = FA5E82962A96448A008DE3F0 /* StateView.swift */; };
 		FA9C8E442A56959300DC4574 /* RatingPreviews.swift in Sources */ = {isa = PBXBuildFile; fileRef = FA9C8E432A56959300DC4574 /* RatingPreviews.swift */; };
+		FABCCD932AC39081008F4D3A /* EMGTestHandler.swift in Sources */ = {isa = PBXBuildFile; fileRef = FABCCD922AC39081008F4D3A /* EMGTestHandler.swift */; };
 		FACA23792A55FBEE0080545A /* DemoModule.framework in Frameworks */ = {isa = PBXBuildFile; fileRef = FA1671D72A5399F700A42DB0 /* DemoModule.framework */; };
 		FAD0107B2AA29ABA007D1AF6 /* TextView.swift in Sources */ = {isa = PBXBuildFile; fileRef = FAD0107A2AA29ABA007D1AF6 /* TextView.swift */; };
 		FAD52BF62A7B5EEA001F1832 /* ExpandingView.swift in Sources */ = {isa = PBXBuildFile; fileRef = FAD52BF12A7B5EEA001F1832 /* ExpandingView.swift */; };
@@ -116,6 +117,7 @@
 		FA44246E2AAA5B27005BFCD9 /* DemoApp.xctestplan */ = {isa = PBXFileReference; lastKnownFileType = text; path = DemoApp.xctestplan; sourceTree = "<group>"; };
 		FA5E82962A96448A008DE3F0 /* StateView.swift */ = {isa = PBXFileReference; fileEncoding = 4; lastKnownFileType = sourcecode.swift; path = StateView.swift; sourceTree = "<group>"; };
 		FA9C8E432A56959300DC4574 /* RatingPreviews.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = RatingPreviews.swift; sourceTree = "<group>"; };
+		FABCCD922AC39081008F4D3A /* EMGTestHandler.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = EMGTestHandler.swift; sourceTree = "<group>"; };
 		FAC167E72A5F41C500D79C23 /* DemoAppUITests.xctest */ = {isa = PBXFileReference; explicitFileType = wrapper.cfbundle; includeInIndex = 0; path = DemoAppUITests.xctest; sourceTree = BUILT_PRODUCTS_DIR; };
 		FACA23752A55FB970080545A /* XCTest.framework */ = {isa = PBXFileReference; lastKnownFileType = wrapper.framework; name = XCTest.framework; path = Platforms/MacOSX.platform/Developer/Library/Frameworks/XCTest.framework; sourceTree = DEVELOPER_DIR; };
 		FAD0107A2AA29ABA007D1AF6 /* TextView.swift */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.swift; path = TextView.swift; sourceTree = "<group>"; };
@@ -183,6 +185,7 @@
 			children = (
 				FAF1BA622A54934200AEBE1B /* .. */,
 				FA1671C22A5367A800A42DB0 /* DemoAppApp.swift */,
+				FABCCD922AC39081008F4D3A /* EMGTestHandler.swift */,
 				FA1671C42A5367A800A42DB0 /* ContentView.swift */,
 				FAD52BF02A7B5EBB001F1832 /* TestViews */,
 				FA1671C62A5367A800A42DB0 /* Assets.xcassets */,
@@ -418,6 +421,7 @@
 				FA40515D2A95B587007A66D4 /* EmptyView.swift in Sources */,
 				FAD52BFA2A7B5EEA001F1832 /* RideOptionsView.swift in Sources */,
 				FA5E82972A96448A008DE3F0 /* StateView.swift in Sources */,
+				FABCCD932AC39081008F4D3A /* EMGTestHandler.swift in Sources */,
 				FA0B20F12A7DA20200EC91D3 /* AnyViewPreview.swift in Sources */,
 				FA0B20FB2A7DCB0100EC91D3 /* PreviewVariants.swift in Sources */,
 				FAD52BF72A7B5EEA001F1832 /* CodeEntryView.swift in Sources */,

DemoApp/DemoApp/EMGTestHandler.swift

@@ -0,0 +1,16 @@
+//
+//  EMGTestHandler.swift
+//  DemoApp
+//
+//  Created by Noah Martin on 9/26/23.
+//
+
+import Foundation
+
+@objc(EMGTestHandler)
+class EMGTestHandler: NSObject {
+  @objc
+  static func setup() {
+    print("Setup called.")
+  }
+}

Sources/SnapshottingSwift/Snapshots.swift

@@ -26,6 +26,7 @@ extension SnapshotError: LocalizedError {
 
 class Snapshots {
   let server = HTTPServer(port: 8080)
+  let testHandler: NSObject.Type? = NSClassFromString("EMGTestHandler") as? NSObject.Type
 
   public init() {
     let windowScene = UIApplication.shared
@@ -66,6 +67,7 @@ class Snapshots {
       let id = pathComponents[3]
 
       var result: [String: String] = [:]
+      testHandler?.perform(NSSelectorFromString("setup"))
       let (imageResult, preview) = await display(typeName: typeName, id: id)
 
       if let displayName = preview.displayName {