FazBrowse GitHub Viewer
|
Trending
|
URL:
|
Home
Tools:
[Download Repo ZIP]
[View Raw Code]
[Original HTTPS Page]
sqlpp11/tests/postgresql/usage/Blob.cpp at optional · rbock/sqlpp11 · GitHub
rbock
/
sqlpp11
Public
Notifications
You must be signed in to change notification settings
Fork
361
Star
2.6k
Code
Issues
2
Pull requests
1
Actions
Projects
Security and quality
0
Insights
Additional navigation options
Code
Issues
Pull requests
Actions
Projects
Security and quality
Insights
Expand file tree
Breadcrumbs
sqlpp11
/
tests
/
postgresql
/
usage
/
Blob.cpp
Copy path
More file actions
More file actions
Latest commit
History
History
History
114 lines (99 loc) · 4.28 KB
Breadcrumbs
sqlpp11
/
tests
/
postgresql
/
usage
/
Blob.cpp
Copy path
File metadata and controls
114 lines (99 loc) · 4.28 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
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
/*
* Copyright (c) 2019-2019, Jaroslav Bisikirski
* Copyright (c) 2021-2021, Roland Bock
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without modification,
* are permitted provided that the following conditions are met:
*
* * Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
* IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
* OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
* OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#
include
<
sqlpp11/postgresql/postgresql.h
>
#
include
<
sqlpp11/sqlpp11.h
>
#
include
<
functional
>
#
include
<
iostream
>
#
include
<
random
>
#
include
"
BlobSample.h
"
#
include
"
make_test_connection.h
"
namespace
sql
=
sqlpp::postgresql;
const
auto
blob = model::BlobSample{};
//
This would be a great fuzzing target :-)
constexpr
size_t
blob_size =
1000
*
1000ul
;
constexpr
size_t
blob_small_size =
999
;
void
verify_blob
(sql::connection& db,
const
std::vector<
uint8_t
>& expected,
uint64_t
id)
{
auto
result =
db
(
select
(blob.
data
).
from
(blob).
where
(blob.
id
== id));
const
auto
& result_row = result.
front
();
if
(!result_row.
data
)
throw
std::runtime_error
(
"
blob data is unpexpectedly NULL for id
"
+
std::to_string
(id));
const
auto
received = *result_row.
data
;
if
(expected.
size
() != received.
size
())
{
std::cerr <<
"
Size mismatch
"
<< std::endl;
throw
std::runtime_error
(
"
Size mismatch
"
+
std::to_string
(expected.
size
()) +
"
!=
"
+
std::to_string
(received.
size
()));
}
std::cerr <<
"
Verifying content
"
<< std::endl;
for
(
size_t
i =
0
; i < expected.
size
(); i++)
{
if
(expected[i] != received[i])
{
std::cerr <<
"
expected[
"
<< i <<
"
]
"
<<
static_cast
<
int
>(expected[i]) <<
"
!= received
"
<<
static_cast
<
int
>(received[i])
<< std::endl;
throw
std::runtime_error
(
"
Content mismatch
"
);
}
}
}
int
Blob
(
int
,
char
*[])
{
sql::connection db =
sql::make_test_connection
();
db.
execute
(
R"(
DROP TABLE IF EXISTS blob_sample;
)"
);
db.
execute
(
R"(
CREATE TABLE blob_sample (
id bigserial PRIMARY KEY,
data bytea
)
)"
);
std::cerr <<
"
Generating data
"
<< blob_size << std::endl;
std::vector<
uint8_t
>
data
(blob_size);
std::uniform_int_distribution<
unsigned
short
>
distribution
(
0
,
255
);
std::mt19937 engine;
auto
generator =
std::bind
(distribution, engine);
std::generate_n
(data.
begin
(), blob_size, generator);
std::vector<
uint8_t
>
data_smaller
(blob_small_size);
std::generate_n
(data_smaller.
begin
(), blob_small_size, generator);
db
(
insert_into
(blob).
set
(blob.
data
= data_smaller));
const
auto
id = db.
last_insert_id
(
"
blob_sample
"
,
"
id
"
);
verify_blob
(db, data_smaller, id);
auto
prepared_insert = db.
prepare
(
insert_into
(blob).
set
(blob.
data
=
parameter
(blob.
data
)));
prepared_insert.
params
.
data
= data;
db
(prepared_insert);
const
auto
prep_id = db.
last_insert_id
(
"
blob_sample
"
,
"
id
"
);
prepared_insert.
params
.
data
.
set_null
();
db
(prepared_insert);
const
auto
null_id = db.
last_insert_id
(
"
blob_sample
"
,
"
id
"
);
verify_blob
(db, data, prep_id);
{
auto
result =
db
(
select
(blob.
data
).
from
(blob).
where
(blob.
id
== null_id));
const
auto
& result_row = result.
front
();
std::cerr <<
"
Null blob is_null:
\t
"
<< std::boolalpha << (result_row.
data
== sqlpp::
nullopt
) << std::endl;
if
(result_row.
data
.
has_value
())
{
throw
std::runtime_error
(
"
Expected NULL blob has value
"
);
}
}
return
0
;
}
Back
|
FazBrowse Home
|
New Git URL