test_openbis.py

import json
from pybis import DataSet
from pybis import Openbis

def test_token(openbis_instance):
    assert openbis_instance.hostname is not None
    new_instance = Openbis(openbis_instance.url)
    new_instance.login()
    assert new_instance.token is not None
    assert new_instance.is_token_valid() is True
    new_instance.logout()
    assert new_instance.is_token_valid() is False

    openbis_instance.save_token()
    another_instance = Openbis(openbis_instance.url)
    assert another_instance.is_token_valid() is True



def test_get_samples_by_id(openbis_instance):
    response = openbis_instance.get_samples('/TEST/TEST-SAMPLE-2-CHILD-1')
    assert response is not None
    assert response['/TEST/TEST-SAMPLE-2-CHILD-1'] is not None


def test_get_samples_by_permid(openbis_instance):
    response = openbis_instance.get_samples('20130415091923485-402')
    assert response is not None
    assert response['20130415091923485-402'] is not None


def test_get_parents(openbis_instance):
    id = '/TEST/TEST-SAMPLE-2'
    response = openbis_instance.get_samples(id)
    assert response is not None
    assert 'parents' in response[id]
    assert 'identifier' in response[id]['parents'][0]
    assert response[id]['parents'][0]['identifier']['identifier'] == '/TEST/TEST-SAMPLE-2-PARENT'


def test_get_dataset_parents(openbis_instance):
    permid = '20130415093804724-403'
    parent_permid = '20130415100158230-407'
    dataset = openbis_instance.get_dataset(permid)
    assert dataset is not None
    parents = dataset.get_parents()
    assert isinstance(parents, list)
    assert parents[0] is not None
    assert isinstance(parents[0], DataSet)
    assert parents[0].permid == parent_permid

    children = parents[0].get_children()
    assert isinstance(children, list)
    assert children[0] is not None
    assert isinstance(children[0], DataSet)


def test_get_dataset_by_permid(openbis_instance):
    permid = '20130412142942295-198'
    permid = '20130412153118625-384'
    dataset = openbis_instance.get_dataset(permid)
    assert dataset is not None
    assert isinstance(dataset, DataSet)
    assert isinstance(dataset, Openbis)
    assert 'dataStore' in dataset.data
    assert 'downloadUrl' in dataset.data['dataStore']
    file_list = dataset.get_file_list(recursive=False)
    assert file_list is not None
    assert isinstance(file_list, list)
    assert len(file_list) == 1

    file_list = dataset.get_file_list(recursive=True)
    assert file_list is not None
    assert len(file_list) > 10