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')