/*
 * Copyright (C) 2015 The Android Open Source Project
 *
 * 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.android.setupwizardlib.test;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertSame;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;

import android.content.Context;
import android.graphics.drawable.Drawable;
import android.graphics.drawable.InsetDrawable;
import android.os.Build;
import androidx.recyclerview.widget.RecyclerView;
import androidx.recyclerview.widget.RecyclerView.Adapter;
import androidx.recyclerview.widget.RecyclerView.ViewHolder;
import android.view.ContextThemeWrapper;
import android.view.LayoutInflater;
import android.view.View;
import android.view.View.MeasureSpec;
import android.view.ViewGroup;
import android.support.test.InstrumentationRegistry;
import android.support.test.filters.SmallTest;
import android.support.test.runner.AndroidJUnit4;
import com.android.setupwizardlib.SetupWizardRecyclerLayout;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;

@RunWith(AndroidJUnit4.class)
@SmallTest
public class SetupWizardRecyclerLayoutTest {

  private Context mContext;

  @Before
  public void setUp() throws Exception {
    mContext =
        new ContextThemeWrapper(
            InstrumentationRegistry.getContext(), R.style.SuwThemeMaterial_Light);
  }

  @Test
  public void testDefaultTemplate() {
    SetupWizardRecyclerLayout layout = new SetupWizardRecyclerLayout(mContext);
    assertRecyclerTemplateInflated(layout);
  }

  @Test
  public void testInflateFromXml() {
    LayoutInflater inflater = LayoutInflater.from(mContext);
    SetupWizardRecyclerLayout layout =
        (SetupWizardRecyclerLayout) inflater.inflate(R.layout.test_recycler_layout, null);
    assertRecyclerTemplateInflated(layout);
  }

  @Test
  public void testGetRecyclerView() {
    SetupWizardRecyclerLayout layout = new SetupWizardRecyclerLayout(mContext);
    assertRecyclerTemplateInflated(layout);
    assertNotNull("getRecyclerView should not be null", layout.getRecyclerView());
  }

  @Test
  public void testAdapter() {
    SetupWizardRecyclerLayout layout = new SetupWizardRecyclerLayout(mContext);
    assertRecyclerTemplateInflated(layout);

    final Adapter adapter = createTestAdapter(1);
    layout.setAdapter(adapter);

    final Adapter gotAdapter = layout.getAdapter();
    // Note: The wrapped adapter should be returned, not the HeaderAdapter.
    assertSame("Adapter got from SetupWizardLayout should be same as set", adapter, gotAdapter);
  }

  @Test
  public void testLayout() {
    SetupWizardRecyclerLayout layout = new SetupWizardRecyclerLayout(mContext);
    assertRecyclerTemplateInflated(layout);

    layout.setAdapter(createTestAdapter(3));

    layout.measure(
        MeasureSpec.makeMeasureSpec(500, MeasureSpec.EXACTLY),
        MeasureSpec.makeMeasureSpec(500, MeasureSpec.EXACTLY));
    layout.layout(0, 0, 500, 500);
    // Test that the layout code doesn't crash.
  }

  @Test
  public void testDividerInsetLegacy() {
    SetupWizardRecyclerLayout layout = new SetupWizardRecyclerLayout(mContext);
    if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.KITKAT) {
      layout.setLayoutDirection(View.LAYOUT_DIRECTION_LTR);
    }
    assertRecyclerTemplateInflated(layout);

    layout.setDividerInset(10);
    assertEquals("Divider inset should be 10", 10, layout.getDividerInset());

    final Drawable divider = layout.getDivider();
    assertTrue("Divider should be instance of InsetDrawable", divider instanceof InsetDrawable);
  }

  @Test
  public void testDividerInsets() {
    SetupWizardRecyclerLayout layout = new SetupWizardRecyclerLayout(mContext);
    if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.KITKAT) {
      layout.setLayoutDirection(View.LAYOUT_DIRECTION_LTR);
    }
    assertRecyclerTemplateInflated(layout);

    layout.setDividerInsets(10, 15);
    assertEquals("Divider inset start should be 10", 10, layout.getDividerInsetStart());
    assertEquals("Divider inset end should be 15", 15, layout.getDividerInsetEnd());

    final Drawable divider = layout.getDivider();
    assertTrue("Divider should be instance of InsetDrawable", divider instanceof InsetDrawable);
  }

  @Test
  public void testTemplateWithNoRecyclerView() {
    try {
      new SetupWizardRecyclerLayout(mContext, R.layout.suw_glif_template, R.id.suw_recycler_view);
      fail("Creating SetupWizardRecyclerLayout with no recycler view should throw exception");
    } catch (Exception e) {
      // pass
    }
  }

  private void assertRecyclerTemplateInflated(SetupWizardRecyclerLayout layout) {
    View recyclerView = layout.findViewById(R.id.suw_recycler_view);
    assertTrue(
        "@id/suw_recycler_view should be a RecyclerView", recyclerView instanceof RecyclerView);

    assertNotNull(
        "Header text view should not be null", layout.findManagedViewById(R.id.suw_layout_title));
    assertNotNull(
        "Decoration view should not be null", layout.findManagedViewById(R.id.suw_layout_decor));
  }

  private Adapter createTestAdapter(final int itemCount) {
    return new Adapter() {
      @Override
      public ViewHolder onCreateViewHolder(ViewGroup parent, int position) {
        return new ViewHolder(new View(parent.getContext())) {};
      }

      @Override
      public void onBindViewHolder(ViewHolder viewHolder, int position) {}

      @Override
      public int getItemCount() {
        return itemCount;
      }
    };
  }
}