Actual test for onLoad

DeclareDesign · Feb 9, 2018 · ee2d9b9 · ee2d9b9
1 parent c05012d
commit ee2d9b9
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 6 deletions.
diff --git a/tests/testthat/test-texreg.R b/tests/testthat/test-texreg.R
@@ -3,21 +3,19 @@
 context("S3 - texreg builds")
 
 test_that("texreg extension works", {
+  skip_if_not_installed("texreg")
+  library(texreg)
 
   model2 <- lm_robust(extra~group, sleep, clusters = "ID")
 
-  library(texreg)
 
   capture.output(treg <- extract(model2, TRUE, TRUE, TRUE, TRUE, TRUE, TRUE))
   expect_is(
     treg,
     "texreg"
   )
 
-  # juicin the stats
-  expect_error(
-    .onLoad("lm_robust", "lm_robust"),
-    NA
-  )
 
 })
+
+
diff --git a/tests/testthat/test-zzz.R b/tests/testthat/test-zzz.R
@@ -0,0 +1,14 @@
+context("zzz.R - .onLoad")
+
+test_that("onLoad makes generics if texreg is present", {
+  skip_if_not_installed("texreg")
+
+  e <- environment(.onLoad)
+
+  environment(.onLoad) <- new.env(parent=parent.env(e))
+  environment(.onLoad)$extract.lm_robust <- e$extract.lm_robust
+
+  expect_null(.onLoad("estimatr", "estimatr"))
+  environment(.onLoad) <- e
+
+})