R/driver-specific-connections.R

In pachamaltese/dbtest-test: A Test Environment for DB Requests

.onLoad <- function(libname, pkgname) {
  set_driver_class(
    pkg = "RSQLite",
    mock_conn = "DBIMockSQLiteConnection",
    real_conn = "SQLiteConnection"
  )
}

set_driver_class <- function(pkg, mock_conn, real_conn) {
  # following the convention from Writing R Extensions
  # https://cran.r-project.org/doc/manuals/r-release/R-exts.html#Suggested-packages
  if (requireNamespace(pkg, quietly = TRUE)) {
    methods::setClass(mock_conn, contains = c("DBIMockConnection", real_conn))
  }
}