def testNoStateFile(self): """Make sure that loadState returns None if the file doesn't exist.""" e = IOError(errno.ENOENT,"ENOENT") self.open('/var/run/debmarshal-networks').AndRaise(e) self.mox.ReplayAll() self.assertEqual(utils.loadState('debmarshal-networks'), None)
def testNoStateFile(self): """Make sure that loadState returns None if the file doesn't exist.""" e = IOError(errno.ENOENT, "ENOENT") self.open('/var/run/debmarshal-networks').AndRaise(e) self.mox.ReplayAll() self.assertEqual(utils.loadState('debmarshal-networks'), None)
def testSuccess(self): """Test successfully loading a state file.""" data = ['foo', 'bar'] state = StringIO.StringIO(pickle.dumps(data)) self.open('/var/run/debmarshal-networks').AndReturn(state) self.mox.ReplayAll() self.assertEqual(data, utils.loadState('debmarshal-networks'))