| import os | import os |
| import datetime | import datetime |
| from nose.tools import assert_equal | from nose.tools import assert_equal |
| from ckanext.ga_report.download_analytics import DownloadAnalytics | from ckanext.ga_report.download_analytics import DownloadAnalytics |
| from ckanext.ga_report.ga_auth import (init_service, get_profile_id) | from ckanext.ga_report.ga_auth import (init_service, get_profile_id) |
| from ckanext.ga_report.ga_model import init_tables | from ckanext.ga_report.ga_model import init_tables |
| class TestAPI: | class TestAPI: |
| @classmethod | @classmethod |
| def setup_class(cls): | def setup_class(cls): |
| if not os.path.exists("token.dat") or not os.path.exists("credentials.json"): | if not os.path.exists("token.dat") or not os.path.exists("credentials.json"): |
| print '*' * 60 | print '*' * 60 |
| print "Tests may not run without first having run the auth process" | print "Tests may not run without first having run the auth process" |
| print '*' * 60 | print '*' * 60 |
| init_tables() | init_tables() |
| @classmethod | @classmethod |
| def teardown_class(cls): | def teardown_class(cls): |
| pass | pass |
| def test_latest(self): | def test_latest(self): |
| svc = init_service("token.dat", "credentials.json") | svc = init_service("token.dat", "credentials.json") |
| try: | try: |
| downloader = DownloadAnalytics(svc, profile_id=get_profile_id(svc)) | downloader = DownloadAnalytics(svc, profile_id=get_profile_id(svc)) |
| downloader.latest() | downloader.latest() |
| except Exception as e: | except Exception as e: |
| assert False, e | assert False, e |
| def test_since(self): | def test_since(self): |
| svc = init_service("token.dat", "credentials.json") | svc = init_service("token.dat", "credentials.json") |
| downloader = DownloadAnalytics(svc, profile_id=get_profile_id(svc)) | downloader = DownloadAnalytics(svc, profile_id=get_profile_id(svc)) |
| try: | try: |
| downloader.since_date(datetime.datetime.now() - datetime.timedelta(days=-30)) | downloader.for_date(datetime.datetime.now() - datetime.timedelta(days=-30)) |
| except Exception as e: | except Exception as e: |
| assert False, e | assert False, e |