def afterSetUp(self): super(Testzenbatchdumper, self).afterSetUp() self.zdumper = BatchDeviceDumper(noopts=1) self.zdumper.options = FakeOptions() self.zdumper.options.regex = '.*' self.zdumper.options.prune = False self.zdumper.options.root = '/zport/dmd/Devices/TestZenBatchDumper' self.zloader = BatchDeviceLoader(noopts=1) self.zloader.options = FakeOptions() self.zloader.options.nomodel = True self.zloader.options.nocommit = False # Ensure that both commands get # a real database connection self.zdumper.dmd = self.zloader.dmd self.log = logging.getLogger("zen.BatchDeviceDumper") self.zdumper.log = self.log self.zloader.log = logging.getLogger("zen.BatchDeviceLoader") # Actually add the organizer we use in testing testRoot = self.zdumper.options.root.rsplit('/', 1)[1] self.zdumper.dmd.Devices.manage_addOrganizer(testRoot)
def afterSetUp(self): super(Testzenbatchloader, self).afterSetUp() self.zloader = BatchDeviceLoader(noopts=1) self.zloader.options = FakeOptions() self.zloader.options.nomodel = True self.zloader.options.nocommit = False self.log = logging.getLogger("zen.BatchDeviceLoader") self.zloader.log = self.log
class Testzenbatchdumper(BaseTestCase): def afterSetUp(self): super(Testzenbatchdumper, self).afterSetUp() self.zdumper = BatchDeviceDumper(noopts=1) self.zdumper.options = FakeOptions() self.zdumper.options.regex = '.*' self.zdumper.options.prune = False self.zdumper.options.root = '/zport/dmd/Devices/TestZenBatchDumper' self.zloader = BatchDeviceLoader(noopts=1) self.zloader.options = FakeOptions() self.zloader.options.nomodel = True self.zloader.options.nocommit = False # Ensure that both commands get # a real database connection self.zdumper.dmd = self.zloader.dmd self.log = logging.getLogger("zen.BatchDeviceDumper") self.zdumper.log = self.log self.zloader.log = logging.getLogger("zen.BatchDeviceLoader") # Actually add the organizer we use in testing testRoot = self.zdumper.options.root.rsplit('/', 1)[1] self.zdumper.dmd.Devices.manage_addOrganizer(testRoot) def testDump(self): olympics = DateTime("2010/02/28") configs = [ "/Devices/TestZenBatchDumper", "device1 cDateTest=%s" % repr(olympics) ] device_list, unparseable = self.zloader.parseDevices(configs) self.zloader.processDevices(device_list) outFile = StringIO() numDevs = self.zdumper.listDeviceTree(outFile) total = len([d for d in self.zdumper.root.getSubDevices()]) self.assert_(total > 0, "Didn't load any devices") self.assert_(numDevs['Devices'] == total, "Dumped %d of %d devices" % (numDevs['Devices'], total)) outText = outFile.getvalue() outFile.close() self.log.info(outText) outConfigs = outText.split('\n') out_device_list, unparseable = self.zloader.parseDevices(outConfigs) self.assert_(numDevs['Devices'] == len(out_device_list)) dev = self.zloader.dmd.Devices.findDevice('device1') self.assert_(dev.cDateTest == olympics)
def importDevices(self, data, options={}): loader = BatchDeviceLoader(noopts=True) if isinstance(data, str): data = data.split('\n') # Set command-line options loader.options.nomodel = True self._setOptions(loader, options) devices, unparseable = loader.parseDevices(data) stats = loader.processDevices(devices) stats['unparseable'] = unparseable return stats
class Testzenbatchdumper(BaseTestCase): def afterSetUp(self): super(Testzenbatchdumper, self).afterSetUp() self.zdumper = BatchDeviceDumper(noopts=1) self.zdumper.options = FakeOptions() self.zdumper.options.regex = '.*' self.zdumper.options.prune = False self.zdumper.options.root = '/zport/dmd/Devices/TestZenBatchDumper' self.zloader = BatchDeviceLoader(noopts=1) self.zloader.options = FakeOptions() self.zloader.options.nomodel = True self.zloader.options.nocommit = False # Ensure that both commands get # a real database connection self.zdumper.dmd = self.zloader.dmd self.log = logging.getLogger("zen.BatchDeviceDumper") self.zdumper.log = self.log self.zloader.log = logging.getLogger("zen.BatchDeviceLoader") # Actually add the organizer we use in testing testRoot = self.zdumper.options.root.rsplit('/', 1)[1] self.zdumper.dmd.Devices.manage_addOrganizer(testRoot) def testDump(self): olympics = DateTime("2010/02/28") configs = ["/Devices/TestZenBatchDumper", "device1 cDateTest=%s" % repr(olympics)] device_list, unparseable = self.zloader.parseDevices(configs) self.zloader.processDevices(device_list) outFile = StringIO() numDevs = self.zdumper.listDeviceTree(outFile) total = len([d for d in self.zdumper.root.getSubDevices() ]) self.assert_(total > 0, "Didn't load any devices") self.assert_(numDevs['Devices'] == total, "Dumped %d of %d devices" % ( numDevs['Devices'], total)) outText = outFile.getvalue() outFile.close() self.log.info(outText) outConfigs = outText.split('\n') out_device_list, unparseable = self.zloader.parseDevices(outConfigs) self.assert_(numDevs['Devices'] == len(out_device_list)) dev = self.zloader.dmd.Devices.findDevice('device1') self.assert_(dev.cDateTest == olympics)
class Testzenbatchloader(BaseTestCase): def afterSetUp(self): super(Testzenbatchloader, self).afterSetUp() self.zloader = BatchDeviceLoader(noopts=1) self.zloader.options = FakeOptions() self.zloader.options.nomodel = True self.zloader.options.nocommit = False self.log = logging.getLogger("zen.BatchDeviceLoader") self.zloader.log = self.log def testSampleConfig(self): """ Is the internal sample config still working? """ configs = self.zloader.sample_configs.split('\n') device_list, unparseable = self.zloader.parseDevices(configs) self.zloader.processDevices(device_list) def testCProps(self): """ Verify that we can set custom properties """ olympics = DateTime("2010/02/28") configs = ["device1 cDateTest=%s" % repr(olympics)] device_list, unparseable = self.zloader.parseDevices(configs) self.zloader.processDevices(device_list) dev = self.zloader.dmd.Devices.findDevice('device1') self.assert_(dev.cDateTest == olympics)