No public description

PiperOrigin-RevId: 758974266

Author: Mobile Ads Developer Relations

source/android-library/app/src/main/java/com/google/android/ads/nativetemplates/TemplateView.java

@@ -72,6 +72,10 @@ public TemplateView(Context context, AttributeSet attrs, int defStyleAttr, int d
     initView(context, attrs);
   }
 
+  public NativeTemplateStyle getStyles() {
+    return styles;
+  }
+
   public void setStyles(NativeTemplateStyle styles) {
     this.styles = styles;
     this.applyStyles();
@@ -82,10 +86,11 @@ public NativeAdView getNativeAdView() {
   }
 
   private void applyStyles() {
-
     Drawable mainBackground = styles.getMainBackgroundColor();
     if (mainBackground != null) {
-      background.setBackground(mainBackground);
+      if (background != null) {
+        background.setBackground(mainBackground);
+      }
       if (primaryView != null) {
         primaryView.setBackground(mainBackground);
       }

source/android-library/app/src/main/java/com/google/unity/ads/nativead/UnityNativeTemplateStyle.java

@@ -21,6 +21,7 @@
 import android.view.LayoutInflater;
 import androidx.annotation.NonNull;
 import androidx.annotation.Nullable;
+import androidx.annotation.VisibleForTesting;
 import com.google.android.ads.nativetemplates.NativeTemplateStyle;
 import com.google.android.ads.nativetemplates.TemplateView;
 import java.util.Objects;
@@ -35,6 +36,8 @@ public final class UnityNativeTemplateStyle {
   @Nullable final UnityNativeTemplateTextStyle secondaryTextStyle;
   @Nullable final UnityNativeTemplateTextStyle tertiaryTextStyle;
 
+  private LayoutInflater layoutInflater;
+
   public UnityNativeTemplateStyle(
       @NonNull UnityNativeTemplateType templateType,
       @Nullable ColorDrawable mainBackgroundColor,
@@ -50,10 +53,21 @@ public UnityNativeTemplateStyle(
     this.tertiaryTextStyle = tertiaryTextStyle;
   }
 
+  private void setLayoutInflater(Context context) {
+    this.layoutInflater =
+        (LayoutInflater) context.getSystemService(Context.LAYOUT_INFLATER_SERVICE);
+  }
+
+  @VisibleForTesting(otherwise = VisibleForTesting.NONE)
+  public void setLayoutInflater(LayoutInflater layoutInflater) {
+    this.layoutInflater = layoutInflater;
+  }
+
   @Nullable
   public TemplateView asTemplateView(Context context) {
-    LayoutInflater layoutInflater =
-        (LayoutInflater) context.getSystemService(Context.LAYOUT_INFLATER_SERVICE);
+    if (layoutInflater == null) {
+      setLayoutInflater(context);
+    }
     TemplateView templateView =
         (TemplateView) layoutInflater.inflate(templateType.resourceId(), null);
     if (templateView == null) {

source/android-library/app/src/test/java/com/google/unity/ads/nativead/UnityNativeTemplateStyleTest.java

@@ -1,19 +1,35 @@
 package com.google.unity.ads.nativead;
 
 import static com.google.common.truth.Truth.assertThat;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.anyInt;
+import static org.mockito.Mockito.when;
 
+import android.content.Context;
 import android.graphics.Color;
 import android.graphics.drawable.ColorDrawable;
+import android.view.LayoutInflater;
+import androidx.test.core.app.ApplicationProvider;
+import com.google.android.ads.nativetemplates.NativeTemplateStyle;
+import com.google.android.ads.nativetemplates.TemplateView;
 import org.junit.Before;
+import org.junit.Rule;
 import org.junit.Test;
 import org.junit.runner.RunWith;
+import org.mockito.Mock;
+import org.mockito.junit.MockitoJUnit;
+import org.mockito.junit.MockitoRule;
 import org.robolectric.RobolectricTestRunner;
 
 /** Tests for {@link UnityNativeTemplateStyle} */
 @RunWith(RobolectricTestRunner.class)
 public final class UnityNativeTemplateStyleTest {
 
+  @Rule public final MockitoRule mockito = MockitoJUnit.rule();
+  @Mock private LayoutInflater mockLayoutInflater;
+
   private UnityNativeTemplateStyle unityNativeTemplateStyle;
+  private Context context;
 
   @Before
   public void setUp() {
@@ -41,6 +57,7 @@ public void setUp() {
                 /* backgroundColor= */ new ColorDrawable(Color.BLUE),
                 /* fontStyle= */ UnityNativeTemplateFontStyle.ITALIC,
                 /* size= */ 10.0));
+    context = ApplicationProvider.getApplicationContext();
   }
 
   @Test
@@ -107,4 +124,37 @@ public void unityNativeTemplateStyle_canBeCompared() throws Exception {
     assertThat(unityNativeTemplateStyle.hashCode())
         .isNotEqualTo(anotherUnityNativeTemplateStyle.hashCode());
   }
+
+  @Test
+  public void unityNativeTemplateStyle_asTemplateView_canBeInflated() throws Exception {
+    // Arrange.
+    TemplateView fakeTemplateView = new TemplateView(context);
+    assertThat(fakeTemplateView.getStyles()).isNull();
+    when(mockLayoutInflater.inflate(anyInt(), any())).thenReturn(fakeTemplateView);
+
+    // Act.
+    unityNativeTemplateStyle.setLayoutInflater(mockLayoutInflater);
+    fakeTemplateView = unityNativeTemplateStyle.asTemplateView(context);
+    NativeTemplateStyle styles = fakeTemplateView.getStyles();
+
+    // Assert (styles should have been applied).
+    assertThat(styles).isNotNull();
+    assertThat(styles.getMainBackgroundColor().getColor()).isEqualTo(Color.RED);
+    assertThat(styles.getCallToActionBackgroundColor().getColor()).isEqualTo(Color.BLUE);
+    assertThat(styles.getCallToActionTypefaceColor()).isEqualTo(Color.GREEN);
+    assertThat(styles.getCallToActionTextTypeface()).isNotNull();
+    assertThat(styles.getCallToActionTextSize()).isEqualTo(1.0f);
+    assertThat(styles.getPrimaryTextBackgroundColor().getColor()).isEqualTo(Color.BLUE);
+    assertThat(styles.getPrimaryTextTypefaceColor()).isEqualTo(Color.RED);
+    assertThat(styles.getPrimaryTextTypeface()).isNotNull();
+    assertThat(styles.getPrimaryTextSize()).isEqualTo(5.0f);
+    assertThat(styles.getSecondaryTextBackgroundColor().getColor()).isEqualTo(Color.BLUE);
+    assertThat(styles.getSecondaryTextTypefaceColor()).isEqualTo(Color.RED);
+    assertThat(styles.getSecondaryTextTypeface()).isNotNull();
+    assertThat(styles.getSecondaryTextSize()).isEqualTo(20.0f);
+    assertThat(styles.getTertiaryTextBackgroundColor().getColor()).isEqualTo(Color.BLUE);
+    assertThat(styles.getTertiaryTextTypefaceColor()).isEqualTo(Color.RED);
+    assertThat(styles.getTertiaryTextTypeface()).isNotNull();
+    assertThat(styles.getTertiaryTextSize()).isEqualTo(10.0f);
+  }
 }