def test_bulk_import():
client = mock.MagicMock()
bulk_import = models.BulkImport(
client,
name="name",
database="database",
table="table",
status="status",
upload_frozen="upload_frozen",
job_id="job_id",
valid_records="valid_records",
error_records="error_records",
valid_parts="valid_parts",
error_parts="error_parts",
)
assert bulk_import.name == "name"
assert bulk_import.database == "database"
assert bulk_import.table == "table"
assert bulk_import.status == "status"
assert bulk_import.job_id == "job_id"
assert bulk_import.valid_records == "valid_records"
assert bulk_import.error_records == "error_records"
assert bulk_import.valid_parts == "valid_parts"
assert bulk_import.error_parts == "error_parts"
assert bulk_import.upload_frozen == "upload_frozen"
def bulk_imports(self):
"""List bulk import sessions
Returns:
a list of :class:`tdclient.models.BulkImport`
"""
return [
models.BulkImport(self, **data) for data in self.api.list_bulk_imports()
]
def bulk_import(self, name):
"""Get a bulk import session
Params:
name (str): name of a bulk import session
Returns: :class:`tdclient.models.BulkImport`
"""
data = self.api.show_bulk_import(name)
return models.BulkImport(self, **data)
def create_bulk_import(self, name, database, table, params=None):
"""Create new bulk import session
Args:
name (str): name of new bulk import session
database (str): name of a database
table (str): name of a table
Returns:
:class:`tdclient.models.BulkImport`
"""
params = {} if params is None else params
self.api.create_bulk_import(name, database, table, params)
return models.BulkImport(self, name=name, database=database, table=table)
def test_bulk_import_delete():
client = mock.MagicMock()
bulk_import = models.BulkImport(client,
name="name",
database="database",
table="table",
status="status",
upload_frozen="upload_frozen",
job_id="job_id",
valid_records="valid_records",
error_records="error_records",
valid_parts="valid_parts",
error_parts="error_parts")
bulk_import.delete()
client.delete_bulk_import.assert_called_with("name")
def test_bulk_import_upload_part():
client = mock.MagicMock()
bulk_import = models.BulkImport(client,
name="name",
database="database",
table="table",
status="status",
upload_frozen="upload_frozen",
job_id="job_id",
valid_records="valid_records",
error_records="error_records",
valid_parts="valid_parts",
error_parts="error_parts")
bulk_import.update = mock.MagicMock()
bulk_import.upload_part("part", b"bytes", 5)
client.bulk_import_upload_part.assert_called_with("name", "part", b"bytes",
5)
assert bulk_import.update.called
