tests/common/src/com/android/tv/testing/TestSingletonApp.java

/*
 * Copyright (C) 2017 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.tv.testing;

import android.app.Application;
import android.media.tv.TvInputManager;
import android.os.AsyncTask;
import com.android.tv.InputSessionManager;
import com.android.tv.MainActivityWrapper;
import com.android.tv.TvSingletons;
import com.android.tv.analytics.Analytics;
import com.android.tv.analytics.Tracker;
import com.android.tv.common.BaseApplication;
import com.android.tv.common.experiments.ExperimentLoader;
import com.android.tv.common.flags.impl.DefaultBackendKnobsFlags;
import com.android.tv.common.flags.impl.DefaultCloudEpgFlags;
import com.android.tv.common.flags.impl.DefaultConcurrentDvrPlaybackFlags;
import com.android.tv.common.flags.impl.DefaultExoplayer2Flags;
import com.android.tv.common.flags.impl.DefaultUiFlags;
import com.android.tv.common.recording.RecordingStorageStatusManager;
import com.android.tv.common.singletons.HasSingletons;
import com.android.tv.common.util.Clock;
import com.android.tv.data.ChannelDataManager;
import com.android.tv.data.PreviewDataManager;
import com.android.tv.data.ProgramDataManager;
import com.android.tv.data.epg.EpgFetcher;
import com.android.tv.data.epg.EpgReader;
import com.android.tv.dvr.DvrDataManager;
import com.android.tv.dvr.DvrManager;
import com.android.tv.dvr.DvrScheduleManager;
import com.android.tv.dvr.DvrWatchedPositionManager;
import com.android.tv.dvr.recorder.RecordingScheduler;
import com.android.tv.perf.PerformanceMonitor;
import com.android.tv.perf.stub.StubPerformanceMonitor;
import com.android.tv.testing.dvr.DvrDataManagerInMemoryImpl;
import com.android.tv.testing.testdata.TestData;
import com.android.tv.tuner.singletons.TunerSingletons;
import com.android.tv.tuner.source.TsDataSourceManager;
import com.android.tv.tuner.source.TunerTsStreamerManager;
import com.android.tv.tuner.tvinput.factory.TunerSessionFactory;
import com.android.tv.tuner.tvinput.factory.TunerSessionFactoryImpl;
import com.android.tv.tunerinputcontroller.BuiltInTunerManager;
import com.android.tv.util.SetupUtils;
import com.android.tv.util.TvInputManagerHelper;
import com.android.tv.util.account.AccountHelper;
import com.google.common.base.Optional;
import java.util.concurrent.Executor;
import javax.inject.Provider;

/** Test application for Live TV. */
public class TestSingletonApp extends Application
        implements TvSingletons, TunerSingletons, HasSingletons<TvSingletons> {
    public final FakeClock fakeClock = FakeClock.createWithCurrentTime();
    public final FakeEpgReader epgReader = new FakeEpgReader(fakeClock);
    public final FakeEpgFetcher epgFetcher = new FakeEpgFetcher();

    public FakeTvInputManagerHelper tvInputManagerHelper;
    public SetupUtils setupUtils;
    public DvrManager dvrManager;
    public DvrDataManager mDvrDataManager;

    private final Provider<EpgReader> mEpgReaderProvider = SingletonProvider.create(epgReader);
    private final Optional<BuiltInTunerManager> mBuiltInTunerManagerOptional = Optional.absent();
    private final DefaultBackendKnobsFlags mBackendKnobs = new DefaultBackendKnobsFlags();
    private final DefaultCloudEpgFlags mCloudEpgFlags = new DefaultCloudEpgFlags();
    private final DefaultUiFlags mUiFlags = new DefaultUiFlags();
    private final DefaultConcurrentDvrPlaybackFlags mConcurrentDvrPlaybackFlags =
            new DefaultConcurrentDvrPlaybackFlags();
    private final TsDataSourceManager.Factory mTsDataSourceManagerFactory =
            new TsDataSourceManager.Factory(() -> new TunerTsStreamerManager(null));
    private final TunerSessionFactoryImpl mTunerSessionFactory =
            new TunerSessionFactoryImpl(
                    new DefaultExoplayer2Flags(),
                    mConcurrentDvrPlaybackFlags,
                    mTsDataSourceManagerFactory);
    private PerformanceMonitor mPerformanceMonitor;
    private ChannelDataManager mChannelDataManager;

    @Override
    public void onCreate() {
        super.onCreate();
        tvInputManagerHelper = new FakeTvInputManagerHelper(this);
        setupUtils = new SetupUtils(this, mBuiltInTunerManagerOptional);
        tvInputManagerHelper.start();
        mChannelDataManager = new ChannelDataManager(this, tvInputManagerHelper);
        mChannelDataManager.start();
        mDvrDataManager = new DvrDataManagerInMemoryImpl(this, fakeClock);
        // HACK reset the singleton for tests
        BaseApplication.sSingletons = this;
    }

    public void loadTestData(TestData testData, long durationMs) {
        tvInputManagerHelper
                .getFakeTvInputManager()
                .add(testData.getTvInputInfo(), TvInputManager.INPUT_STATE_CONNECTED);
        testData.init(this, fakeClock, durationMs);
    }

    @Override
    public Analytics getAnalytics() {
        return null;
    }

    @Override
    public void handleInputCountChanged() {}

    @Override
    public ChannelDataManager getChannelDataManager() {
        return mChannelDataManager;
    }

    @Override
    public boolean isChannelDataManagerLoadFinished() {
        return false;
    }

    @Override
    public ProgramDataManager getProgramDataManager() {
        return null;
    }

    @Override
    public boolean isProgramDataManagerCurrentProgramsLoadFinished() {
        return false;
    }

    @Override
    public PreviewDataManager getPreviewDataManager() {
        return null;
    }

    @Override
    public DvrDataManager getDvrDataManager() {
        return mDvrDataManager;
    }

    @Override
    public DvrScheduleManager getDvrScheduleManager() {
        return null;
    }

    @Override
    public DvrManager getDvrManager() {
        return dvrManager;
    }

    @Override
    public RecordingScheduler getRecordingScheduler() {
        return null;
    }

    @Override
    public DvrWatchedPositionManager getDvrWatchedPositionManager() {
        return null;
    }

    @Override
    public InputSessionManager getInputSessionManager() {
        return new InputSessionManager(this);
    }

    @Override
    public Tracker getTracker() {
        return null;
    }

    @Override
    public TvInputManagerHelper getTvInputManagerHelper() {
        return tvInputManagerHelper;
    }

    @Override
    public Provider<EpgReader> providesEpgReader() {
        return mEpgReaderProvider;
    }

    @Override
    public EpgFetcher getEpgFetcher() {
        return epgFetcher;
    }

    @Override
    public SetupUtils getSetupUtils() {
        return setupUtils;
    }

    @Override
    public Optional<BuiltInTunerManager> getBuiltInTunerManager() {
        return mBuiltInTunerManagerOptional;
    }

    @Override
    public ExperimentLoader getExperimentLoader() {
        return new ExperimentLoader();
    }

    @Override
    public MainActivityWrapper getMainActivityWrapper() {
        return null;
    }

    @Override
    public AccountHelper getAccountHelper() {
        return null;
    }

    @Override
    public Clock getClock() {
        return fakeClock;
    }

    @Override
    public RecordingStorageStatusManager getRecordingStorageStatusManager() {
        return null;
    }

    @Override
    public boolean isRunningInMainProcess() {
        return false;
    }

    @Override
    public PerformanceMonitor getPerformanceMonitor() {
        if (mPerformanceMonitor == null) {
            mPerformanceMonitor = new StubPerformanceMonitor();
        }
        return mPerformanceMonitor;
    }

    @Override
    public String getEmbeddedTunerInputId() {
        return "com.android.tv/.tuner.tvinput.TunerTvInputService";
    }

    @Override
    public Executor getDbExecutor() {
        return AsyncTask.SERIAL_EXECUTOR;
    }

    @Override
    public DefaultBackendKnobsFlags getBackendKnobs() {
        return mBackendKnobs;
    }

    @Override
    public DefaultCloudEpgFlags getCloudEpgFlags() {
        return mCloudEpgFlags;
    }

    @Override
    public DefaultUiFlags getUiFlags() {
        return mUiFlags;
    }

    @Override
    public BuildType getBuildType() {
        return BuildType.ENG;
    }

    @Override
    public DefaultConcurrentDvrPlaybackFlags getConcurrentDvrPlaybackFlags() {
        return mConcurrentDvrPlaybackFlags;
    }

    public TunerSessionFactory getTunerSessionFactory() {
        return mTunerSessionFactory;
    }

    @Override
    public TvSingletons singletons() {
        return this;
    }
}