michael@0: from b2g_update_test import B2GUpdateSmokeTestCase, OTA, FOTA
michael@0: import os
michael@0: 
michael@0: this_dir = os.path.dirname(__file__)
michael@0: 
michael@0: class OTASimple(B2GUpdateSmokeTestCase):
michael@0:     JS_PATH          = os.path.join(this_dir, "update_smoketest_ota_simple.js")
michael@0:     START_WITH_BUILD = "start"
michael@0:     STAGE_PREFS      = {
michael@0:         "b2g.update.apply-idle-timeout": 0,
michael@0:         "b2g.update.apply-prompt-timeout": 5 * 60 * 1000
michael@0:     }
michael@0:     TIMEOUT     = 5 * 60 * 1000
michael@0: 
michael@0:     def test_ota_simple_complete(self):
michael@0:         self.stage_update(build="finish", mar="complete")
michael@0:         self.execute_smoketest()
michael@0: 
michael@0:     def test_ota_simple_partial(self):
michael@0:         self.stage_update(build="finish", mar="partial")
michael@0:         self.execute_smoketest()