class ClassTest(unittest.TestCase):
def setUp(self):
self.mock_ao = MagicMock()
self.mock_ao.clone.return_value = self.mock_ao
self.testobj = ORapi(self.mock_ao)
def test_str(self):
assert 'ORapi' in str(self.testobj)
def test_session_property(self):
# The session property is defined in the base class but we
# will unit test it here anyway, to make sure nobody changes
# the base class in a way that breaks us.
aw = self.testobj
original_session = aw.session
assert original_session
# test the setter and getter
fake_session = MagicMock()
aw.session = fake_session
assert aw.session is fake_session
# put the real one back
aw.session = original_session
assert aw.session is original_session
def test_get(self):
hdrs = {'fake-header': 'fake-value'}
expected = 'unit test get result'
self.mock_ao.get.return_value = expected
suffix = 'milk'
actual = self.testobj.get(suffix, headers=hdrs)
self.mock_ao.get.assert_called_with(suffix, headers=hdrs)
assert actual == expected
def test_search(self):
hdrs = {'fake-header': 'fake-value'}
expected = 'unit test search result'
self.mock_ao.get.return_value = expected
# the leading ? character is optional so try both.
qstring = '?name=marmaduke'
qs2 = qstring[1:]
# default suffix should be "search"
actual = self.testobj.search(pattern=qstring)
self.mock_ao.get.assert_called_with('search' + qstring, headers=None)
assert actual == expected
actual = self.testobj.search(pattern=qs2, headers=hdrs)
self.mock_ao.get.assert_called_with('search' + qstring, headers=hdrs)
assert actual == expected
# try a different suffix
suffix = 'toraiocht'
actual = self.testobj.search(pattern=qstring, suffix=suffix)
self.mock_ao.get.assert_called_with(suffix + qstring, headers=None)
assert actual == expected
actual = self.testobj.search(pattern=qs2, headers=hdrs, suffix=suffix)
self.mock_ao.get.assert_called_with(suffix + qstring, headers=hdrs)
assert actual == expected
class ClassTest(unittest.TestCase):
def setUp(self):
self.mock_ao = MagicMock()
self.mock_ao.clone.return_value = self.mock_ao
self.testobj = ORapi(self.mock_ao)
def test_str(self):
assert 'ORapi' in str(self.testobj)
def test_get(self):
hdrs = {'fake-header': 'fake-value'}
expected = 'unit test get result'
self.mock_ao.get.return_value = expected
suffix = 'milk'
actual = self.testobj.get(suffix, headers=hdrs)
self.mock_ao.get.assert_called_with(suffix, headers=hdrs)
assert actual == expected
def test_search(self):
hdrs = {'fake-header': 'fake-value'}
expected = 'unit test search result'
self.mock_ao.get.return_value = expected
# the leading ? character is optional so try both.
qstring = '?name=marmaduke'
qs2 = qstring[1:]
# default suffix should be "search"
actual = self.testobj.search(pattern=qstring)
self.mock_ao.get.assert_called_with('search' + qstring, headers=None)
assert actual == expected
actual = self.testobj.search(pattern=qs2, headers=hdrs)
self.mock_ao.get.assert_called_with('search' + qstring, headers=hdrs)
assert actual == expected
# try a different suffix
suffix = 'toraiocht'
actual = self.testobj.search(pattern=qstring, suffix=suffix)
self.mock_ao.get.assert_called_with(suffix + qstring, headers=None)
assert actual == expected
actual = self.testobj.search(pattern=qs2, headers=hdrs, suffix=suffix)
self.mock_ao.get.assert_called_with(suffix + qstring, headers=hdrs)
assert actual == expected