tests/testthat/test-utils.R

In jonthegeek/factory: Build Function Factories

test_that("body_replace replaces bits of bodies", {
  test_fun <- function(x) {
    x^exp
  }
  body(test_fun) <- body_replace(body(test_fun), quote(exp), quote(!!exp))
  expected_fun <- function(x) {
    x^!!exp
  }
  expect_identical(test_fun, expected_fun)
})

test_that("body_insert errors appropriately.", {
  fun <- function(x) x + 1
  expect_error(
    body_insert(
      fn_body = body(fun),
      insertion = quote("Should not matter")
    ),
    regexp = "Please wrap your function"
  )
})

test_that("body_replace doesn't freak out with NULL.", {
  test_fun <- function(x_vector) {
    matrix(x_vector, ncol = 1, dimnames = list(NULL, "x"))
  }
  test_body <- body(test_fun)
  expect_error(
    body_replace(
      fn_body = test_body,
      target = quote(x_vector),
      replacement = quote("got it!")
    ),
    NA
  )
})