1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38
|
"""FlexScheduleRule object test module"""
import unittest
from unittest.mock import patch
import uuid
from rachiopy import FlexSchedulerule
from tests.constants import BASE_API_URL, AUTHTOKEN, RESPONSE200
class TestFlexScheduleRuleMethods(unittest.TestCase):
"""Class containing the FlexScheduleRule object test cases."""
def setUp(self):
self.flexschedulerule = FlexSchedulerule(AUTHTOKEN)
def test_init(self):
"""Test if the constructor works as expected."""
self.assertEqual(self.flexschedulerule.authtoken, AUTHTOKEN)
@patch("requests.Session.request")
def test_get(self, mock):
"""Test if the get method works as expected."""
mock.return_value = RESPONSE200
flexscheduleruleid = uuid.uuid4()
self.flexschedulerule.get(flexscheduleruleid)
args, kwargs = mock.call_args
# Check that the mock function is called with the rights args.
self.assertEqual(
args[1],
f"{BASE_API_URL}/flexschedulerule/{flexscheduleruleid}",
)
self.assertEqual(args[0], "GET")
self.assertEqual(kwargs["data"], None)
|