@@ -148,7 +148,8 @@ def test_hive_arguments(self):
148
148
{ "database" : "db1", "tableName" : "table1" }
149
149
],
150
150
"dryRun" : false,
151
- "replicateImpalaMetadata" : true
151
+ "replicateImpalaMetadata" : true,
152
+ "numThreads" : 4
152
153
}'''
153
154
args = utils .deserialize (RAW , ApiHiveReplicationArguments )
154
155
self .assertEquals ('vst2' , args .sourceService .peerName )
@@ -163,6 +164,7 @@ def test_hive_arguments(self):
163
164
self .assertEquals ("db1" , args .tableFilters [0 ].database )
164
165
self .assertEquals ("table1" , args .tableFilters [0 ].tableName )
165
166
self .assertTrue (args .replicateImpalaMetadata )
167
+ self .assertEquals (4 , args .numThreads )
166
168
167
169
def test_hive_cloud_arguments (self ):
168
170
RAW = '''{
@@ -239,7 +241,20 @@ def test_hive_results(self):
239
241
},
240
242
"dryRun" : false,
241
243
"runAsUser" : "systest",
242
- "runOnSourceAsUser" : "hdfs"
244
+ "runOnSourceAsUser" : "hdfs",
245
+ "statsAvailable" : true,
246
+ "dbProcessed" : 1,
247
+ "tableProcessed" : 10,
248
+ "partitionProcessed" : 100,
249
+ "functionProcessed" : 1,
250
+ "indexProcessed" : 1,
251
+ "statsProcessed" : 100,
252
+ "dbExpected" : 10,
253
+ "tableExpected" : 100,
254
+ "partitionExpected" : 1000,
255
+ "functionExpected" : 10,
256
+ "indexExpected" : 10,
257
+ "statsExpected" : 1000
243
258
}'''
244
259
res = utils .deserialize (RAW , ApiHiveReplicationResult )
245
260
self .assertEquals ('EXPORT' , res .phase )
@@ -266,6 +281,19 @@ def test_hive_results(self):
266
281
self .assertFalse (res .dryRun )
267
282
self .assertEquals (res .runAsUser , 'systest' )
268
283
self .assertEquals (res .runOnSourceAsUser , 'hdfs' )
284
+ self .assertTrue (res .statsAvailable )
285
+ self .assertEquals (1 , res .dbProcessed )
286
+ self .assertEquals (10 , res .tableProcessed )
287
+ self .assertEquals (100 , res .partitionProcessed )
288
+ self .assertEquals (1 , res .functionProcessed )
289
+ self .assertEquals (1 , res .indexProcessed )
290
+ self .assertEquals (100 , res .statsProcessed )
291
+ self .assertEquals (10 , res .dbExpected )
292
+ self .assertEquals (100 , res .tableExpected )
293
+ self .assertEquals (1000 , res .partitionExpected )
294
+ self .assertEquals (10 , res .functionExpected )
295
+ self .assertEquals (10 , res .indexExpected )
296
+ self .assertEquals (1000 , res .statsExpected )
269
297
270
298
def test_schedule (self ):
271
299
RAW = '''{
0 commit comments