Esempio n. 1
0
def pseudo_ls8_dataset4(index: Index, initialised_postgres_db: PostgresDb,
                        pseudo_ls8_type: DatasetType,
                        pseudo_ls8_dataset2: Dataset) -> Dataset:
    # Same as 2, but a different path/row
    id_ = str(uuid.uuid4())
    dataset_doc = copy.deepcopy(pseudo_ls8_dataset2.metadata_doc)
    dataset_doc['id'] = id_
    dataset_doc['image'] = {
        'satellite_ref_point_start': {
            'x': 116,
            'y': 85
        },
        'satellite_ref_point_end': {
            'x': 116,
            'y': 87
        },
    }

    with initialised_postgres_db.connect() as connection:
        was_inserted = connection.insert_dataset(dataset_doc, id_,
                                                 pseudo_ls8_type.id)
        assert was_inserted
        d = index.datasets.get(id_)
        # The dataset should have been matched to the telemetry type.
        assert d.type.id == pseudo_ls8_type.id
        return d
Esempio n. 2
0
def test_index_duplicate_dataset(index: Index, initialised_postgres_db: PostgresDb,
                                 local_config,
                                 default_metadata_type)->None:
    dataset_type = index.products.add_document(_pseudo_telemetry_dataset_type)
    assert not index.datasets.has(_telemetry_uuid)

    with initialised_postgres_db.begin() as transaction:
        was_inserted = transaction.insert_dataset(
            _telemetry_dataset,
            _telemetry_uuid,
            dataset_type.id
        )

    assert was_inserted
    assert index.datasets.has(_telemetry_uuid)

    # Insert again.
    with initialised_postgres_db.connect() as connection:
        was_inserted = connection.insert_dataset(
            _telemetry_dataset,
            _telemetry_uuid,
            dataset_type.id
        )
        assert was_inserted is False

    assert index.datasets.has(_telemetry_uuid)