test_dcat.py 2.33 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
# Copyright 2018 Markus Scheidgen
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an"AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import pytest
from datetime import datetime

18
from nomad import infrastructure, config
19
from nomad.datamodel import EntryMetadata
Markus Scheidgen's avatar
Markus Scheidgen committed
20
from nomad.app.dcat.mapping import Mapping
21
22
23
24

from tests.conftest import clear_elastic
from tests.app.test_app import BlueprintClient

Maja-Olivia Lenz's avatar
Maja-Olivia Lenz committed
25

26
27
28
29
30
31
@pytest.fixture(scope='session')
def api(session_client):
    return BlueprintClient(session_client, '/dcat')


@pytest.fixture(scope='module')
Maja-Olivia Lenz's avatar
Maja-Olivia Lenz committed
32
def example_entry(test_user, other_test_user):
33
34
35
36
37
38
39
40
41
42
43
44

    entry = EntryMetadata(
        calc_id='test-id',
        upload_id='upload-id',
        upload_time=datetime.now(),
        last_processing=datetime.now(),
        uploader=test_user,
        coauthors=[other_test_user],
        comment='this is a calculation comment',
        formula='H20',
        published=True)

Markus Scheidgen's avatar
Markus Scheidgen committed
45
    yield entry
46
47


Markus Scheidgen's avatar
Markus Scheidgen committed
48
49
50
51
def test_mapping(example_entry):
    mapping = Mapping()
    mapping.map_entry(example_entry)
    assert mapping.g is not None
Maja-Olivia Lenz's avatar
Maja-Olivia Lenz committed
52
    # print(mapping.g.serialize(format='ttl').decode('utf-8'))
Markus Scheidgen's avatar
Markus Scheidgen committed
53
54


Maja-Olivia Lenz's avatar
Maja-Olivia Lenz committed
55
56
57
def test_get_dataset(elastic_infra, api, example_entry):
    clear_elastic(elastic_infra)
    example_entry.a_elastic.index()
58
59
60
    calc_id = 'test-id'
    rv = api.get('/datasets/%s' % calc_id)
    assert rv.status_code == 200
Maja-Olivia Lenz's avatar
Maja-Olivia Lenz committed
61
62

    clear_elastic(elastic_infra)
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79


def test_get_catalog(elastic_infra, api, example_entry):
    clear_elastic(elastic_infra)

    for i in range(1, 11):
        example_entry.calc_id = 'test-id-%d' % i
        example_entry.upload_time = datetime(2000, 1, 1)
        example_entry.last_processing = datetime(2020, 1, i)
        example_entry.a_elastic.index()

    infrastructure.elastic_client.indices.refresh(index=config.elastic.index_name)

    rv = api.get('/catalog/?after=test-id-3&modified_since=2020-01-07&format=nt')
    assert rv.status_code == 200

    clear_elastic(elastic_infra)