import unittest

from . import test_data

from sia import util

class UtilTest(unittest.TestCase):
    def test_get_valid_root_elements_single(self):
        valid_elements = util.get_valid_root_elements(test_data.echo_action_schema)
        self.assertEqual(valid_elements, {'test_tag'})

    def test_split_response_single_element(self):
        response = "Some reasoning here\n<test_tag>content</test_tag>"
        valid_elements = {'test_tag'}
        result = util.split_response(response, valid_elements)
        self.assertEqual(result.reasoning, "Some reasoning here")
        self.assertEqual(result.actions, "<test_tag>content</test_tag>")