Skip to content

Commit 75ed9b0

Browse files
committed
Remove the batch_insert function from tests
We support batch insertion on all backends now.
1 parent 6428f4d commit 75ed9b0

11 files changed

Lines changed: 35 additions & 44 deletions

File tree

diesel/src/query_builder/insert_statement.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ impl<T, Op> IncompleteInsertStatement<T, Op> {
3434
}
3535
}
3636

37-
pub trait IntoInsertStatement<Tab, Op=Insert, Ret=NoReturningClause> {
37+
pub trait IntoInsertStatement<Tab, Op, Ret> {
3838
type InsertStatement;
3939

4040
fn into_insert_statement(self, target: Tab, operator: Op, returning: Ret)

diesel_tests/tests/associations.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -142,11 +142,11 @@ fn conn_with_test_data() -> (TestConnection, User, User, User) {
142142
let tess = find_user_by_name("Tess", &connection);
143143
let jim = find_user_by_name("Jim", &connection);
144144
let new_posts = vec![sean.new_post("Hello", None), sean.new_post("World", None)];
145-
batch_insert(&new_posts, posts::table, &connection);
145+
insert(&new_posts).into(posts::table).execute(&connection).unwrap();
146146
let new_posts = vec![tess.new_post("Hello 2", None), tess.new_post("World 2", None)];
147-
batch_insert(&new_posts, posts::table, &connection);
147+
insert(&new_posts).into(posts::table).execute(&connection).unwrap();
148148
let new_posts = vec![jim.new_post("Hello 3", None), jim.new_post("World 3", None)];
149-
batch_insert(&new_posts, posts::table, &connection);
149+
insert(&new_posts).into(posts::table).execute(&connection).unwrap();
150150

151151
(connection, sean, tess, jim)
152152
}

diesel_tests/tests/bench.rs

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ use self::schema::{
1313
users, posts, comments,
1414
NewUser, NewPost, NewComment,
1515
User, Post, Comment,
16-
TestConnection, batch_insert,
16+
TestConnection,
1717
};
1818
use diesel::*;
1919

@@ -40,7 +40,7 @@ macro_rules! bench_trivial_query {
4040
let data: Vec<_> = (0..$n).map(|i| {
4141
NewUser::new(&format!("User {}", i), None)
4242
}).collect();
43-
batch_insert(&data, users::table, &conn);
43+
insert(&data).into(users::table).execute(&conn).unwrap();
4444

4545
b.iter(|| {
4646
users::table.load::<User>(&conn).unwrap()
@@ -54,7 +54,7 @@ macro_rules! bench_trivial_query {
5454
let data: Vec<_> = (0..$n).map(|i| {
5555
NewUser::new(&format!("User {}", i), None)
5656
}).collect();
57-
batch_insert(&data, users::table, &conn);
57+
insert(&data).into(users::table).execute(&conn).unwrap();
5858

5959
b.iter(|| {
6060
users::table.into_boxed().load::<User>(&conn).unwrap()
@@ -86,7 +86,7 @@ macro_rules! bench_medium_complex_query {
8686
let hair_color = if i % 2 == 0 { "black" } else { "brown" };
8787
NewUser::new(&format!("User {}", i), Some(hair_color))
8888
}).collect();
89-
batch_insert(&data, users::table, &conn);
89+
insert(&data).into(users::table).execute(&conn).unwrap();
9090

9191
b.iter(|| {
9292
use schema::users::dsl::*;
@@ -105,7 +105,7 @@ macro_rules! bench_medium_complex_query {
105105
let hair_color = if i % 2 == 0 { "black" } else { "brown" };
106106
NewUser::new(&format!("User {}", i), Some(hair_color))
107107
}).collect();
108-
batch_insert(&data, users::table, &conn);
108+
insert(&data).into(users::table).execute(&conn).unwrap();
109109

110110
b.iter(|| {
111111
use schema::users::dsl::*;
@@ -140,7 +140,7 @@ fn loading_associations_sequentially(b: &mut Bencher) {
140140
let hair_color = if i % 2 == 0 { "black" } else { "brown" };
141141
NewUser::new(&format!("User {}", i), Some(hair_color))
142142
}).collect();
143-
batch_insert(&data, users::table, &conn);
143+
insert(&data).into(users::table).execute(&conn).unwrap();
144144
let all_users = users::table.load::<User>(&conn).unwrap();
145145
let data: Vec<_> = all_users.iter().flat_map(|user| {
146146
let user_id = user.id;
@@ -149,7 +149,7 @@ fn loading_associations_sequentially(b: &mut Bencher) {
149149
NewPost::new(user_id, &title, None)
150150
})
151151
}).collect();
152-
batch_insert(&data, posts::table, &conn);
152+
insert(&data).into(posts::table).execute(&conn).unwrap();
153153
let all_posts = posts::table.load::<Post>(&conn).unwrap();
154154
let data: Vec<_> = all_posts.iter().flat_map(|post| {
155155
let post_id = post.id;
@@ -161,7 +161,7 @@ fn loading_associations_sequentially(b: &mut Bencher) {
161161
let comment_data: Vec<_> = data.iter().map(|&(ref title, post_id)| {
162162
NewComment(post_id, &title)
163163
}).collect();
164-
batch_insert(&comment_data, comments::table, &conn);
164+
insert(&comment_data).into(comments::table).execute(&conn).unwrap();
165165

166166
// ACTUAL BENCHMARK
167167
b.iter(|| {

diesel_tests/tests/expressions/mod.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
mod date_and_time;
22
mod ops;
33

4-
use schema::{connection, NewUser, batch_insert, connection_with_sean_and_tess_in_users_table};
4+
use schema::{connection, NewUser, connection_with_sean_and_tess_in_users_table};
55
use schema::users::dsl::*;
66
use diesel::*;
77
use diesel::backend::Backend;
@@ -79,7 +79,7 @@ fn max_returns_same_type_as_expression_being_maximized() {
7979
NewUser::new("C", None),
8080
NewUser::new("A", None),
8181
];
82-
batch_insert(data, users, &connection);
82+
insert(data).into(users).execute(&connection).unwrap();
8383
assert_eq!(Ok("C".to_string()), source.first(&connection));
8484
connection.execute("DELETE FROM users WHERE name = 'C'").unwrap();
8585
assert_eq!(Ok("B".to_string()), source.first(&connection));
@@ -161,7 +161,7 @@ fn function_with_multiple_arguments() {
161161

162162
let connection = connection();
163163
let new_users = vec![NewUser::new("Sean", Some("black")), NewUser::new("Tess", None)];
164-
batch_insert(&new_users, users, &connection);
164+
insert(&new_users).into(users).execute(&connection).unwrap();
165165

166166
let expected_data = vec!["black".to_string(), "Tess".to_string()];
167167
let data = users.select(coalesce(hair_color, name))

diesel_tests/tests/filter.rs

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ fn filter_by_equality_on_nullable_columns() {
4040
NewUser::new("Tess", Some("brown")),
4141
NewUser::new("Jim", Some("black")),
4242
];
43-
batch_insert(&data, users, &connection);
43+
insert(&data).into(users).execute(&connection).unwrap();
4444

4545
let data = users.load::<User>(&connection).unwrap();
4646
let sean = data[0].clone();
@@ -62,7 +62,7 @@ fn filter_by_is_not_null_on_nullable_columns() {
6262
NewUser::new("Derek", Some("red")),
6363
NewUser::new("Gordon", None),
6464
];
65-
batch_insert(&data, users, &connection);
65+
insert(&data).into(users).execute(&connection).unwrap();
6666
let data = users.load::<User>(&connection).unwrap();
6767
let derek = data[0].clone();
6868

@@ -79,7 +79,7 @@ fn filter_by_is_null_on_nullable_columns() {
7979
NewUser::new("Derek", Some("red")),
8080
NewUser::new("Gordon", None),
8181
];
82-
batch_insert(&data, users, &connection);
82+
insert(&data).into(users).execute(&connection).unwrap();
8383
let data = users.load::<User>(&connection).unwrap();
8484
let gordon = data[1].clone();
8585

@@ -129,7 +129,7 @@ fn filter_then_select() {
129129

130130
let connection = connection();
131131
let data = vec![NewUser::new("Sean", None), NewUser::new("Tess", None)];
132-
batch_insert(&data, users, &connection);
132+
insert(&data).into(users).execute(&connection).unwrap();
133133

134134
assert_eq!(Ok("Sean".to_string()),
135135
users.filter(name.eq("Sean")).select(name).first(&connection));
@@ -151,7 +151,7 @@ fn filter_on_multiple_columns() {
151151
NewUser::new("Tess", Some("black")),
152152
NewUser::new("Tess", Some("brown")),
153153
];
154-
batch_insert(data, users, &connection);
154+
insert(data).into(users).execute(&connection).unwrap();
155155
let data = users.load::<User>(&connection).unwrap();
156156
let black_haired_sean = data[0].clone();
157157
let brown_haired_sean = data[1].clone();
@@ -183,7 +183,7 @@ fn filter_called_twice_means_same_thing_as_and() {
183183
NewUser::new("Tess", Some("black")),
184184
NewUser::new("Tess", Some("brown")),
185185
];
186-
batch_insert(data, users, &connection);
186+
insert(data).into(users).execute(&connection).unwrap();
187187
let data = users.load::<User>(&connection).unwrap();
188188
let black_haired_sean = data[0].clone();
189189
let brown_haired_sean = data[1].clone();

diesel_tests/tests/filter_operators.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ fn filter_by_like() {
9191
NewUser::new("Tess Griffin", None),
9292
NewUser::new("Jim", None),
9393
];
94-
batch_insert(&data, users, &connection);
94+
insert(&data).into(users).execute(&connection).unwrap();
9595
let data = users.load::<User>(&connection).unwrap();
9696
let sean = data[0].clone();
9797
let tess = data[1].clone();

diesel_tests/tests/find.rs

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,10 @@ fn find_with_composite_pk() {
4545
let third_following = Following { user_id: 2, post_id: 1, email_notifications: false };
4646

4747
let connection = connection();
48-
batch_insert(&[first_following, second_following, third_following], followings, &connection);
48+
insert(&vec![first_following, second_following, third_following])
49+
.into(followings)
50+
.execute(&connection)
51+
.unwrap();
4952

5053
assert_eq!(Ok(first_following), followings.find((1, 1)).first(&connection));
5154
assert_eq!(Ok(second_following), followings.find((1, 2)).first(&connection));

diesel_tests/tests/insert.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ fn insert_records() {
1010
NewUser::new("Tess", None),
1111
];
1212

13-
batch_insert(new_users, users, &connection);
13+
insert(new_users).into(users).execute(&connection).unwrap();
1414
let actual_users = users.load::<User>(&connection).unwrap();
1515

1616
let expected_users = vec![
@@ -127,7 +127,7 @@ fn insert_returning_count_returns_number_of_rows_inserted() {
127127
BaldUser { name: "Sean".to_string() },
128128
BaldUser { name: "Tess".to_string() },
129129
];
130-
let count = batch_insert(new_users, users, &connection);
130+
let count = insert(new_users).into(users).execute(&connection).unwrap();
131131
let second_count = insert(&BaldUser { name: "Guy".to_string() }).into(users).execute(&connection).unwrap();
132132

133133
assert_eq!(2, count);
@@ -154,7 +154,7 @@ fn insert_borrowed_content() {
154154
BorrowedUser { name: "Sean" },
155155
BorrowedUser { name: "Tess" },
156156
];
157-
batch_insert(new_users, users, &connection);
157+
insert(new_users).into(users).execute(&connection).unwrap();
158158

159159
let actual_users = users.load::<User>(&connection).unwrap();
160160
let expected_users = vec![

diesel_tests/tests/joins.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -220,15 +220,15 @@ fn join_through_other() {
220220
let connection = connection_with_sean_and_tess_in_users_table();
221221

222222
insert(&NewUser::new("Jim", None)).into(users).execute(&connection).unwrap();
223-
batch_insert(&vec![
223+
insert(&vec![
224224
NewPost::new(1, "Hello", None), NewPost::new(2, "World", None),
225225
NewPost::new(1, "Hello again!", None),
226-
], posts::table, &connection);
226+
]).into(posts::table).execute(&connection).unwrap();
227227
let posts = posts::table.load::<Post>(&connection).unwrap();
228-
batch_insert(&vec![
228+
insert(&vec![
229229
NewComment(posts[0].id, "OMG"), NewComment(posts[1].id, "WTF"),
230230
NewComment(posts[2].id, "Best post ever!!!")
231-
], comments::table, &connection);
231+
]).into(comments::table).execute(&connection).unwrap();
232232
let comments = comments::table.load::<Comment>(&connection).unwrap();
233233

234234
let data = users.inner_join(comments::table).load(&connection)

diesel_tests/tests/order.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ fn order_by_column() {
1111
NewUser::new("Tess", None),
1212
NewUser::new("Jim", None),
1313
];
14-
batch_insert(&data, users, &conn);
14+
insert(&data).into(users).execute(&conn).unwrap();
1515
let data = users.load::<User>(&conn).unwrap();
1616
let sean = &data[0];
1717
let tess = &data[1];
@@ -48,7 +48,7 @@ fn order_by_descending_column() {
4848
NewUser::new("Tess", None),
4949
NewUser::new("Jim", None),
5050
];
51-
batch_insert(&data, users, &conn);
51+
insert(&data).into(users).execute(&conn).unwrap();
5252
let data = users.load::<User>(&conn).unwrap();
5353
let sean = &data[0];
5454
let tess = &data[1];

0 commit comments

Comments
 (0)