Merge pull request #5 from scitran/tests

add tests

Author: cgc

.travis.yml

@@ -7,4 +7,6 @@ install:
   - "pip install --upgrade pip"
   - "pip install -r requirements.txt"
 
-script: "make lint"
+script:
+  - "make lint"
+  - "make test"

Makefile

@@ -1,3 +1,8 @@
+test:
+	# hack to avoid needing to install this package
+	# from http://stackoverflow.com/a/34140498
+	python -m pytest tests
+
 lint:
 	flake8 examples scitran_client
 

requirements.txt

@@ -33,3 +33,5 @@ urllib3==1.15.1
 websocket-client==0.37.0
 flake8==3.0.4
 pdoc==0.3.2
+pytest==3.0.6
+requests-mock==1.3.0

tests/fixtures/test.csv

@@ -0,0 +1,4 @@
+condition,onset
+red,0
+blue,3
+red,6

tests/test_query_builder.py

@@ -0,0 +1,16 @@
+from scitran_client import query, Projects, Sessions, Groups
+
+
+def test_query():
+    assert query(Sessions).filter(
+        Projects.label.match('ADHD cuing task'),
+        Groups.label.match('ADHDLab'),
+    ) == dict(
+        path='sessions',
+        projects=dict(filtered=dict(filter={'and': [
+            dict(query=dict(match=dict(label='ADHD cuing task'))),
+        ]})),
+        groups=dict(filtered=dict(filter={'and': [
+            dict(query=dict(match=dict(label='ADHDLab'))),
+        ]})),
+    )

tests/test_st_auth.py

@@ -0,0 +1,16 @@
+import requests_mock
+from scitran_client.st_auth import _is_valid_token
+
+host = 'https://flywheel.io'
+
+
+@requests_mock.mock()
+def test_is_valid_token(mock):
+    mock.get('{}/api/users/self'.format(host), status_code=200)
+    assert _is_valid_token(host, 'yep')
+
+
+@requests_mock.mock()
+def test_is_valid_token_invalid(mock):
+    mock.get('{}/api/users/self'.format(host), status_code=401)
+    assert not _is_valid_token(host, 'yep')

tests/test_st_client.py

@@ -0,0 +1,8 @@
+from scitran_client import compute_file_hash
+import os
+
+
+def test_compute_file_hash():
+    assert compute_file_hash(
+        os.path.join(os.path.dirname(__file__), 'fixtures/test.csv')
+    ) == 'v0-sha384-301d915f78736ff43dd396b5607cade4dffc0cd31c94bb2b80aff005cac042d8826a0a766c5dc2884a942cf960177378'