import unittest
from trustcafeapiwrapper.utils.get_parent_pksk_from_path import get_parent_pksk_from_path
class TestGetParentPkskFromPath(unittest.TestCase):
    def test_root_path(self):
        self.assertEqual(get_parent_pksk_from_path('/'), 'maintrunk#maintrunk')

    def test_userprofile_path(self):
        self.assertEqual(get_parent_pksk_from_path('/user/johndoe'), 'userprofile#johndoe')

    def test_subwiki_path(self):
        self.assertEqual(get_parent_pksk_from_path('/branch/12345'), 'subwiki#12345')

    def test_invalid_entity(self):
        with self.assertRaises(ValueError):
            get_parent_pksk_from_path('/invalid/12345')