tests/testthat/test-dbQuoteIdentifier.R

In r-dbi/RMariaDB: Database Interface and MariaDB Driver

test_that("quoting string", {
  con <- mariadbDefault()
  on.exit(dbDisconnect(con))

  quoted <- dbQuoteIdentifier(con, "Robert'); DROP TABLE Students;--")
  expect_s4_class(quoted, 'SQL')
  expect_equal(as.character(quoted),
    "`Robert'); DROP TABLE Students;--`")
})

test_that("quoting SQL", {
  con <- mariadbDefault()
  on.exit(dbDisconnect(con))

  quoted <- dbQuoteIdentifier(con, SQL("Robert'); DROP TABLE Students;--"))
  expect_s4_class(quoted, 'SQL')
  expect_equal(as.character(quoted),
    "Robert'); DROP TABLE Students;--")
})

test_that("quoting Id", {
  con <- mariadbDefault()
  on.exit(dbDisconnect(con))

  quoted <- dbQuoteIdentifier(con, Id(schema = 'Robert', table = 'Students;--'))
  expect_s4_class(quoted, 'SQL')
  expect_equal(as.character(quoted),
    "`Robert`.`Students;--`")
})

test_that("quoting Id with column, #254", {
  con <- mariadbDefault()
  on.exit(dbDisconnect(con))

  quoted <- dbQuoteIdentifier(con, Id(schema = 'Robert', table = 'Students;--', column = "dormitory"))
  expect_s4_class(quoted, 'SQL')
  expect_equal(as.character(quoted),
    "`Robert`.`Students;--`.`dormitory`")
})

test_that("quoting Id with column, unordered", {
  con <- mariadbDefault()
  on.exit(dbDisconnect(con))

  quoted <- dbQuoteIdentifier(con, Id(column = "dormitory", table = 'Students;--'))
  expect_s4_class(quoted, 'SQL')
  expect_equal(as.character(quoted),
    "`Students;--`.`dormitory`")
})

test_that("quoting errors", {
  con <- mariadbDefault()
  on.exit(dbDisconnect(con))

  expect_error(Id(table = 'Robert', table = 'Students;--'))
})

test_that("unquoting identifier - SQL with quotes", {
  con <- mariadbDefault()
  on.exit(dbDisconnect(con))

  expect_equal(dbUnquoteIdentifier(con, SQL('`Students;--`')),
    list(Id(table = 'Students;--')))

  expect_equal(dbUnquoteIdentifier(con, SQL('`Robert`.`Students;--`')),
    list(Id(schema = 'Robert', table = 'Students;--')))

  expect_equal(dbUnquoteIdentifier(con, SQL('`Rob``ert`.`Students;--`')),
    list(Id(schema = 'Rob`ert', table = 'Students;--')))

  expect_equal(dbUnquoteIdentifier(con, SQL('`Rob.ert`.`Students;--`')),
    list(Id(schema = 'Rob.ert',  table = 'Students;--')))

  expect_error(dbUnquoteIdentifier(con, SQL('`Robert.`Students`')),
    "^Can't unquote")
})

test_that("unquoting identifier - SQL without quotes", {
  con <- mariadbDefault()
  on.exit(dbDisconnect(con))

  expect_equal(dbUnquoteIdentifier(con, SQL('Students')),
    list(Id(table = 'Students')))

  expect_equal(dbUnquoteIdentifier(con, SQL('Robert.Students')),
    list(Id(schema = 'Robert', table = 'Students')))

  expect_error(dbUnquoteIdentifier(con, SQL('Rob``ert.Students')),
    "^Can't unquote")
})

test_that("unquoting identifier - Id", {
  con <- mariadbDefault()
  on.exit(dbDisconnect(con))

  expect_equal(dbUnquoteIdentifier(con,
    Id(schema = 'Robert', table = 'Students;--')),
  list(Id(schema = 'Robert', table = 'Students;--')))
})