class UbootExpectTestCase(unittest.TestCase): @classmethod def setUpClass(cls): verbose = True dryrun = False logger = utils.logger.init_global_logger('UbootExpect') logger.setLevel(logging.DEBUG) streamhandler = logging.StreamHandler() streamhandler.setFormatter(logging.Formatter('%(msg)s')) if verbose: streamhandler.setLevel(logging.DEBUG) else: streamhandler.setLevel(logging.INFO) logger.addHandler(streamhandler) utils.executer.init_global_executer(dryrun=dryrun, enable_colors=False, verbose=verbose) def setUp(self): dryrun = False self.uboot = UbootExpect() self.uboot.console_logger = utils.logger.get_global_logger() self.uboot.dryrun = dryrun cmd = 'termnet %s %s' % (test_telnet_host, test_telnet_port) ret = self.uboot.open_comm(cmd) self.assertTrue(ret) ret = self.uboot.sync() self.assertTrue(ret) def tearDown(self): self.uboot.close_comm() def test_cmd(self): cmd = 'nand info' self.uboot.cmd(cmd) cmd = 'dhcp' self.uboot.cmd(cmd, prompt_timeout=None) self.uboot.cancel_cmd() def test_expect(self): cmd = 'nand info' response = 'Device 0' self.uboot.cmd(cmd, prompt_timeout=None) found, line = self.uboot.expect(response) self.assertTrue(found) if found: print 'Found line: %s' % line def test_env(self): var = 'foo' value = 'bar' self.uboot.set_env(var, value) if self.uboot.dryrun is False: self.assertEqual(self.uboot.get_env(var), value) var = "bootcmd" value = "'nboot 0x82000000 0 \${kernel_offset}'" self.uboot.set_env(var, value) if self.uboot.dryrun is False: self.assertEqual(self.uboot.get_env(var), 'nboot 0x82000000 0 ${kernel_offset}') self.uboot.save_env()