@@ -109,7 +109,7 @@ def test_replace
109109 task = Bigquery . configure ( config , schema , processor_count )
110110 any_instance_of ( BigqueryClient ) do |obj |
111111 mock ( obj ) . get_dataset ( config [ 'dataset' ] )
112- mock ( obj ) . create_table_if_not_exists ( config [ 'temp_table' ] )
112+ mock ( obj ) . create_table_if_not_exists ( config [ 'temp_table' ] , { :options => { "expiration_time" => nil } } )
113113 mock ( obj ) . create_table_if_not_exists ( config [ 'table' ] )
114114 mock ( obj ) . copy ( config [ 'temp_table' ] , config [ 'table' ] , write_disposition : 'WRITE_TRUNCATE' )
115115 mock ( obj ) . delete_table ( config [ 'temp_table' ] )
@@ -122,7 +122,7 @@ def test_replace_with_partitioning
122122 task = Bigquery . configure ( config , schema , processor_count )
123123 any_instance_of ( BigqueryClient ) do |obj |
124124 mock ( obj ) . get_dataset ( config [ 'dataset' ] )
125- mock ( obj ) . create_table_if_not_exists ( config [ 'temp_table' ] )
125+ mock ( obj ) . create_table_if_not_exists ( config [ 'temp_table' ] , { :options => { "expiration_time" => nil } } )
126126 mock ( obj ) . create_table_if_not_exists ( config [ 'table' ] )
127127 mock ( obj ) . copy ( config [ 'temp_table' ] , config [ 'table' ] , write_disposition : 'WRITE_TRUNCATE' )
128128 mock ( obj ) . delete_table ( config [ 'temp_table' ] )
@@ -138,7 +138,7 @@ def test_replace_backup
138138 any_instance_of ( BigqueryClient ) do |obj |
139139 mock ( obj ) . get_dataset ( config [ 'dataset' ] )
140140 mock ( obj ) . get_dataset ( config [ 'dataset_old' ] )
141- mock ( obj ) . create_table_if_not_exists ( config [ 'temp_table' ] )
141+ mock ( obj ) . create_table_if_not_exists ( config [ 'temp_table' ] , { :options => { "expiration_time" => nil } } )
142142 mock ( obj ) . create_table_if_not_exists ( config [ 'table' ] )
143143 mock ( obj ) . create_table_if_not_exists ( config [ 'table_old' ] , dataset : config [ 'dataset_old' ] )
144144
@@ -158,7 +158,7 @@ def test_replace_backup_auto_create_dataset
158158 mock ( obj ) . create_dataset ( config [ 'dataset' ] )
159159 mock ( obj ) . create_dataset ( config [ 'dataset_old' ] , reference : config [ 'dataset' ] )
160160 mock ( obj ) . create_table_if_not_exists ( config [ 'table' ] )
161- mock ( obj ) . create_table_if_not_exists ( config [ 'temp_table' ] )
161+ mock ( obj ) . create_table_if_not_exists ( config [ 'temp_table' ] , { :options => { "expiration_time" => nil } } )
162162 mock ( obj ) . create_table_if_not_exists ( config [ 'table_old' ] , dataset : config [ 'dataset_old' ] )
163163
164164 mock ( obj ) . get_table_or_partition ( config [ 'table' ] )
@@ -176,7 +176,7 @@ def test_replace_backup_with_partitioning
176176 any_instance_of ( BigqueryClient ) do |obj |
177177 mock ( obj ) . get_dataset ( config [ 'dataset' ] )
178178 mock ( obj ) . get_dataset ( config [ 'dataset_old' ] )
179- mock ( obj ) . create_table_if_not_exists ( config [ 'temp_table' ] )
179+ mock ( obj ) . create_table_if_not_exists ( config [ 'temp_table' ] , { :options => { "expiration_time" => nil } } )
180180 mock ( obj ) . create_table_if_not_exists ( config [ 'table' ] )
181181 mock ( obj ) . create_table_if_not_exists ( config [ 'table_old' ] , dataset : config [ 'dataset_old' ] )
182182
@@ -196,7 +196,7 @@ def test_append
196196 task = Bigquery . configure ( config , schema , processor_count )
197197 any_instance_of ( BigqueryClient ) do |obj |
198198 mock ( obj ) . get_dataset ( config [ 'dataset' ] )
199- mock ( obj ) . create_table_if_not_exists ( config [ 'temp_table' ] )
199+ mock ( obj ) . create_table_if_not_exists ( config [ 'temp_table' ] , { :options => { "expiration_time" => nil } } )
200200 mock ( obj ) . create_table_if_not_exists ( config [ 'table' ] )
201201 mock ( obj ) . copy ( config [ 'temp_table' ] , config [ 'table' ] , write_disposition : 'WRITE_APPEND' )
202202 mock ( obj ) . delete_table ( config [ 'temp_table' ] )
@@ -209,7 +209,7 @@ def test_append_with_partitioning
209209 task = Bigquery . configure ( config , schema , processor_count )
210210 any_instance_of ( BigqueryClient ) do |obj |
211211 mock ( obj ) . get_dataset ( config [ 'dataset' ] )
212- mock ( obj ) . create_table_if_not_exists ( config [ 'temp_table' ] )
212+ mock ( obj ) . create_table_if_not_exists ( config [ 'temp_table' ] , { :options => { "expiration_time" => nil } } )
213213 mock ( obj ) . create_table_if_not_exists ( config [ 'table' ] )
214214 mock ( obj ) . copy ( config [ 'temp_table' ] , config [ 'table' ] , write_disposition : 'WRITE_APPEND' )
215215 mock ( obj ) . delete_table ( config [ 'temp_table' ] )
0 commit comments