Esempio n. 1
0
    def test___dumpStats(t):
        ZenHub._ZenHub__dumpStats(t.zh)

        t.zh._service_manager.getStatusReport.assert_called_once_with()
        t.zh.log.info.assert_called_once_with(
            "\n%s\n",
            t.zh._service_manager.getStatusReport.return_value
        )
        t.zh._service_manager.reportWorkerStatus.assert_called_once_with()
Esempio n. 2
0
    def test_sighandler_USR2(t, _notify, _ReportWorkerStatus):
        ZenHub.sighandler_USR2(t.zh, signum="unused", frame="unused")

        t.zh._status_reporter.getReport.assert_called_once_with()
        t.zh.log.info.assert_called_once_with(
            "\n%s\n", t.zh._status_reporter.getReport.return_value,
        )
        _notify.assert_called_once_with(_ReportWorkerStatus.return_value)
        _ReportWorkerStatus.assert_called_once_with()
    def test_sighandler_USR1(t, signal, _super):
        t.zh.profiler = Mock(name='profiler', spec_set=['dump_stats'])
        t.zh.options = Mock(name='options', profiling=True)
        signum = sentinel.signum
        frame = sentinel.frame

        ZenHub.sighandler_USR1(t.zh, signum=signum, frame=frame)

        t.zh.profiler.dump_stats.assert_called_with()
        _super.assert_called_with(ZenHub, t.zh)
        _super.return_value.sighandler_USR1.assert_called_with(signum, frame)
Esempio n. 4
0
    def test_sighandler_USR1(t, signal, super):
        '''Daemon function
        when signal USR1 is recieved, broadcast it to all worker processes
        '''
        t.zh.profiler = Mock(name='profiler', spec_set=['dump_stats'])
        t.zh.options = Mock(name='options', profiling=True)
        signum = sentinel.signum
        frame = sentinel.frame

        ZenHub.sighandler_USR1(t.zh, signum=signum, frame=frame)

        t.zh.profiler.dump_stats.assert_called_with()
        super.assert_called_with(ZenHub, t.zh)
        super.return_value.sighandler_USR1.assert_called_with(
            signum, frame
        )
    def setUp(t):
        # Patch out the ZenHub __init__ method, due to excessive side-effects
        t.init_patcher = patch.object(
            ZCmdBase,
            '__init__',
            autospec=True,
            return_value=None,
        )
        t.init_patcher.start()
        t.addCleanup(t.init_patcher.stop)

        # Mock out attributes set by ZCmdBase
        t.zcmdbase_patchers = [
            patch.object(ZenHub, 'dmd', create=True),
            patch.object(ZenHub, 'log', create=True),
            patch.object(ZenHub, 'options', create=True),
            patch.object(ZenHub, 'niceDoggie', create=True),
            patch.object(
                ZenHub,
                'storage',
                create=True,
                set_spec=['poll_invalidations'],
            ),
        ]
        for patcher in t.zcmdbase_patchers:
            patcher.start()
            t.addCleanup(patcher.stop)

        # Patch external dependencies
        needs_patching = [
            "reactor",
            "XmlRpcManager",
            "make_server_factory",
            "getCredentialCheckers",
            "make_service_manager",
            "make_pools",
            "ZenHubStatusReporter",
            "StatsMonitor",
            "InvalidationManager",
            "MetricManager",
            "notify",
            "ContinuousProfiler",
            "load_config_override",
            "load_config",
            "IHubConfProvider",
            "provideUtility",
            "register_legacy_worklist_metrics",
        ]
        t.patchers = {}
        for target in needs_patching:
            patched = patch(
                "{src}.{target}".format(target=target, **PATH),
                autospec=True,
            )
            t.patchers[target] = patched
            setattr(t, target, patched.start())
            t.addCleanup(patched.stop)

        t.zh = ZenHub()
Esempio n. 6
0
class TestZenHub(BaseTestCase):

    layer = ZenossTestCaseLayer

    base = 7000
    xbase = 8000

    def afterSetUp(self):
        super(TestZenHub, self).afterSetUp()
        global count
        count += 1
        base = self.base + count
        xbase = self.xbase + count
        self.before, sys.argv = sys.argv, [
            'run',
            '--pbport=%d' % base,
            '--xmlrpcport=%d' % xbase, '--workers=0'
        ]
        self.zenhub = ZenHub()
        from zope.component import getGlobalSiteManager
        # The call to zenhub above overrides the queue so we need to
        # re-override it
        gsm = getGlobalSiteManager()
        queue = DummyQueuePublisher()
        gsm.registerUtility(queue, IQueuePublisher)

    def beforeTearDown(self):
        sys.argv = self.before
        super(TestZenHub, self).beforeTearDown()

    def testPbRegistration(self):
        from twisted.spread.jelly import unjellyableRegistry
        self.assertTrue(unjellyableRegistry.has_key('DataMaps.ObjectMap'))
        self.assertTrue(
            unjellyableRegistry.has_key(
                'Products.DataCollector.plugins.DataMaps.ObjectMap'))

    def testGetService(self):
        client = TestClient(self, self.base + count)
        self.zenhub.main()
        self.assertTrue(client.success)

    def testSendEvent(self):
        client = SendEventClient(self, self.base + count)
        self.zenhub.main()
        self.assertTrue(client.success)
Esempio n. 7
0
 def afterSetUp(self):
     super(TestZenHub, self).afterSetUp()
     global count
     count += 1
     base = self.base + count
     xbase = self.xbase + count
     self.before, sys.argv = sys.argv, [
         'run',
         '--pbport=%d' % base,
         '--xmlrpcport=%d' % xbase, '--workers=0'
     ]
     self.zenhub = ZenHub()
     from zope.component import getGlobalSiteManager
     # The call to zenhub above overrides the queue so we need to
     # re-override it
     gsm = getGlobalSiteManager()
     queue = DummyQueuePublisher()
     gsm.registerUtility(queue, IQueuePublisher)
Esempio n. 8
0
class TestZenHub(BaseTestCase):

    layer = ZenossTestCaseLayer

    base = 7000
    xbase = 8000

    def afterSetUp(self):
        super(TestZenHub, self).afterSetUp()
        global count
        count += 1
        base = self.base + count
        xbase = self.xbase + count
        self.before, sys.argv = sys.argv, ['run',
                                           '--pbport=%d' % base,
                                           '--xmlrpcport=%d' % xbase,
                                           '--workers=0']
        self.zenhub = ZenHub()
        from zope.component import getGlobalSiteManager
        # The call to zenhub above overrides the queue so we need to
        # re-override it
        gsm = getGlobalSiteManager()
        queue = DummyQueuePublisher()
        gsm.registerUtility(queue, IQueuePublisher)

    def beforeTearDown(self):
        sys.argv = self.before
        super(TestZenHub, self).beforeTearDown()

    def testPbRegistration(self):
        from twisted.spread.jelly import unjellyableRegistry
        self.assertTrue(unjellyableRegistry.has_key('DataMaps.ObjectMap'))
        self.assertTrue(unjellyableRegistry.has_key('Products.DataCollector.plugins.DataMaps.ObjectMap'))

    def testGetService(self):
        client = TestClient(self, self.base + count)
        self.zenhub.main()
        self.assertTrue(client.success)

    def testSendEvent(self):
        client = SendEventClient(self, self.base + count)
        self.zenhub.main()
        self.assertTrue(client.success)
 def afterSetUp(self):
     super(TestZenHub, self).afterSetUp()
     global count
     count += 1
     base = self.base + count
     xbase = self.xbase + count
     self.before, sys.argv = sys.argv, ['run',
                                        '--pbport=%d' % base,
                                        '--xmlrpcport=%d' % xbase,
                                        '--workers=0']
     self.zenhub = ZenHub()
     from zope.component import getGlobalSiteManager
     # The call to zenhub above overrides the queue so we need to
     # re-override it
     gsm = getGlobalSiteManager()
     queue = DummyQueuePublisher()
     gsm.registerUtility(queue, IQueuePublisher)
Esempio n. 10
0
 def test_sighandler_USR2(t, reactor):
     '''Daemon function
     when signal USR2 is recieved, broadcast it to all worker processes
     '''
     ZenHub.sighandler_USR2(t.zh, signum='unused', frame='unused')
     reactor.callLater.assert_called_once_with(0, t.zh._ZenHub__dumpStats)
Esempio n. 11
0
    def test___init__(
        t,
        _super,
        load_config,
        notify,
        HubWillBeCreatedEvent,
        ContinuousProfiler,
        reactor,
        zenPath,
        HubCreatedEvent,
        App_Start,
        signal,
        load_config_override,
        XmlRpcManager,
        make_server_factory,
        getCredentialCheckers,
        make_service_manager,
        make_pools,
        ZenHubStatusReporter,
        StatsMonitor,
        MetricManager,
        InvalidationManager,
        provideUtility,
        register_legacy_worklist_metrics,
    ):
        # Mock out attributes set by the parent class
        # Because these changes are made on the class, they must be reversable
        t.zenhub_patchers = [
            patch.object(ZenHub, 'dmd', create=True),
            patch.object(ZenHub, 'log', create=True),
            patch.object(ZenHub, 'options', create=True),
            patch.object(ZenHub, 'getRRDStats', autospec=True),
            patch.object(ZenHub, '_getConf', autospec=True),
            patch.object(ZenHub, 'sendEvent', autospec=True),
            patch.object(ZenHub, 'storage', create=True),
        ]

        for patcher in t.zenhub_patchers:
            patcher.start()
            t.addCleanup(patcher.stop)

        ZenHub._getConf.return_value.id = 'config_id'
        ZenHub.storage.mock_add_spec(['poll_invalidations'])

        zh = ZenHub()

        t.assertIsInstance(zh, ZenHub)
        # Skip Metrology validation for now due to complexity
        _super.return_value.__init__assert_called_with(ZenHub, zh)
        load_config.assert_called_with("hub.zcml", ZENHUB_MODULE)
        HubWillBeCreatedEvent.assert_called_with(zh)
        notify.assert_has_calls([call(HubWillBeCreatedEvent.return_value)])
        # Performance Profiling
        ContinuousProfiler.assert_called_with('zenhub', log=zh.log)
        zh.profiler.start.assert_called_with()

        # 'counters' is a ZenHub API.
        t.assertIs(zh.counters, StatsMonitor.return_value.counters)

        t.assertIsInstance(zh.shutdown, bool)
        t.assertFalse(zh.shutdown)

        t.assertIs(zh._monitor, StatsMonitor.return_value)
        t.assertIs(zh._status_reporter, ZenHubStatusReporter.return_value)
        t.assertIs(zh._pools, make_pools.return_value)
        t.assertIs(zh._service_manager, make_service_manager.return_value)
        t.assertIs(zh._server_factory, make_server_factory.return_value)
        t.assertIs(zh._xmlrpc_manager, XmlRpcManager.return_value)
        register_legacy_worklist_metrics.assert_called_once_with()

        # Event Handler shortcut
        t.assertEqual(zh.zem, zh.dmd.ZenEventManager)

        # Messageing config, including work and invalidations
        # Patched internal import of Products.ZenMessaging.queuemessaging
        load_config_override.assert_called_with(
            'twistedpublisher.zcml',
            QUEUEMESSAGING_MODULE,
        )
        HubCreatedEvent.assert_called_with(zh)
        notify.assert_called_with(HubCreatedEvent.return_value)
        zh.sendEvent.assert_called_with(
            zh,
            eventClass=App_Start,
            summary='zenhub started',
            severity=0,
        )

        StatsMonitor.assert_called_once_with()
        ZenHubStatusReporter.assert_called_once_with(
            StatsMonitor.return_value, )
        make_pools.assert_called_once_with()
        make_service_manager.assert_called_once_with(make_pools.return_value)
        getCredentialCheckers.assert_called_once_with(
            zh.options.passwordfile, )
        make_server_factory.assert_called_once_with(
            make_pools.return_value,
            make_service_manager.return_value,
            getCredentialCheckers.return_value,
        )
        XmlRpcManager.assert_called_once_with(
            zh.dmd,
            getCredentialCheckers.return_value[0],
        )

        MetricManager.assert_called_with(daemon_tags={
            'zenoss_daemon': 'zenhub',
            'zenoss_monitor': zh.options.monitor,
            'internal': True,
        }, )
        t.assertEqual(zh._metric_manager, MetricManager.return_value)
        t.assertEqual(
            zh._invalidation_manager,
            InvalidationManager.return_value,
        )
        provideUtility.assert_called_once_with(zh._metric_manager)

        signal.signal.assert_called_with(signal.SIGUSR2, zh.sighandler_USR2)
Esempio n. 12
0
        self._zenhub = zenhub

    def check(self):
        pass


class HubWillBeCreatedEvent(object):  # noqa: D101
    implements(IHubWillBeCreatedEvent)

    def __init__(self, hub):
        self.hub = hub


class HubCreatedEvent(object):  # noqa: D101
    implements(IHubCreatedEvent)

    def __init__(self, hub):
        self.hub = hub


class ParserReadyForOptionsEvent(object):  # noqa: D101
    implements(IParserReadyForOptionsEvent)

    def __init__(self, parser):
        self.parser = parser


if __name__ == '__main__':
    from Products.ZenHub.zenhub import ZenHub
    ZenHub().main()