diff --git a/firebase-inappmessaging/src/androidTest/java/com/google/firebase/inappmessaging/FirebaseInAppMessagingFlowableTest.java b/firebase-inappmessaging/src/androidTest/java/com/google/firebase/inappmessaging/FirebaseInAppMessagingFlowableTest.java index 6d22c28532c..18b78b7f903 100644 --- a/firebase-inappmessaging/src/androidTest/java/com/google/firebase/inappmessaging/FirebaseInAppMessagingFlowableTest.java +++ b/firebase-inappmessaging/src/androidTest/java/com/google/firebase/inappmessaging/FirebaseInAppMessagingFlowableTest.java @@ -58,6 +58,7 @@ import com.google.firebase.inappmessaging.CommonTypesProto.TriggeringCondition; import com.google.firebase.inappmessaging.FirebaseInAppMessagingDisplayCallbacks.InAppMessagingDismissType; import com.google.firebase.inappmessaging.FirebaseInAppMessagingDisplayCallbacks.InAppMessagingErrorReason; +import com.google.firebase.inappmessaging.internal.AbtIntegrationHelper; import com.google.firebase.inappmessaging.internal.DisplayCallbacksFactory; import com.google.firebase.inappmessaging.internal.MetricsLoggerClient; import com.google.firebase.inappmessaging.internal.ProgramaticContextualTriggers; @@ -167,6 +168,8 @@ public class FirebaseInAppMessagingFlowableTest { @Mock private FirebaseInstanceId instanceId; @Mock private TestDeviceHelper testDeviceHelper; @Mock private Subscriber firebaseEventSubscriber; + @Mock private AbtIntegrationHelper abtIntegrationHelper; + private TestSubscriber subscriber; private FirebaseInAppMessaging instance; private TestForegroundNotifier foregroundNotifier; @@ -246,6 +249,7 @@ public void setUp() { appComponentBuilder = DaggerTestAppComponent.builder() .universalComponent(universalComponent) + .testAbTestingModule(new TestAbTestingModule(abtIntegrationHelper)) .testEngagementMetricsLoggerClientModule( new TestEngagementMetricsLoggerClientModule(app, engagementMetricsLoggerInterface)) .grpcClientModule(new GrpcClientModule(app)) diff --git a/firebase-inappmessaging/src/androidTest/java/com/google/firebase/inappmessaging/TestAbTestingModule.java b/firebase-inappmessaging/src/androidTest/java/com/google/firebase/inappmessaging/TestAbTestingModule.java new file mode 100644 index 00000000000..01e6328c7dd --- /dev/null +++ b/firebase-inappmessaging/src/androidTest/java/com/google/firebase/inappmessaging/TestAbTestingModule.java @@ -0,0 +1,34 @@ +// Copyright 2020 Google LLC +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +package com.google.firebase.inappmessaging; + +import com.google.firebase.inappmessaging.internal.AbtIntegrationHelper; +import dagger.Module; +import dagger.Provides; + +/** Test Provider for ABTesting Module */ +@Module +class TestAbTestingModule { + private final AbtIntegrationHelper abtIntegrationHelper; + + TestAbTestingModule(AbtIntegrationHelper abtIntegrationHelper) { + this.abtIntegrationHelper = abtIntegrationHelper; + } + + @Provides + AbtIntegrationHelper providesAbtIntegrationHelper() { + return this.abtIntegrationHelper; + } +} diff --git a/firebase-inappmessaging/src/androidTest/java/com/google/firebase/inappmessaging/TestAppComponent.java b/firebase-inappmessaging/src/androidTest/java/com/google/firebase/inappmessaging/TestAppComponent.java index 0763fa85675..d01c1099dd6 100644 --- a/firebase-inappmessaging/src/androidTest/java/com/google/firebase/inappmessaging/TestAppComponent.java +++ b/firebase-inappmessaging/src/androidTest/java/com/google/firebase/inappmessaging/TestAppComponent.java @@ -27,6 +27,7 @@ // Test modules TestApiClientModule.class, TestEngagementMetricsLoggerClientModule.class, + TestAbTestingModule.class, // Real modules GrpcClientModule.class