git.iko.soy/reshape
on main help sign in
reshape/tests/remove_column.rs
217 lines 5.2 kB history raw
1mod common;
2use common::Test;
4#[test]
5fn remove_column() {
6 let mut test = Test::new("Remove column");
8 test.first_migration(
9 r#"
10 name = "create_user_table"
12 [[actions]]
13 type = "create_table"
14 name = "users"
15 primary_key = ["id"]
17 [[actions.columns]]
18 name = "id"
19 type = "INTEGER"
21 [[actions.columns]]
22 name = "name"
23 type = "TEXT"
24 "#,
25 );
27 test.second_migration(
28 r#"
29 name = "remove_name_column"
31 [[actions]]
32 type = "remove_column"
33 table = "users"
34 column = "name"
35 down = "'TEST_DOWN_VALUE'"
36 "#,
37 );
39 test.intermediate(|old_db, new_db| {
40 // Insert using old schema and ensure it can be retrieved through new schema
41 old_db
42 .simple_query("INSERT INTO users(id, name) VALUES (1, 'John Doe')")
43 .unwrap();
44 let results = new_db
45 .query("SELECT id FROM users WHERE id = 1", &[])
46 .unwrap();
47 assert_eq!(1, results.len());
48 assert_eq!(1, results[0].get::<_, i32>("id"));
50 // Ensure the name column is not accesible through the new schema
51 assert!(new_db.query("SELECT id, name FROM users", &[]).is_err());
53 // Insert using new schema and ensure the down function is correctly applied
54 new_db
55 .simple_query("INSERT INTO users(id) VALUES (2)")
56 .unwrap();
57 let result = old_db
58 .query_opt("SELECT name FROM users WHERE id = 2", &[])
59 .unwrap();
60 assert_eq!(
61 Some("TEST_DOWN_VALUE"),
62 result.as_ref().map(|row| row.get("name"))
63 );
64 });
66 test.run();
69#[test]
70fn remove_column_with_index() {
71 let mut test = Test::new("Remove column");
73 test.first_migration(
74 r#"
75 name = "create_user_table"
77 [[actions]]
78 type = "create_table"
79 name = "users"
80 primary_key = ["id"]
82 [[actions.columns]]
83 name = "id"
84 type = "INTEGER"
86 [[actions.columns]]
87 name = "name"
88 type = "TEXT"
90 [[actions]]
91 type = "add_index"
92 table = "users"
94 [actions.index]
95 name = "name_idx"
96 columns = ["name"]
97 "#,
98 );
100 test.second_migration(
101 r#"
102 name = "remove_name_column"
104 [[actions]]
105 type = "remove_column"
106 table = "users"
107 column = "name"
108 down = "'TEST_DOWN_VALUE'"
109 "#,
110 );
112 test.after_completion(|db| {
113 // Ensure index has been removed after the migration is complete
114 let count: i64 = db
115 .query(
116 "
117 SELECT COUNT(*)
118 FROM pg_catalog.pg_index
119 JOIN pg_catalog.pg_class ON pg_index.indexrelid = pg_class.oid
120 WHERE pg_class.relname = 'name_idx'
121 ",
122 &[],
123 )
124 .unwrap()
125 .first()
126 .map(|row| row.get(0))
127 .unwrap();
129 assert_eq!(0, count, "expected index to not exist");
130 });
132 test.run();
135#[test]
136fn remove_column_with_complex_down() {
137 let mut test = Test::new("Remove column complex");
139 test.first_migration(
140 r#"
141 name = "create_tables"
143 [[actions]]
144 type = "create_table"
145 name = "users"
146 primary_key = ["id"]
148 [[actions.columns]]
149 name = "id"
150 type = "INTEGER"
152 [[actions.columns]]
153 name = "email"
154 type = "TEXT"
156 [[actions]]
157 type = "create_table"
158 name = "profiles"
159 primary_key = ["user_id"]
161 [[actions.columns]]
162 name = "user_id"
163 type = "INTEGER"
165 [[actions.columns]]
166 name = "email"
167 type = "TEXT"
168 "#,
169 );
171 test.second_migration(
172 r#"
173 name = "remove_users_email_column"
175 [[actions]]
176 type = "remove_column"
177 table = "users"
178 column = "email"
180 [actions.down]
181 table = "profiles"
182 value = "profiles.email"
183 where = "users.id = profiles.user_id"
184 "#,
185 );
187 test.after_first(|db| {
188 db.simple_query("INSERT INTO users (id, email) VALUES (1, 'test@example.com')")
189 .unwrap();
190 db.simple_query("INSERT INTO profiles (user_id, email) VALUES (1, 'test@example.com')")
191 .unwrap();
192 });
194 test.intermediate(|old_db, new_db| {
195 new_db
196 .simple_query("UPDATE profiles SET email = 'test2@example.com' WHERE user_id = 1")
197 .unwrap();
199 // Ensure new email was propagated to users table in old schema
200 let email: String = old_db
201 .query(
202 "
203 SELECT email
204 FROM users
205 WHERE id = 1
206 ",
207 &[],
208 )
209 .unwrap()
210 .first()
211 .map(|row| row.get("email"))
212 .unwrap();
213 assert_eq!("test2@example.com", email);
214 });
216 test.run();