Esempio n. 1
0
def test_service_generator():
    
    import ginkgo.core
    class MyService(ginkgo.core.Service): pass

    expected_children = [('hi', MyService()), ('also named', MyService())]
    expected_main = MyService()

    def service():
        for name, child in expected_children:
            yield name, child
        yield expected_main
        
    from ginkgo.runner import Runner
    
    Runner._args = ['run', '-C', 'config', '-u', 'nobody']
    Runner._opener = mock_open({'{}/config'.format(os.getcwd()): ""})

    runner = Runner()
    
    service_gen = service()
        
    children, main_service = runner._expand_service_generators(service_gen)

    print children, main_service
    assert children == expected_children
    assert main_service == expected_main
Esempio n. 2
0
def test_privileged_configuration():
    from ginkgo.runner import Runner
    from ginkgo import config
    import os

    uids = {}

    Runner._args = ['run', '-C', 'config', '-u', 'nobody']
    Runner._opener = mock_open({'{}/config'.format(os.getcwd()): ""})

    runner = Runner()

    # mock out the os getuid/setuid modules
    import os
    props = {'uid': 0, 'gid': 0}
    def getuid(): return props['uid']
    def setuid(uid): props['uid'] = uid
    def getgid(): return props['gid']
    def setgid(gid): props['gid'] = gid

    os.getuid = getuid
    os.setuid = setuid
    os.getgid = getgid
    os.setgid = setgid

    # capture the uid of the service at start and stop time
    def service():
        import ginkgo.core
        class Service(ginkgo.core.Service):
            def do_start(self):
                print "asked to do_start"
                uids['start'] = os.getuid()

                def stop(): self.stop()
                gevent.spawn_later(0.1, stop)
            
            def do_stop(self):
                print "asked to do_stop"
                uids['stop'] = os.getuid()
                print "done stopping"

        return Service()

    config.load({'service': service,
                 '_allow_early_gevent_import_for_tests': True})

    print "before action"
    runner.do_action()
    
    assert uids['start'] != runner.uid
    assert uids['stop'] == runner.uid