src/main/java/com/google/android/libraries/feed/testing/modelprovider/FakeModelProviderFactory.java

// Copyright 2018 The Feed Authors.
//
// 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.android.libraries.feed.testing.modelprovider;

import com.google.android.libraries.feed.api.internal.modelprovider.ModelProvider;
import com.google.android.libraries.feed.api.internal.modelprovider.ModelProvider.ViewDepthProvider;
import com.google.android.libraries.feed.api.internal.modelprovider.ModelProviderFactory;
import com.google.android.libraries.feed.common.functional.Predicate;
import com.google.search.now.feed.client.StreamDataProto.StreamStructure;
import com.google.search.now.feed.client.StreamDataProto.UiContext;

/**
 * Fake for tests using {@link ModelProviderFactory}. Functionality should be added to this class as
 * needed.
 */
public class FakeModelProviderFactory implements ModelProviderFactory {

  private FakeModelProvider modelProvider;
  private String createSessionId;
  /*@Nullable*/ private ViewDepthProvider viewDepthProvider;

  @Override
  public ModelProvider create(String sessionId, UiContext uiContext) {
    this.modelProvider = new FakeModelProvider();
    this.createSessionId = sessionId;
    return modelProvider;
  }

  @Override
  public ModelProvider createNew(
      /*@Nullable*/ ViewDepthProvider viewDepthProvider, UiContext uiContext) {
    this.viewDepthProvider = viewDepthProvider;
    this.modelProvider = new FakeModelProvider();
    return modelProvider;
  }

  @Override
  public ModelProvider createNew(
      /*@Nullable*/ ViewDepthProvider viewDepthProvider,
      /*@Nullable*/ Predicate<StreamStructure> filterPredicate,
      UiContext uiContext) {
    this.viewDepthProvider = viewDepthProvider;
    this.modelProvider = new FakeModelProvider();
    return modelProvider;
  }

  public FakeModelProvider getLatestModelProvider() {
    return modelProvider;
  }

  public String getLatestCreateSessionId() {
    return createSessionId;
  }

  /*@Nullable*/
  public ViewDepthProvider getViewDepthProvider() {
    return viewDepthProvider;
  }
}