def test_bulk_item_insert(
postgres_core: CoreCrudClient,
postgres_transactions: TransactionsClient,
postgres_bulk_transactions: BulkTransactionsClient,
load_test_data: Callable,
):
coll = load_test_data("test_collection.json")
postgres_transactions.create_collection(coll, request=MockStarletteRequest)
item = load_test_data("test_item.json")
items = []
for _ in range(10):
_item = deepcopy(item)
_item["id"] = str(uuid.uuid4())
items.append(_item)
fc = postgres_core.item_collection(coll["id"],
request=MockStarletteRequest)
assert len(fc["features"]) == 0
postgres_bulk_transactions.bulk_item_insert(items=items)
fc = postgres_core.item_collection(coll["id"],
request=MockStarletteRequest)
assert len(fc["features"]) == 10
for item in items:
postgres_transactions.delete_item(item["id"],
item["collection"],
request=MockStarletteRequest)
def test_bulk_item_insert(
postgres_transactions: TransactionsClient,
postgres_bulk_transactions: BulkTransactionsClient,
load_test_data: Callable,
):
coll = Collection.parse_obj(load_test_data("test_collection.json"))
postgres_transactions.create_collection(coll, request=MockStarletteRequest)
item = Item.parse_obj(load_test_data("test_item.json"))
items = []
for _ in range(10):
_item = item.dict()
_item["id"] = str(uuid.uuid4())
items.append(_item)
postgres_bulk_transactions.bulk_item_insert(Items(items=items))
for item in items:
postgres_transactions.delete_item(item["id"],
request=MockStarletteRequest)
def test_bulk_item_insert_chunked(
postgres_transactions: TransactionsClient,
postgres_bulk_transactions: BulkTransactionsClient,
load_test_data: Callable,
):
coll = load_test_data("test_collection.json")
postgres_transactions.create_collection(coll, request=MockStarletteRequest)
item = load_test_data("test_item.json")
items = []
for _ in range(10):
_item = deepcopy(item)
_item["id"] = str(uuid.uuid4())
items.append(_item)
postgres_bulk_transactions.bulk_item_insert(items=items, chunk_size=2)
for item in items:
postgres_transactions.delete_item(item["id"],
item["collection"],
request=MockStarletteRequest)
from stac_fastapi.extensions.core import (
FieldsExtension,
QueryExtension,
SortExtension,
TransactionExtension,
)
from stac_fastapi.extensions.third_party import BulkTransactionExtension
from stac_fastapi.sqlalchemy.config import SqlalchemySettings
from stac_fastapi.sqlalchemy.core import CoreCrudClient
from stac_fastapi.sqlalchemy.session import Session
from stac_fastapi.sqlalchemy.transactions import (
BulkTransactionsClient,
TransactionsClient,
)
settings = SqlalchemySettings()
session = Session.create_from_settings(settings)
api = StacApi(
settings=settings,
extensions=[
TransactionExtension(client=TransactionsClient(session=session)),
BulkTransactionExtension(client=BulkTransactionsClient(
session=session)),
FieldsExtension(),
QueryExtension(),
SortExtension(),
],
client=CoreCrudClient(session=session),
)
app = api.app
def postgres_bulk_transactions(db_session):
return BulkTransactionsClient(session=db_session)