1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123
|
# function translation ----------------------------------------------------
test_that("custom scalar translated correctly", {
local_con(simulate_mysql())
expect_equal(translate_sql(as.logical(1L)), sql("IF(1, TRUE, FALSE)"))
expect_equal(translate_sql(str_locate("abc", "b")), sql("REGEXP_INSTR('abc', 'b')"))
expect_equal(translate_sql(str_replace_all("abc", "b", "d")), sql("REGEXP_REPLACE('abc', 'b', 'd')"))
})
test_that("custom aggregators translated correctly", {
local_con(simulate_mysql())
expect_equal(translate_sql(str_flatten(y, ","), window = FALSE), sql("GROUP_CONCAT(`y` SEPARATOR ',')"))
})
test_that("use CHAR type for as.character", {
local_con(simulate_mysql())
expect_equal(translate_sql(as.character(x)), sql("CAST(`x` AS CHAR)"))
})
test_that("custom stringr functions translated correctly", {
local_con(simulate_mysql())
expect_equal(translate_sql(str_c(x, y)), sql("CONCAT_WS('', `x`, `y`)"))
expect_equal(translate_sql(str_detect(x, y)), sql("`x` REGEXP `y`"))
expect_equal(translate_sql(str_like(x, y)), sql("`x` LIKE `y`"))
expect_equal(translate_sql(str_like(x, y, FALSE)), sql("`x` LIKE BINARY `y`"))
expect_equal(translate_sql(str_locate(x, y)), sql("REGEXP_INSTR(`x`, `y`)"))
expect_equal(translate_sql(str_replace_all(x, y, z)), sql("REGEXP_REPLACE(`x`, `y`, `z`)"))
})
# verbs -------------------------------------------------------------------
test_that("generates custom sql", {
con <- simulate_mysql()
expect_snapshot(sql_table_analyze(con, in_schema("schema", "tbl")))
expect_snapshot(sql_query_explain(con, sql("SELECT * FROM table")))
lf <- lazy_frame(x = 1, con = con)
expect_snapshot(left_join(lf, lf, by = "x", na_matches = "na"))
expect_snapshot(error = TRUE, full_join(lf, lf, by = "x"))
expect_snapshot(slice_sample(lf, n = 1))
expect_snapshot(copy_inline(con, tibble(x = 1:2, y = letters[1:2])) %>% remote_query())
})
test_that("`sql_query_update_from()` is correct", {
df_y <- lazy_frame(
a = 2:3, b = c(12L, 13L), c = -(2:3), d = c("y", "z"),
con = simulate_mysql(),
.name = "df_y"
) %>%
mutate(c = c + 1)
expect_snapshot(
sql_query_update_from(
con = simulate_mysql(),
x_name = ident("df_x"),
y = df_y,
by = c("a", "b"),
update_values = sql(
c = "COALESCE(`df_x`.`c`, `...y`.`c`)",
d = "`...y`.`d`"
),
returning_cols = c("a", b2 = "b")
)
)
})
# live database -----------------------------------------------------------
test_that("logicals converted to integer correctly", {
db <- copy_to_test("MariaDB", data.frame(x = c(TRUE, FALSE, NA)))
expect_identical(db %>% pull(), c(1L, 0L, NA))
})
test_that("can explain", {
db <- copy_to_test("MariaDB", data.frame(x = 1:3))
expect_snapshot(db %>% mutate(y = x + 1) %>% explain())
})
test_that("can overwrite temp tables", {
con <- src_test("MariaDB")
df1 <- tibble(x = 1)
copy_to(con, df1, "test-df", temporary = TRUE)
df2 <- tibble(x = 2)
db2 <- copy_to(con, df2, "test-df", temporary = TRUE, overwrite = TRUE)
expect_equal(collect(db2), df2)
})
test_that("can update", {
con <- src_test("MariaDB")
df_x <- tibble(a = 1:3, b = 11:13, c = 1:3, d = c("a", "b", "c"))
x <- copy_to(con, df_x, "df_x", temporary = TRUE, overwrite = TRUE)
df_y <- tibble(a = 2:3, b = c(12L, 13L), c = -(2:3), d = c("y", "z"))
y <- copy_to(con, df_y, "df_y", temporary = TRUE, overwrite = TRUE) %>%
mutate(c = c + 1)
# `RETURNING` in an `UPDATE` clause is not (yet) supported for MariaDB
# https://jira.mariadb.org/browse/MDEV-5092
expect_equal(
rows_update(
x, y,
by = c("a", "b"),
unmatched = "ignore",
in_place = TRUE
) %>%
collect(),
tibble(
a = 1:3,
b = 11:13,
c = c(1L, -1L, -2L),
d = c("a", "y", "z")
)
)
})
|