"""DR semantic regression tests independent of ROS transport."""
import json
import sys
import tempfile
import unittest
from pathlib import Path

ROOT = Path(__file__).resolve().parents[1]
sys.path[:0] = [str(ROOT), str(ROOT / 'coordinator'), str(ROOT / 'robobrain')]


class DrSemanticsTests(unittest.TestCase):
    def test_localization_adapter_preserves_all_dr_context_fields(self):
        sys.path.insert(0,str(ROOT/'ros2/robobrain_services'))
        from robobrain_services.nodes import perception_goal
        from types import SimpleNamespace as S
        goal=S(task_id='t',subtask_id='s',target_ref='water',target_description='bottle',
               shelf_id='shelf_A',column_id='2',tier_id='3',station_binding_ref='binding-9',
               expected_geometry_epoch=7,capture_after=S(sec=1,nanosec=5),timeout=S(sec=2,nanosec=0))
        result=perception_goal(goal,'localize')
        for key in ('shelf_id','column_id','tier_id','station_binding_ref','expected_geometry_epoch'):
            self.assertEqual(result[key],getattr(goal,key))

    def test_shelf_tier_can_be_empty_without_guessing(self):
        # BT p15 and RB02 allow an unknown tier; execution must decide whether
        # its chosen route can use the observation without that calibration.
        from robot_robobrain.service import BrainService
        from robot_robobrain.backends import FixtureBackend
        from robot_robobrain.observations import Observation
        with tempfile.TemporaryDirectory() as folder:
            image = Path(folder) / 'frame.jpg'; image.write_bytes(b'fixture')
            raw = dict(status='SUCCEEDED', shelf_id='shelf_A', side_id='FRONT',
                       column_id='1', tier_id='', confidence=.95)
            service = BrainService(FixtureBackend(json.dumps(raw)), folder)
            observation = Observation('obs', 100, 'camera', str(image), 'observe_A', 1, 'shelf_A')
            goal = dict(task_id='t', subtask_id='s', target_ref='water', source_region_ref='shelf_A',
                        observation_station_id='observe_A', station_registry_version=1, capture_after=90, timeout=1)
            result = service.shelf(goal, observation, 110, max_age_ns=50)
            self.assertEqual(result['status'], 'SUCCEEDED')
            self.assertEqual(result['tier_id'], '')



if __name__ == '__main__':
    unittest.main()