...

tests/client.py

@@ -0,0 +1,55 @@
+#!/usr/bin/env python
+
+"""Tests for `pyteamsnap` package."""
+
+
+import unittest
+from unittest import TestCase
+
+from pyteamsnap import client
+from os import getenv
+from pyteamsnap.models.base import BaseApiObject
+import vcr
+TEAMSNAP_TOKEN = getenv('TEAMSNAP_TOKEN')
+TEAMSNAP_TEAM = getenv('TEAMSNAP_TEAM')
+TEAMSNAP_EVENT = getenv('TEAMSNAP_EVENT')
+
+vcr_options = {
+    'decode_compressed_response': True,
+    'cassette_library_dir':'tests/fixtures/cassettes',
+    'filter_headers':['authorization']
+}
+
+class ClientTestCase(TestCase):
+    """Tests for `pyteamsnap` package."""
+
+    @classmethod
+    def setUpClass(cls) -> None:
+        """Set up test fixtures, if any."""
+        cls.TEAMSNAP_TOKEN = getenv('TEAMSNAP_TOKEN')
+        cls.TEAMSNAP_TEAM = int(getenv('TEAMSNAP_TEAM'))
+        cls.TEAMSNAP_EVENT = getenv('TEAMSNAP_EVENT')
+        with vcr.use_cassette("client.yml", **vcr_options):
+            cls.client = client.TeamSnap(token=TEAMSNAP_TOKEN)
+        cls.cassette = vcr.use_cassette(f"{cls.__name__}.yml", **vcr_options)
+        super().setUpClass()
+
+
+    def test_bulkload(self):
+        from pyteamsnap.models import Event, EventLineup, EventLineupEntry, AvailabilitySummary, Member
+        with self.cassette:
+            bulk_load = self.client.bulk_load(
+                team_id=self.TEAMSNAP_TEAM,
+                event__id=self.TEAMSNAP_EVENT,
+                types=[
+                    Event,
+                    EventLineup,
+                    EventLineupEntry,
+                    AvailabilitySummary,
+                    Member
+                ])
+
+        self.assertIsInstance(bulk_load, list)
+        return bulk_load
+
+