@@ -91,6 +91,7 @@ def test_delete_in_advance
91
91
mock ( obj ) . get_dataset ( config [ 'dataset' ] )
92
92
mock ( obj ) . delete_table_or_partition ( config [ 'table' ] )
93
93
mock ( obj ) . create_table_if_not_exists ( config [ 'table' ] )
94
+ mock ( obj ) . patch_table
94
95
end
95
96
Bigquery . transaction ( config , schema , processor_count , &control )
96
97
end
@@ -102,6 +103,7 @@ def test_delete_in_advance_with_partitioning
102
103
mock ( obj ) . get_dataset ( config [ 'dataset' ] )
103
104
mock ( obj ) . delete_table_or_partition ( config [ 'table' ] )
104
105
mock ( obj ) . create_table_if_not_exists ( config [ 'table' ] )
106
+ mock ( obj ) . patch_table
105
107
end
106
108
Bigquery . transaction ( config , schema , processor_count , &control )
107
109
end
@@ -117,6 +119,7 @@ def test_replace
117
119
mock ( obj ) . create_table_if_not_exists ( config [ 'table' ] )
118
120
mock ( obj ) . copy ( config [ 'temp_table' ] , config [ 'table' ] , write_disposition : 'WRITE_TRUNCATE' )
119
121
mock ( obj ) . delete_table ( config [ 'temp_table' ] )
122
+ mock ( obj ) . patch_table
120
123
end
121
124
Bigquery . transaction ( config , schema , processor_count , &control )
122
125
end
@@ -130,6 +133,7 @@ def test_replace_with_partitioning
130
133
mock ( obj ) . create_table_if_not_exists ( config [ 'table' ] )
131
134
mock ( obj ) . copy ( config [ 'temp_table' ] , config [ 'table' ] , write_disposition : 'WRITE_TRUNCATE' )
132
135
mock ( obj ) . delete_table ( config [ 'temp_table' ] )
136
+ mock ( obj ) . patch_table
133
137
end
134
138
Bigquery . transaction ( config , schema , processor_count , &control )
135
139
end
@@ -151,6 +155,7 @@ def test_replace_backup
151
155
152
156
mock ( obj ) . copy ( config [ 'temp_table' ] , config [ 'table' ] , write_disposition : 'WRITE_TRUNCATE' )
153
157
mock ( obj ) . delete_table ( config [ 'temp_table' ] )
158
+ mock ( obj ) . patch_table
154
159
end
155
160
Bigquery . transaction ( config , schema , processor_count , &control )
156
161
end
@@ -170,6 +175,7 @@ def test_replace_backup_auto_create_dataset
170
175
171
176
mock ( obj ) . copy ( config [ 'temp_table' ] , config [ 'table' ] , write_disposition : 'WRITE_TRUNCATE' )
172
177
mock ( obj ) . delete_table ( config [ 'temp_table' ] )
178
+ mock ( obj ) . patch_table
173
179
end
174
180
Bigquery . transaction ( config , schema , processor_count , &control )
175
181
end
@@ -189,6 +195,7 @@ def test_replace_backup_with_partitioning
189
195
190
196
mock ( obj ) . copy ( config [ 'temp_table' ] , config [ 'table' ] , write_disposition : 'WRITE_TRUNCATE' )
191
197
mock ( obj ) . delete_table ( config [ 'temp_table' ] )
198
+ mock ( obj ) . patch_table
192
199
end
193
200
Bigquery . transaction ( config , schema , processor_count , &control )
194
201
end
0 commit comments