import unittest
from baim.retrieval_audit import bm25
from baim.mind2web_smoke import normalize_task


class RetrievalTests(unittest.TestCase):
    def test_broad_candidates_keep_tabs_without_allowing_sensitive_controls(self):
        rows=[dict(role='tab',name='Account'),dict(role='generic',name='Account',sensitive=True),
              dict(role='button',name='Other',enabled=False)]
        self.assertEqual(bm25('Account',rows),[0])

    def test_history_excludes_current_and_future_actions(self):
        candidate=dict(tag='button',attributes='{}',backend_node_id='1')
        step=dict(pos_candidates=[candidate],neg_candidates=[],cleaned_html='<button backend_node_id="1">A</button>',
                  action_uid='a',operation=dict(op='CLICK'))
        task=dict(confirmed_task='Goal',action_reprs=['previous','current'],actions=[step,{**step,'action_uid':'b'}])
        rows=list(normalize_task(task))
        self.assertEqual(rows[0]['history'],[])
        self.assertEqual(rows[1]['history'],['previous'])