1 files changed, 154 insertions, 0 deletions
diff --git a/tests/unit/src/com/android/tv/util/MockApplicationSingletons.java b/tests/unit/src/com/android/tv/util/MockApplicationSingletons.java
new file mode 100644
index 00000000..4cfc7f8a
--- /dev/null
+++ b/tests/unit/src/com/android/tv/util/MockApplicationSingletons.java
@@ -0,0 +1,154 @@
+/*
+ * 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.util;
+
+import android.content.Context;
+
+import com.android.tv.ApplicationSingletons;
+import com.android.tv.InputSessionManager;
+import com.android.tv.MainActivityWrapper;
+import com.android.tv.TvApplication;
+import com.android.tv.analytics.Analytics;
+import com.android.tv.analytics.Tracker;
+import com.android.tv.config.RemoteConfig;
+import com.android.tv.data.ChannelDataManager;
+import com.android.tv.data.PreviewDataManager;
+import com.android.tv.data.ProgramDataManager;
+import com.android.tv.dvr.DvrDataManager;
+import com.android.tv.dvr.DvrManager;
+import com.android.tv.dvr.DvrScheduleManager;
+import com.android.tv.dvr.DvrStorageStatusManager;
+import com.android.tv.dvr.DvrWatchedPositionManager;
+import com.android.tv.dvr.recorder.RecordingScheduler;
+import com.android.tv.perf.PerformanceMonitor;
+
+/**
+ * Mock {@link ApplicationSingletons} class.
+ */
+public class MockApplicationSingletons implements ApplicationSingletons {
+    private final TvApplication mApp;
+
+    private PerformanceMonitor mPerformanceMonitor;
+
+    public MockApplicationSingletons(Context context) {
+        mApp = (TvApplication) context.getApplicationContext();
+    }
+
+    @Override
+    public Analytics getAnalytics() {
+        return mApp.getAnalytics();
+    }
+
+    @Override
+    public ChannelDataManager getChannelDataManager() {
+        return mApp.getChannelDataManager();
+    }
+
+    @Override
+    public boolean isChannelDataManagerLoadFinished() {
+        return mApp.isChannelDataManagerLoadFinished();
+    }
+
+    @Override
+    public ProgramDataManager getProgramDataManager() {
+        return mApp.getProgramDataManager();
+    }
+
+    @Override
+    public boolean isProgramDataManagerCurrentProgramsLoadFinished() {
+        return mApp.isProgramDataManagerCurrentProgramsLoadFinished();
+    }
+
+    @Override
+    public PreviewDataManager getPreviewDataManager() {
+        return mApp.getPreviewDataManager();
+    }
+
+    @Override
+    public DvrDataManager getDvrDataManager() {
+        return mApp.getDvrDataManager();
+    }
+
+    @Override
+    public DvrStorageStatusManager getDvrStorageStatusManager() {
+        return mApp.getDvrStorageStatusManager();
+    }
+
+    @Override
+    public DvrScheduleManager getDvrScheduleManager() {
+        return mApp.getDvrScheduleManager();
+    }
+
+    @Override
+    public DvrManager getDvrManager() {
+        return mApp.getDvrManager();
+    }
+
+    @Override
+    public RecordingScheduler getRecordingScheduler() {
+        return mApp.getRecordingScheduler();
+    }
+
+    @Override
+    public DvrWatchedPositionManager getDvrWatchedPositionManager() {
+        return mApp.getDvrWatchedPositionManager();
+    }
+
+    @Override
+    public InputSessionManager getInputSessionManager() {
+        return mApp.getInputSessionManager();
+    }
+
+    @Override
+    public Tracker getTracker() {
+        return mApp.getTracker();
+    }
+
+    @Override
+    public TvInputManagerHelper getTvInputManagerHelper() {
+        return mApp.getTvInputManagerHelper();
+    }
+
+    @Override
+    public MainActivityWrapper getMainActivityWrapper() {
+        return mApp.getMainActivityWrapper();
+    }
+
+    @Override
+    public AccountHelper getAccountHelper() {
+        return mApp.getAccountHelper();
+    }
+
+    @Override
+    public RemoteConfig getRemoteConfig() {
+        return mApp.getRemoteConfig();
+    }
+
+    @Override
+    public boolean isRunningInMainProcess() {
+        return mApp.isRunningInMainProcess();
+    }
+
+    @Override
+    public PerformanceMonitor getPerformanceMonitor() {
+        return mPerformanceMonitor != null ? mPerformanceMonitor : mApp.getPerformanceMonitor();
+    }
+
+    public void setPerformanceMonitor(PerformanceMonitor performanceMonitor) {
+        mPerformanceMonitor = performanceMonitor;
+    }
+}