tasks/app/src/androidTest/java/org/tasks/injection/TestModule.java

package org.tasks.injection;

import androidx.room.Room;
import android.content.Context;
import com.todoroo.astrid.dao.Database;
import com.todoroo.astrid.dao.TaskDao;
import dagger.Module;
import dagger.Provides;
import org.tasks.data.AlarmDao;
import org.tasks.data.CaldavDao;
import org.tasks.data.DeletionDao;
import org.tasks.data.FilterDao;
import org.tasks.data.GoogleTaskDao;
import org.tasks.data.GoogleTaskListDao;
import org.tasks.data.LocationDao;
import org.tasks.data.TagDao;
import org.tasks.data.TagDataDao;
import org.tasks.data.TaskAttachmentDao;
import org.tasks.data.TaskListMetadataDao;
import org.tasks.data.UserActivityDao;
import org.tasks.notifications.NotificationDao;
import org.tasks.preferences.PermissionChecker;
import org.tasks.preferences.PermissivePermissionChecker;

@Module
public class TestModule {

  private final Context context;

  TestModule(Context context) {
    this.context = context;
  }

  @Provides
  @ApplicationScope
  public Database getDatabase() {
    return Room.inMemoryDatabaseBuilder(context, Database.class)
        .fallbackToDestructiveMigration()
        .build();
  }

  @Provides
  public NotificationDao getNotificationDao(Database appDatabase) {
    return appDatabase.notificationDao();
  }

  @Provides
  public TagDataDao getTagDataDao(Database database) {
    return database.getTagDataDao();
  }

  @Provides
  public UserActivityDao getUserActivityDao(Database database) {
    return database.getUserActivityDao();
  }

  @Provides
  public TaskListMetadataDao getTaskListMetadataDao(Database database) {
    return database.getTaskListMetadataDao();
  }

  @Provides
  public GoogleTaskListDao getGoogleTaskListDao(Database database) {
    return database.getGoogleTaskListDao();
  }

  @Provides
  public AlarmDao getAlarmDao(Database database) {
    return database.getAlarmDao();
  }

  @Provides
  public GoogleTaskDao getGoogleTaskDao(Database database) {
    return database.getGoogleTaskDao();
  }

  @Provides
  public TagDao getTagDao(Database database) {
    return database.getTagDao();
  }

  @Provides
  public LocationDao getLocationDao(Database database) {
    return database.getLocationDao();
  }

  @Provides
  public TaskDao getTaskDao(Database database) {
    TaskDao taskDao = database.getTaskDao();
    taskDao.initialize(context);
    return taskDao;
  }

  @Provides
  public CaldavDao getCaldavDao(Database database) {
    return database.getCaldavDao();
  }

  @Provides
  public FilterDao getFilterDao(Database database) {
    return database.getFilterDao();
  }

  @Provides
  public DeletionDao getDeletionDao(Database database) {
    return database.getDeletionDao();
  }

  @Provides
  public TaskAttachmentDao getTaskAttachmentDao(Database database) {
    return database.getTaskAttachmentDao();
  }

  @ApplicationScope
  @Provides
  @ForApplication
  public Context getContext() {
    return context;
  }

  @Provides
  public PermissionChecker getPermissionChecker() {
    return new PermissivePermissionChecker(context);
  }
}