@@ -21,7 +21,13 @@ class DelegateBridgeTests: XCTestCase {
21
21
var delegateBridge = DelegateBridge ( )
22
22
23
23
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)
25
31
delegateBridge. displayMessage ( inAppMessage, displayDelegate: TestDelegate ( ) )
26
32
27
33
DispatchQueue . main. async {
@@ -36,7 +42,13 @@ class DelegateBridgeTests: XCTestCase {
36
42
}
37
43
38
44
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)
40
52
delegateBridge. displayMessage ( inAppMessage, displayDelegate: TestDelegate ( ) )
41
53
42
54
DispatchQueue . main. async {
@@ -55,14 +67,4 @@ class DelegateBridgeTests: XCTestCase {
55
67
}
56
68
57
69
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
- }
68
70
}
0 commit comments