19 lines
No EOL
683 B
Python
19 lines
No EOL
683 B
Python
import unittest
|
|
from trustcafeapiwrapper.utils.get_entity_from_str import get_entity_from_str
|
|
|
|
class TestGetEntityFromStr(unittest.TestCase):
|
|
def test_branch(self):
|
|
self.assertEqual(get_entity_from_str('branch'), 'subwiki')
|
|
|
|
def test_user(self):
|
|
self.assertEqual(get_entity_from_str('user'), 'userprofile')
|
|
|
|
def test_subwiki(self):
|
|
self.assertEqual(get_entity_from_str('subwiki'), 'subwiki')
|
|
|
|
def test_userprofile(self):
|
|
self.assertEqual(get_entity_from_str('userprofile'), 'userprofile')
|
|
|
|
def test_invalid_entity(self):
|
|
with self.assertRaises(ValueError):
|
|
get_entity_from_str('invalidentity') |