1# Copyright 2024 The Chromium Authors 2# Use of this source code is governed by a BSD-style license that can be 3# found in the LICENSE file. 4 5import datetime as dt 6 7from crossbench.probes.screenshot import ScreenshotProbe 8from crossbench.runner.run import Run 9from tests.crossbench.mock_helper import MockStory 10from tests.crossbench.runner.groups.base import BaseRunGroupTestCase 11from tests.crossbench.runner.helper import MockProbe 12 13 14class RunTestCase(BaseRunGroupTestCase): 15 16 def test_find_probe_context(self): 17 self.runner.attach_probe(MockProbe()) 18 session = self.default_session() 19 run = Run(self.runner, session, MockStory("mock story"), 1, False, 20 "1_default", 1, "test run", dt.timedelta(minutes=1), True) 21 session.set_ready() 22 with session.open(): 23 self.assertIsNotNone(run.find_probe_context(MockProbe)) 24 self.assertIsNone(run.find_probe_context(ScreenshotProbe)) 25