35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
import unittest
|
|
|
|
from src.constants import TEST_NODE_IDS as NODE_IDS, TEST_TOKEN as TOKEN
|
|
from src.data.github.graphql import RawCommit, get_commits
|
|
|
|
|
|
class TestTemplate(unittest.TestCase):
|
|
def test_get_commits(self):
|
|
node_ids = get_commits(access_token=TOKEN, node_ids=NODE_IDS, catch_errors=True)
|
|
|
|
# assert returns equal number of commits
|
|
self.assertIsInstance(node_ids, list)
|
|
self.assertEqual(len(node_ids), len(NODE_IDS))
|
|
|
|
self.assertIsInstance(node_ids[0], RawCommit)
|
|
|
|
def test_get_commits_invalid_access_token(self):
|
|
node_ids = get_commits(access_token="", node_ids=NODE_IDS, catch_errors=True)
|
|
|
|
# assert returns list of Nones
|
|
self.assertIsInstance(node_ids, list)
|
|
self.assertEqual(len(node_ids), len(NODE_IDS))
|
|
self.assertIsInstance(node_ids[0], type(None))
|
|
|
|
def test_get_commits_invalid_node_ids(self):
|
|
node_ids = get_commits(
|
|
access_token=TOKEN,
|
|
node_ids=[NODE_IDS[0], "", NODE_IDS[1]],
|
|
catch_errors=True,
|
|
)
|
|
|
|
self.assertIsInstance(node_ids[0], RawCommit)
|
|
self.assertIsInstance(node_ids[1], type(None))
|
|
self.assertIsInstance(node_ids[2], RawCommit)
|