From f6fe818f17e662f27fd5ffdfe14b8644599ec3c2 Mon Sep 17 00:00:00 2001
From: KostiantynTselykh <konstantin.tseluch@openx.com>
Date: Fri, 11 Jun 2021 11:28:19 +0300
Subject: [PATCH] feat(NativeAdUnit): - Appending omid tracker on adunit
 creation

tests:
- Added test to verify event tracker list contains OMID event
---
 .../rendering/bidding/display/NativeAdUnit.java    | 14 +++++++++++++-
 .../bidding/display/NativeAdUnitTest.java          |  6 +++++-
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/PrebidMobile/PrebidMobile-rendering/src/main/java/org/prebid/mobile/rendering/bidding/display/NativeAdUnit.java b/PrebidMobile/PrebidMobile-rendering/src/main/java/org/prebid/mobile/rendering/bidding/display/NativeAdUnit.java
index 9d5947c67..56fdee8cf 100644
--- a/PrebidMobile/PrebidMobile-rendering/src/main/java/org/prebid/mobile/rendering/bidding/display/NativeAdUnit.java
+++ b/PrebidMobile/PrebidMobile-rendering/src/main/java/org/prebid/mobile/rendering/bidding/display/NativeAdUnit.java
@@ -27,8 +27,11 @@
 import org.prebid.mobile.rendering.errors.AdException;
 import org.prebid.mobile.rendering.models.AdConfiguration;
 import org.prebid.mobile.rendering.models.ntv.NativeAdConfiguration;
+import org.prebid.mobile.rendering.models.ntv.NativeEventTracker;
 import org.prebid.mobile.rendering.utils.logger.LogUtil;
 
+import java.util.ArrayList;
+
 import androidx.annotation.NonNull;
 import androidx.annotation.Nullable;
 
@@ -46,7 +49,7 @@ public NativeAdUnit(Context context, String configId,
                         @NonNull
                                 NativeAdConfiguration nativeAdConfiguration) {
         super(context, configId, null);
-        mAdUnitConfig.setNativeAdConfiguration(nativeAdConfiguration);
+        setupNativeAdConfiguration(nativeAdConfiguration);
     }
 
     @Override
@@ -92,4 +95,13 @@ public void fetchDemand(
         mNativeFetchCompleteListener = listener;
         super.fetchDemand(null, mOnFetchCompleteListener);
     }
+
+    private void setupNativeAdConfiguration(NativeAdConfiguration nativeAdConfiguration) {
+        ArrayList<NativeEventTracker.EventTrackingMethod> methodList = new ArrayList<>();
+        methodList.add(NativeEventTracker.EventTrackingMethod.JS);
+        NativeEventTracker tracker = new NativeEventTracker(NativeEventTracker.EventType.OMID,
+                                                            methodList);
+        nativeAdConfiguration.addTracker(tracker);
+        mAdUnitConfig.setNativeAdConfiguration(nativeAdConfiguration);
+    }
 }
diff --git a/PrebidMobile/PrebidMobile-rendering/src/test/java/org/prebid/mobile/rendering/bidding/display/NativeAdUnitTest.java b/PrebidMobile/PrebidMobile-rendering/src/test/java/org/prebid/mobile/rendering/bidding/display/NativeAdUnitTest.java
index d3fa28866..3ae5585ef 100644
--- a/PrebidMobile/PrebidMobile-rendering/src/test/java/org/prebid/mobile/rendering/bidding/display/NativeAdUnitTest.java
+++ b/PrebidMobile/PrebidMobile-rendering/src/test/java/org/prebid/mobile/rendering/bidding/display/NativeAdUnitTest.java
@@ -30,6 +30,7 @@
 import org.prebid.mobile.rendering.errors.AdException;
 import org.prebid.mobile.rendering.models.AdConfiguration;
 import org.prebid.mobile.rendering.models.ntv.NativeAdConfiguration;
+import org.prebid.mobile.rendering.models.ntv.NativeEventTracker;
 import org.prebid.mobile.test.utils.WhiteBox;
 
 import static org.junit.Assert.assertEquals;
@@ -55,7 +56,10 @@ public class NativeAdUnitTest {
     @Before
     public void setup() {
         MockitoAnnotations.initMocks(this);
-        mSpyNativeAdUnit = spy(new NativeAdUnit(mMockContext, "123", new NativeAdConfiguration()));
+        final NativeAdConfiguration nativeAdConfiguration = new NativeAdConfiguration();
+        mSpyNativeAdUnit = spy(new NativeAdUnit(mMockContext, "123", nativeAdConfiguration));
+
+        assertEquals(NativeEventTracker.EventType.OMID, nativeAdConfiguration.getTrackers().get(0).getEventType());
     }
 
     @Test