michael@0: from b2g_update_test import B2GUpdateTestCase, OTA, FOTA
michael@0: import os
michael@0: 
michael@0: update_test_dir = os.path.dirname(os.path.dirname(__file__))
michael@0: 
michael@0: class UpdateTestOTASimple(B2GUpdateTestCase):
michael@0:     def setUp(self):
michael@0:         prefs = {
michael@0:             "b2g.update.apply-idle-timeout": 0
michael@0:         }
michael@0:         mar_path = os.path.join(update_test_dir, "unit", "data", "simple.mar")
michael@0:         self.stage_update(complete_mar=mar_path, prefs=prefs)
michael@0:         B2GUpdateTestCase.setUp(self)
michael@0: 
michael@0:     def test_ota_simple(self):
michael@0:         self.marionette.set_script_timeout(60 * 1000 * 5)
michael@0: 
michael@0:         ota_simple_js = os.path.join(os.path.dirname(__file__),
michael@0:                                      "update_test_ota_simple.js")
michael@0:         self.execute_update_test(ota_simple_js, apply=OTA)