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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
#' spec_result_send_query
#' @family Arrow specifications
#' @usage NULL
#' @format NULL
#' @keywords NULL
spec_arrow_send_query_arrow <- list(
arrow_send_query_formals = function() {
# <establish formals of described functions>
expect_equal(names(formals(dbSendQueryArrow)), c("conn", "statement", "..."))
},
arrow_send_query_trivial = function(con) {
#' @return
#' `dbSendQueryArrow()` returns
res <- expect_visible(dbSendQueryArrow(con, trivial_query()))
#' an S4 object that inherits from [DBIResultArrow-class].
expect_s4_class(res, "DBIResultArrow")
#' The result set can be used with [dbFetchArrow()] to extract records.
expect_equal(check_arrow(dbFetchArrow(res))[[1]], 1.5)
#' Once you have finished using a result, make sure to clear it
#' with [dbClearResult()].
dbClearResult(res)
},
#'
arrow_send_query_closed_connection = function(ctx, closed_con) {
#' @section Failure modes:
#' An error is raised when issuing a query over a closed
expect_error(dbSendQueryArrow(closed_con, trivial_query()))
},
arrow_send_query_invalid_connection = function(ctx, invalid_con) {
#' or invalid connection,
expect_error(dbSendQueryArrow(invalid_con, trivial_query()))
},
arrow_send_query_non_string = function(con) {
#' or if the query is not a non-`NA` string.
expect_error(dbSendQueryArrow(con, character()))
expect_error(dbSendQueryArrow(con, letters))
expect_error(dbSendQueryArrow(con, NA_character_))
},
arrow_send_query_syntax_error = function(ctx, con) {
skip_if_not_dbitest(ctx, "1.7.99.5")
#' An error is also raised if the syntax of the query is invalid
#' and all query parameters are given (by passing the `params` argument)
#' or the `immediate` argument is set to `TRUE`.
#'
#' @section Failure modes:
expect_error(dbSendQueryArrow(con, "SELLECT", params = list()))
expect_error(dbSendQueryArrow(con, "SELLECT", immediate = TRUE))
},
#' @section Additional arguments:
#' The following arguments are not part of the `dbSendQueryArrow()` generic
#' (to improve compatibility across backends)
#' but are part of the DBI specification:
#' - `params` (default: `NULL`)
#' - `immediate` (default: `NULL`)
#'
#' They must be provided as named arguments.
#' See the "Specification" sections for details on their usage.
arrow_send_query_result_valid = function(ctx, con) {
skip_if_not_dbitest(ctx, "1.7.99.6")
#' @section Specification:
#' No warnings occur under normal conditions.
expect_warning(res <- dbSendQueryArrow(con, trivial_query()), NA)
#' When done, the DBIResult object must be cleared with a call to
#' [dbClearResult()].
dbClearResult(res)
},
#
arrow_send_query_stale_warning = function(ctx) {
skip_if_not_dbitest(ctx, "1.7.99.7")
#' Failure to clear the result set leads to a warning
#' when the connection is closed.
con <- connect(ctx)
on.exit(dbDisconnect(con))
expect_warning(dbSendQueryArrow(con, trivial_query()), NA)
expect_warning({
dbDisconnect(con)
gc()
})
on.exit(NULL)
},
#'
arrow_send_query_only_one_result_set = function(ctx, con) {
skip_if_not_dbitest(ctx, "1.7.99.8")
#' If the backend supports only one open result set per connection,
res1 <- dbSendQueryArrow(con, trivial_query())
#' issuing a second query invalidates an already open result set
#' and raises a warning.
expect_warning(res2 <- dbSendQueryArrow(con, "SELECT 2"))
expect_false(dbIsValid(res1))
#' The newly opened result set is valid
expect_true(dbIsValid(res2))
#' and must be cleared with `dbClearResult()`.
dbClearResult(res2)
},
#'
arrow_send_query_params = function(ctx, con) {
skip_if_not_dbitest(ctx, "1.7.99.9")
#' The `param` argument allows passing query parameters, see [dbBind()] for details.
placeholder_funs <- get_placeholder_funs(ctx)
for (placeholder_fun in placeholder_funs) {
placeholder <- placeholder_fun(1)
query <- paste0("SELECT ", placeholder, " + 1.0 AS a")
values <- trivial_values(3) - 1
params <- stats::setNames(list(values), names(placeholder))
rs <- dbSendQueryArrow(con, query, params = params)
ret <- dbFetch(rs)
expect_equal(ret, trivial_df(3), info = placeholder)
dbClearResult(rs)
}
},
arrow_send_query_immediate = function(ctx, con, table_name) {
skip_if_not_dbitest(ctx, "1.7.99.10")
#' @inheritSection spec_result_get_query Specification for the `immediate` argument
res <- expect_visible(dbSendQueryArrow(con, trivial_query(), immediate = TRUE))
expect_s4_class(res, "DBIResultArrow")
expect_error(dbGetRowsAffected(res), NA)
dbClearResult(res)
},
NULL
)
|
