Skip to content

Add export FHIR store to GCS test. #1804

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 7 commits into from
Nov 21, 2018
Merged

Add export FHIR store to GCS test. #1804

Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
39fa5d0
Add export FHIR store to GCS test.
noerog Oct 30, 2018
a895cbe
Merge branch 'master' into patch-2
engelke Nov 20, 2018
4937bab
Update import/export tests
noerog Nov 21, 2018
ce3120b
Merge branch 'master' into patch-2
noerog Nov 21, 2018
6f49de9
Merge branch 'master' into patch-2
engelke Nov 21, 2018
938d131
Merge branch 'master' into patch-2
engelke Nov 21, 2018
3517f96
Merge branch 'master' into patch-2
engelke Nov 21, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
81 changes: 80 additions & 1 deletion healthcare/api-client/fhir/fhir_stores_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,8 @@
import sys
import time

from google.cloud import storage

# Add datasets for bootstrapping datasets for testing
sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'datasets')) # noqa
import datasets
Expand All @@ -30,7 +32,12 @@
dataset_id = 'test_dataset_{}'.format(int(time.time()))
fhir_store_id = 'test_fhir_store-{}'.format(int(time.time()))
pubsub_topic = 'test_pubsub_topic-{}'.format(int(time.time()))
resource_type = 'Patient'

gcs_uri = os.environ['CLOUD_STORAGE_BUCKET']
RESOURCES = os.path.join(os.path.dirname(__file__), 'resources')
source_file_name = 'Patient.json'
resource_file = os.path.join(RESOURCES, source_file_name)
import_object = gcs_uri + '/' + source_file_name


@pytest.fixture(scope='module')
Expand Down Expand Up @@ -126,3 +133,75 @@ def test_patch_fhir_store(test_dataset, capsys):
out, _ = capsys.readouterr()

assert 'Patched FHIR store' in out


def test_import_fhir_store_gcs(test_dataset, capsys):
fhir_stores.create_fhir_store(
service_account_json,
api_key,
project_id,
cloud_region,
dataset_id,
fhir_store_id)

storage_client = storage.Client()
bucket = storage_client.get_bucket(gcs_uri)
blob = bucket.blob(source_file_name)

blob.upload_from_filename(resource_file)

fhir_stores.import_fhir_store(
service_account_json,
api_key,
project_id,
cloud_region,
dataset_id,
fhir_store_id,
import_object)

# Clean up
blob.delete()

fhir_stores.delete_fhir_store(
service_account_json,
api_key,
project_id,
cloud_region,
dataset_id,
fhir_store_id)

out, _ = capsys.readouterr()

assert 'Imported FHIR resources' in out


def test_export_fhir_store_gcs(test_dataset, capsys):
fhir_stores.create_fhir_store(
service_account_json,
api_key,
project_id,
cloud_region,
dataset_id,
fhir_store_id)

fhir_stores.export_fhir_store_gcs(
service_account_json,
api_key,
project_id,
cloud_region,
dataset_id,
fhir_store_id,
gcs_uri)

# Clean up
fhir_stores.delete_fhir_store(
service_account_json,
api_key,
project_id,
cloud_region,
dataset_id,
fhir_store_id)

out, _ = capsys.readouterr()

assert 'Exported FHIR resources to bucket' in out