Skip to content

Commit 0eec2bb

Browse files
committed
Let's try direct instantiation of messages in unit tests
1 parent 34ba6ac commit 0eec2bb

File tree

1 file changed

+14
-12
lines changed

1 file changed

+14
-12
lines changed

FirebaseInAppMessaging/Swift/Tests/Unit/CustomInAppMessageDisplayViewModifierTests.swift

Lines changed: 14 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,13 @@ class DelegateBridgeTests: XCTestCase {
2121
var delegateBridge = DelegateBridge()
2222

2323
func testClearsInAppMessageOnDismiss() {
24-
let inAppMessage = MockFIAM(messageID: "messageID")
24+
let inAppMessage = InAppMessagingDisplayMessage(messageID: "messageID",
25+
campaignName: "testCampaign",
26+
renderAsTestMessage: false,
27+
messageType: FIRInAppMessagingDisplayMessageType
28+
.card,
29+
triggerType: FIRInAppMessagingDisplayTriggerType
30+
.onAppForeground)
2531
delegateBridge.displayMessage(inAppMessage, displayDelegate: TestDelegate())
2632

2733
DispatchQueue.main.async {
@@ -36,7 +42,13 @@ class DelegateBridgeTests: XCTestCase {
3642
}
3743

3844
func testClearsInAppMessageOnClick() {
39-
let inAppMessage = MockFIAM(messageID: "messageID")
45+
let inAppMessage = InAppMessagingDisplayMessage(messageID: "messageID",
46+
campaignName: "testCampaign",
47+
renderAsTestMessage: false,
48+
messageType: FIRInAppMessagingDisplayMessageType
49+
.card,
50+
triggerType: FIRInAppMessagingDisplayTriggerType
51+
.onAppForeground)
4052
delegateBridge.displayMessage(inAppMessage, displayDelegate: TestDelegate())
4153

4254
DispatchQueue.main.async {
@@ -55,14 +67,4 @@ class DelegateBridgeTests: XCTestCase {
5567
}
5668

5769
class TestDelegate: NSObject, InAppMessagingDisplayDelegate {}
58-
59-
class MockFIAM: InAppMessagingDisplayMessage {
60-
init(messageID: String) {
61-
super.init(messageID: messageID,
62-
campaignName: "testCampaign",
63-
renderAsTestMessage: false,
64-
messageType: FIRInAppMessagingDisplayMessageType.card,
65-
triggerType: FIRInAppMessagingDisplayTriggerType.onAppForeground)
66-
}
67-
}
6870
}

0 commit comments

Comments
 (0)