def test_arg_jdbc_url():
task_config = {
'source': {
'driver': 'mysql',
'host': 'example.com',
'port': 12345,
'db': 'portal',
'user': '******',
'password': '******'
},
'hive': {
'db': 'raw_data',
'table_prefix': 'p_'
},
'map_types': {
'hive': {'VARBINARY': 'HiveType1', 'UUID': 'HiveType2'},
'java': {'VARBINARY': 'JavaType1', 'VARCHAR': 'JavaType2'}
},
'sqoop_args': {
'user': {
'direct': True,
'm': 5
}
}
}
task = TaskConfig(task_config, {})
table = Table('user', 'TABLE')
columns = [Column('id', 'VARCHAR'), Column('img', 'VARBINARY')]
cmd = SqoopCmd(task, table, columns)
assert cmd.args() == {
'connect': 'jdbc:mysql://example.com:12345/portal',
'hive-import': True,
'hive-table': 'raw_data.p_user',
'map-column-hive': 'img=HiveType1',
'map-column-java': 'id=JavaType2,img=JavaType1',
'password': '******'hadoop-rocks\'',
'table': 'user',
'username': '******',
'direct': True,
'm': 5
}
def test_args2str(k, v, exp):
assert SqoopCmd._arg2str(k, v) == exp
