tests/testthat/test-geelite.R

In geeLite: Building and Managing Local Databases from 'Google Earth Engine'

suppressMessages(suppressWarnings({
  library(rnaturalearthdata)
  library(geojsonio)
  library(jsonlite)
  library(RSQLite)
  library(withr)
}))

# ------------------------------------------------------------------------------

test_that("Testing geeLite Package Pipeline", {

  benchmark <- TRUE

  if (!check_rgee_ready()) {

    benchmark <- FALSE

  } else {

    # Define test modes and initialize timing storage
    modes <- c("local", "drive")
    times <- list()

    for (mode in modes) {

      # Start timer
      start_time <- Sys.time()

      # Set up test environment
      test_path <- local_tempdir()

      # ------------------------------------------------------------------------
      # Step 1: Generate configuration file
      # ------------------------------------------------------------------------

      regions <- c("SO", "YE")
      source <- list(
        "MODIS/061/MOD13A2" = list(
          "NDVI" = c("mean", "sd")
        )
      )
      start <- "2020-01-01"
      resol <- 3

      set_config(path = test_path,
                 regions = regions,
                 source = source,
                 start = start,
                 resol = resol,
                 verbose = FALSE)

      # Check if the config file is created
      config_file <- file.path(test_path, "config/config.json")
      expect_true(file.exists(config_file))

      # ------------------------------------------------------------------------
      # Step 2: Create database
      # ------------------------------------------------------------------------

      run_geelite(path = test_path,
                  mode = mode,
                  verbose = FALSE)

      # Check if the database is created
      db_file <- file.path(test_path, "data/geelite.db")
      expect_true(file.exists(db_file))

      # Check the content of the database
      con <- dbConnect(SQLite(), dbname = db_file)
      df <- dbReadTable(con, "MODIS/061/MOD13A2")
      expect_equal(sort(unique(df$band)), sort("NDVI"))
      expect_equal(sort(unique(df$zonal_stat)), sort(c("mean", "sd")))
      dbDisconnect(con)

      # Testing get_config and get_state functions
      config <- get_config(path = test_path)
      state <- get_state(path = test_path)
      expect_equal(config, state)

      # Testing read_db function
      db <- read_db(path = test_path)
      expect_equal(names(db), c("grid", "MODIS/061/MOD13A2/NDVI/mean",
                                "MODIS/061/MOD13A2/NDVI/sd"))
      rm(db)

      # ------------------------------------------------------------------------
      # Step 3: Modify configuration file
      # ------------------------------------------------------------------------

      modify_config(
        path = test_path,
        keys = list("regions", c("source", "MODIS/061/MOD13A2", "NDVI")),
        new_values = list(c("SO", "KE"), c("mean", "max")),
        verbose = FALSE
      )

      # Check if the config file is updated
      config <- fromJSON(config_file)
      expected_source <- list(
        `MODIS/061/MOD13A2` = list(
          NDVI = c("mean", "max")
        )
      )
      expect_equal(config$source, expected_source)

      # ------------------------------------------------------------------------
      # Step 4: Update database
      # ------------------------------------------------------------------------

      run_geelite(path = test_path,
                  mode = mode,
                  verbose = FALSE)

      # Check the content of the updated database
      con <- dbConnect(SQLite(), dbname = db_file)
      df <- dbReadTable(con, "MODIS/061/MOD13A2")
      expect_equal(sort(unique(df$band)), sort(c("NDVI")))
      expect_equal(sort(unique(df$zonal_stat)), sort(c("mean", "max")))
      grid <- dbReadTable(con, "grid")
      expect_equal(sort(unique(grid$iso)), sort(c("SO", "KE")))
      dbDisconnect(con)

      # Measure total execution time
      end_time <- Sys.time()
      total_time <- round(difftime(end_time, start_time, units = "secs"), 2)

      # Store the time for this mode
      times[[mode]] <- total_time

    }
  }

  # Print execution times after both tests are complete
  if (benchmark == TRUE) {
    cat("\n\n========== Execution Time Summary ==========\n")
    cat(sprintf("Local Mode: %s seconds\n", times[["local"]]))
    cat(sprintf("Drive Mode: %s seconds\n", times[["drive"]]))
    cat("===========================================\n\n")
  }
})