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