88import sys
99import time
1010
11- import mozversion
12- import requests
13-
1411sys .path .append (os .path .dirname (__file__ ))
1512
1613from felt_tests import FeltTests
1714
1815
19- class FeltUpdatesApplyFromFelt (FeltTests ):
20- EXTRA_PREFS = {
21- "app.update.log" : True ,
22- "app.update.disabledForTesting" : False ,
23- "app.update.BITS.enabled" : False ,
24- "enterprise.felt_tests.is_updates_testing" : True ,
25- "enterprise.felt_tests.read_update_url_from_prefs" : True ,
26- }
27-
28- def setup (self ):
29- self ._logger .info ("Enabling updates" )
30- version_info = mozversion .get_version (binary = self ._driver .instance .binary )
31- requests .post (
32- f"http://localhost:{ self .console_port } ,
33- data = version_info ,
34- )
35- self ._logger .info (f"Version: { version_info } )
36- self ._logger .info ("Updates ready" )
37- super ().setup ()
38-
16+ class FeltUpdatesBase (FeltTests ):
3917 def get_update_config_file_path (self ):
4018 self ._driver .set_context ("chrome" )
4119 rv = self ._driver .execute_script (
@@ -47,30 +25,55 @@ def get_update_config_file_path(self):
4725 self ._driver .set_context ("content" )
4826 return rv
4927
50- def test_felt_updates_apply_from_felt (self ):
51- self ._update_root = os .path .dirname (self .get_update_config_file_path ())
28+ def get_latest_update_from_history (self ):
29+ with self ._driver .using_context (self ._driver .CONTEXT_CHROME ):
30+ result = self ._driver .execute_async_script ("""
31+ const callback = arguments[arguments.length - 1];
32+ Cc["@mozilla.org/updates/update-manager;1"]
33+ .getService(Ci.nsIUpdateManager)
34+ .getHistory()
35+ .then(history => callback(history[0]))
36+ .catch(err => callback({ error: err.toString() }));
37+ """ )
38+
39+ assert "error" not in result , f"UpdateManager failed: { result .get ('error' )}
40+ return result
41+
42+ def assert_latest_update_url (self , state , expected_start , expected_end ):
43+ update_data = self .get_latest_update_from_history ()
44+
45+ assert update_data is not None , "Update data payload is missing."
46+ assert update_data ["state" ] == state , (
47+ f"Update state { update_data ['state' ]} { state }
48+ )
5249
53- self ._logger .info ("Updates ready: running tests" )
54- self .run_verify_update_ui ()
55- self .run_verify_update_check_run ()
56- # This is required since in marionette we use MAR that are not signed
57- # so we cannot reach the real restart point
58- # We also cannot close the window without loosing our Marionette access
59- with self ._driver .using_prefs ({
60- "enterprise.felt.previousBuildID" : "20250701120000"
61- }):
62- self .reload_chrome_window ()
63- self .run_verify_update_applied ()
50+ patch = update_data .get ("selectedPatch" , {})
51+ assert patch is not None , "The update history entry does not contain any patch."
6452
65- def teardown (self ):
66- self .run_updates_cleanup ()
53+ url = patch .get ("URL" )
54+ # Split query string that does not matter
55+ final_url = patch .get ("finalURL" ).split ("?" )[0 ]
6756
68- self ._logger .info ("Disabling updates" )
69- requests .post (f"http://localhost:{ self .console_port } )
57+ assert url is not None , "Patch is missing a URL descriptor."
58+ assert url == final_url , (
59+ f"URL mismatch: '{ url } { final_url }
60+ )
61+ assert url .startswith (expected_start ), (
62+ f"Expected URL to start with '{ expected_start } { url }
63+ )
64+ assert url .endswith (expected_end ), (
65+ f"Expected URL to end with '{ expected_end } { url }
66+ )
7067
68+ def run_felt_updates_apply (self ):
7169 # We are not going to start the browser so do not try to close it
7270 self ._manually_closed_child = True
73- super ().teardown ()
71+
72+ self ._update_root = os .path .dirname (self .get_update_config_file_path ())
73+
74+ self ._logger .info ("Updates ready: running tests" )
75+ self .run_verify_update_ui ()
76+ self .run_verify_update_check_run ()
7477
7578 def run_verify_update_ui (self ):
7679 self ._logger .info ("Checking update UI ..." )
0 commit comments