@@ -951,6 +951,11 @@ def test_run(self, before_run, after_launch, start_profiling, interaction, stop_
951951 web_experiment .browsers = [mock_browser ]
952952 web_experiment .run (mock_device , path , run , 'chrome' )
953953
954+ kwargs = {
955+ "browser" : mock_browser ,
956+ "app" : mock_browser .package_name
957+ }
958+
954959 mock_manager = Mock ()
955960 mock_manager .attach_mock (before_run , "before_run_managed" )
956961 mock_manager .attach_mock (after_launch , "after_launch_managed" )
@@ -962,13 +967,17 @@ def test_run(self, before_run, after_launch, start_profiling, interaction, stop_
962967
963968 web_experiment .run (mock_device , path , run , 'chrome' )
964969
965- expected_calls = [call .before_run_managed (mock_device , path , run , browser = mock_browser ),
966- call .after_launch_managed (mock_device , path , run , browser = mock_browser ),
967- call .start_profiling_managed (mock_device , path , run , browser = mock_browser ),
968- call .interaction_managed (mock_device , path , run , browser = mock_browser ),
969- call .stop_profiling_managed (mock_device , path , run , browser = mock_browser ),
970- call .before_close_managed (mock_device , path , run , browser = mock_browser ),
971- call .after_run_managed (mock_device , path , run , browser = mock_browser )]
970+ expected_calls = [call .before_run_managed (mock_device , path , run , ** kwargs ),
971+ call .after_launch_managed (mock_device , path , run , ** kwargs ),
972+ call .start_profiling_managed (mock_device , path , run , ** kwargs ),
973+ call .interaction_managed (mock_device , path , run , ** kwargs ),
974+ call .stop_profiling_managed (mock_device , path , run , ** kwargs ),
975+ call .before_close_managed (mock_device , path , run , ** kwargs ),
976+ call .after_run_managed (mock_device , path , run , ** kwargs )]
977+
978+ from pprint import pprint
979+ pprint (mock_manager .mock_calls )
980+ pprint (expected_calls )
972981 assert mock_manager .mock_calls == expected_calls
973982
974983 @patch ('AndroidRunner.WebExperiment.WebExperiment.after_run' )
@@ -994,8 +1003,10 @@ def test_premature_stoppable_run(self, queue, premature_stoppable_run_run, prema
9941003 web_experiment .browsers = [mock_browser ]
9951004 web_experiment .run_stopping_condition_config = run_stopping_condition_config
9961005 web_experiment .queue = queue
1006+
9971007 kwargs = {
998- 'browser' : mock_browser
1008+ 'browser' : mock_browser ,
1009+ 'app' : mock_browser .package_name
9991010 }
10001011
10011012 queue_value = None
0 commit comments