test appi client
This commit is contained in:
parent
d136f37b5f
commit
9c372fd266
3 changed files with 50 additions and 2 deletions
47
tests/apiclient.py
Normal file
47
tests/apiclient.py
Normal file
|
|
@ -0,0 +1,47 @@
|
|||
import unittest
|
||||
from trustcafeapiwrapper.apiclient import APIClient
|
||||
class TestAPIClient(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.api_client = APIClient(
|
||||
client_id="test_client_id",
|
||||
client_secret="test_client_secret",
|
||||
debug=False
|
||||
)
|
||||
|
||||
def test_initialization(self):
|
||||
self.assertEqual(self.api_client.client_id.get_secret_value(), "test_client_id")
|
||||
self.assertEqual(self.api_client.client_secret.get_secret_value(), "test_client_secret")
|
||||
self.assertFalse(self.api_client.debug)
|
||||
|
||||
def test_token_management(self):
|
||||
# Mock token data
|
||||
token_data = {
|
||||
"access_token": "test_access_token",
|
||||
"access_token_timeout": 9999999999 # Far future timestamp
|
||||
}
|
||||
self.api_client.set_token(token_data)
|
||||
self.assertEqual(self.api_client._access_token, "test_access_token")
|
||||
self.assertEqual(self.api_client._access_token_timeout, 9999999999)
|
||||
self.assertTrue(self.api_client.is_token_valid())
|
||||
|
||||
def test_run_job_with_string(self):
|
||||
# Mock a job function in the jobs module
|
||||
def mock_job(api_client, arg1, arg2):
|
||||
return f"Job executed with {arg1} and {arg2}"
|
||||
|
||||
# Dynamically add the mock job to the jobs module for testing
|
||||
import types
|
||||
import sys
|
||||
mock_module = types.ModuleType("mock_jobs")
|
||||
setattr(mock_module, "mock_job", mock_job)
|
||||
sys.modules["trustcafeapiwrapper.jobs.mock_jobs"] = mock_module
|
||||
|
||||
result = self.api_client.run_job("mock_jobs.mock_job", "value1", "value2")
|
||||
self.assertEqual(result, "Job executed with value1 and value2")
|
||||
|
||||
def test_run_job_with_function(self):
|
||||
def mock_job(api_client, arg1):
|
||||
return f"Job executed with {arg1}"
|
||||
|
||||
result = self.api_client.run_job(mock_job, "value1")
|
||||
self.assertEqual(result, "Job executed with value1")
|
||||
|
|
@ -8,10 +8,11 @@ from tests.utils.make_comment_sk import TestMakeCommentSk
|
|||
from tests.utils.make_post_sk import TestMakePostSk
|
||||
from tests.utils.get_child_spksk_from_paths import TestGetChildSpkskFromPaths
|
||||
|
||||
|
||||
from tests.wrappers.create_post import TestCreatePost
|
||||
from tests.wrappers.create_comment import TestCreateComment
|
||||
from tests.wrappers.react import TestReact
|
||||
from tests.wrappers.vote import TestVoteCast
|
||||
|
||||
from tests.apiclient import TestAPIClient
|
||||
|
||||
unittest.main()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue