Ejemplo n.º 1
0
	def testFileWrite(self):
		tmp_dir = tempfile.mktemp('-roxlib-test')
		os.mkdir(tmp_dir)
		root = self.master.root
		tmp_file = join(tmp_dir, 'new')
		def run():
			response = root.open(tmp_file, 'w')
			yield response
			stream = response.result

			assert os.path.isfile(tmp_file)

			response = root.write(stream, 'Hello\n')
			yield response
			assert response.result == None

			response = root.close(stream)
			yield response
			assert response.result is None

			assert file(tmp_file).read() == 'Hello\n'

			response = root.close(stream)
			yield response
			try:
				response.result
				assert 0, 'Expected an exception!'
			except KeyError:
				pass

			g.main_quit()
		tasks.Task(run())
		g.main()
		shutil.rmtree(tmp_dir)
Ejemplo n.º 2
0
    def testOutputBlocker(self):
        readable, writeable = os.pipe()

        def run():
            # Fill the input buffer...
            sent = 0
            while True:
                ob = tasks.OutputBlocker(writeable)
                tb = tasks.TimeoutBlocker(0.2)
                yield ob, tb
                if ob.happened:
                    sent += os.write(writeable, 'Hello\n')
                else:
                    assert tb.happened
                    break
            assert sent > 0
            #print "send %d bytes" % sent

            # Read it all back...
            got = 0
            while got < sent:
                got += len(os.read(readable, sent - got))

            ob = tasks.OutputBlocker(writeable)
            tb = tasks.TimeoutBlocker(0.2)
            yield ob, tb
            assert ob.happened
            assert not tb.happened

            g.main_quit()

        tasks.Task(run())
        g.main()
Ejemplo n.º 3
0
    def testIdleBlocker(self):
        def run():
            yield None
            g.main_quit()

        tasks.Task(run())
        g.main()
Ejemplo n.º 4
0
	def testOutputBlocker(self):
		readable, writeable = os.pipe()
		def run():
			# Fill the input buffer...
			sent = 0
			while True:
				ob = tasks.OutputBlocker(writeable)
				tb = tasks.TimeoutBlocker(0.2)
				yield ob, tb
				if ob.happened:
					sent += os.write(writeable, 'Hello\n')
				else:
					assert tb.happened
					break
			assert sent > 0
			#print "send %d bytes" % sent

			# Read it all back...
			got = 0
			while got < sent:
				got += len(os.read(readable, sent - got))

			ob = tasks.OutputBlocker(writeable)
			tb = tasks.TimeoutBlocker(0.2)
			yield ob, tb
			assert ob.happened
			assert not tb.happened

			g.main_quit()
		tasks.Task(run())
		g.main()
Ejemplo n.º 5
0
	def testFinished(self):
		readable, writeable = os.pipe()
		got = []

		def run(fail = False):
			for x in range(3):
				got.append(x)
				yield None
			if fail:
				raise Exception("Fail")

		def wait_for(t1, expected):
			yield t1.finished
			assert got == expected
			g.main_quit()

		t1 = tasks.Task(run())
		tasks.Task(wait_for(t1, [0, 1, 2]))
		assert not t1.finished.happened
		g.main()
		assert t1.finished.happened

		old = rox.report_exception
		try:
			rox.report_exception = lambda: (got.append(False), g.main_quit())
			got = []
			t2 = tasks.Task(run(fail = True))
			tasks.Task(wait_for(t2, [0, 1, 2, False]))
			g.main()
		finally:
			rox.report_exception = old
Ejemplo n.º 6
0
    def testFileRead(self):
        tmp_file = tempfile.mktemp(suffix='-roxlib-test')
        s = file(tmp_file, 'w')
        s.write('Hello\n')
        s.close()
        root = self.master.root

        def run():
            response = root.open(tmp_file)
            yield response
            stream = response.result

            response = root.read(stream, 5)
            yield response
            assert "Hello" == response.result

            response = root.close(stream)
            yield response
            assert response.result is None

            g.main_quit()

        tasks.Task(run())
        g.main()
        os.unlink(tmp_file)
Ejemplo n.º 7
0
    def testInputBlocker(self):
        readable, writeable = os.pipe()

        def run():
            ib = tasks.InputBlocker(readable)
            tb = tasks.TimeoutBlocker(0.2)
            yield ib, tb
            assert not ib.happened
            assert tb.happened
            os.write(writeable, "!")

            tb = tasks.TimeoutBlocker(0.2)
            yield ib, tb
            assert ib.happened
            assert not tb.happened

            assert os.read(readable, 1) == '!'
            os.close(writeable)
            ib = tasks.InputBlocker(readable)
            yield ib
            assert ib.happened

            g.main_quit()

        tasks.Task(run())
        g.main()
Ejemplo n.º 8
0
    def testFinished(self):
        readable, writeable = os.pipe()
        got = []

        def run(fail=False):
            for x in range(3):
                got.append(x)
                yield None
            if fail:
                raise Exception("Fail")

        def wait_for(t1, expected):
            yield t1.finished
            assert got == expected
            g.main_quit()

        t1 = tasks.Task(run())
        tasks.Task(wait_for(t1, [0, 1, 2]))
        assert not t1.finished.happened
        g.main()
        assert t1.finished.happened

        old = rox.report_exception
        try:
            rox.report_exception = lambda: (got.append(False), g.main_quit())
            got = []
            t2 = tasks.Task(run(fail=True))
            tasks.Task(wait_for(t2, [0, 1, 2, False]))
            g.main()
        finally:
            rox.report_exception = old
Ejemplo n.º 9
0
	def testTimeoutBlocker(self):
		def run():
			start = time.time()
			yield tasks.TimeoutBlocker(0.6)
			end = time.time()
			assert end > start + 0.5
			g.main_quit()
		tasks.Task(run())
		g.main()
Ejemplo n.º 10
0
    def testTimeoutBlocker(self):
        def run():
            start = time.time()
            yield tasks.TimeoutBlocker(0.6)
            end = time.time()
            assert end > start + 0.5
            g.main_quit()

        tasks.Task(run())
        g.main()
Ejemplo n.º 11
0
	def wait(self):
		"""Run a recursive mainloop until the command terminates.
		Raises an exception on error."""
		if self.child is None:
			self.start()
		self.waiting = True
		while not self.done:
			g.main()
		if self.done is not True:
			raise self.done
Ejemplo n.º 12
0
    def wait(self):
        """Run a recursive mainloop until the command terminates.
		Raises an exception on error."""
        if self.child is None:
            self.start()
        self.waiting = True
        while not self.done:
            g.main()
        if self.done is not True:
            raise self.done
Ejemplo n.º 13
0
 def get_response(self):
     if self.invisible.xmlrpc_response is None:
         self.waiting = True
         try:
             g.main()
         finally:
             self.waiting = False
     assert self.invisible.xmlrpc_response is not None
     retval, method = xmlrpclib.loads(self.invisible.xmlrpc_response)
     assert len(retval) == 1
     return retval[0]
Ejemplo n.º 14
0
	def get_response(self):
		if self.invisible.xmlrpc_response is None:
			self.waiting = True
			try:
				g.main()
			finally:
				self.waiting = False
		assert self.invisible.xmlrpc_response is not None
		retval, method = xmlrpclib.loads(self.invisible.xmlrpc_response)
		assert len(retval) == 1
		return retval[0]
Ejemplo n.º 15
0
	def testMissing(self):
		def run():
			response = self.master.root.missing('foo')
			yield response
			try:
				response.result
				assert 0, 'Expected an exception!'
			except AttributeError:
				pass
			g.main_quit()
		tasks.Task(run())
		g.main()
Ejemplo n.º 16
0
	def testUnlink(self):
		tmp = tempfile.mktemp('-roxlib-test')
		file(tmp, 'w').close()
		def run():
			assert os.path.isfile(tmp)
			response = self.master.root.unlink(tmp)
			yield response
			assert response.result is None
			assert not os.path.exists(tmp)
			g.main_quit()
		tasks.Task(run())
		g.main()
Ejemplo n.º 17
0
	def testRmTree(self):
		tmp_dir = tempfile.mktemp('-roxlib-test')
		os.mkdir(tmp_dir)
		def run():
			assert os.path.isdir(tmp_dir)
			response = self.master.root.rmtree(tmp_dir)
			yield response
			assert response.result is None
			assert not os.path.exists(tmp_dir)
			g.main_quit()
		tasks.Task(run())
		g.main()
Ejemplo n.º 18
0
    def testMissing(self):
        def run():
            response = self.master.root.missing('foo')
            yield response
            try:
                response.result
                assert 0, 'Expected an exception!'
            except AttributeError:
                pass
            g.main_quit()

        tasks.Task(run())
        g.main()
Ejemplo n.º 19
0
	def testTooSoon(self):
		def run():
			response = self.master.root.invoke()
			try:
				response.result
				assert 0, 'Expected an exception!'
			except Exception:
				pass
			yield response
			response.result
			g.main_quit()
		tasks.Task(run())
		g.main()
Ejemplo n.º 20
0
 def get_response(self):
     if self.response is None:
         self.waiting = True
         tag = gobject.timeout_add(60 * 1000, self.timed_out)
         try:
             g.main()
         finally:
             self.waiting = False
             gobject.source_remove(tag)
     assert self.response is not None
     retval, method = xmlrpclib.loads(self.response)
     assert len(retval) == 1
     return retval[0]
Ejemplo n.º 21
0
    def testTooSoon(self):
        def run():
            response = self.master.root.invoke()
            try:
                response.result
                assert 0, 'Expected an exception!'
            except Exception:
                pass
            yield response
            response.result
            g.main_quit()

        tasks.Task(run())
        g.main()
Ejemplo n.º 22
0
    def testUnlink(self):
        tmp = tempfile.mktemp('-roxlib-test')
        file(tmp, 'w').close()

        def run():
            assert os.path.isfile(tmp)
            response = self.master.root.unlink(tmp)
            yield response
            assert response.result is None
            assert not os.path.exists(tmp)
            g.main_quit()

        tasks.Task(run())
        g.main()
Ejemplo n.º 23
0
    def testRmTree(self):
        tmp_dir = tempfile.mktemp('-roxlib-test')
        os.mkdir(tmp_dir)

        def run():
            assert os.path.isdir(tmp_dir)
            response = self.master.root.rmtree(tmp_dir)
            yield response
            assert response.result is None
            assert not os.path.exists(tmp_dir)
            g.main_quit()

        tasks.Task(run())
        g.main()
Ejemplo n.º 24
0
	def testChmod(self):
		tmp_file = tempfile.mktemp(suffix = '-roxlib-test')
		s = file(tmp_file, 'w')
		s.close()
		root = self.master.root
		os.chmod(tmp_file, 0700)

		def run():
			assert os.stat(tmp_file).st_mode & 0777 == 0700
			response = root.chmod(tmp_file, 0655)
			yield response
			response.result
			assert os.stat(tmp_file).st_mode & 0777 == 0655
			g.main_quit()
		tasks.Task(run())
		g.main()
		os.unlink(tmp_file)
Ejemplo n.º 25
0
    def testChmod(self):
        tmp_file = tempfile.mktemp(suffix='-roxlib-test')
        s = file(tmp_file, 'w')
        s.close()
        root = self.master.root
        os.chmod(tmp_file, 0700)

        def run():
            assert os.stat(tmp_file).st_mode & 0777 == 0700
            response = root.chmod(tmp_file, 0655)
            yield response
            response.result
            assert os.stat(tmp_file).st_mode & 0777 == 0655
            g.main_quit()

        tasks.Task(run())
        g.main()
        os.unlink(tmp_file)
Ejemplo n.º 26
0
	def testRename(self):
		tmp_dir = tempfile.mktemp('-roxlib-test')
		os.mkdir(tmp_dir)
		root = self.master.root
		f = file(join(tmp_dir, 'old'), 'w')
		f.write('Hello\n')
		f.close()

		def run():
			response = root.rename(join(tmp_dir, 'old'),
					    join(tmp_dir, 'new'))
			yield response
			assert response.result == None

			assert file(join(tmp_dir, 'new')).read() == 'Hello\n'

			g.main_quit()
		tasks.Task(run())
		g.main()
		shutil.rmtree(tmp_dir)
Ejemplo n.º 27
0
    def testRename(self):
        tmp_dir = tempfile.mktemp('-roxlib-test')
        os.mkdir(tmp_dir)
        root = self.master.root
        f = file(join(tmp_dir, 'old'), 'w')
        f.write('Hello\n')
        f.close()

        def run():
            response = root.rename(join(tmp_dir, 'old'), join(tmp_dir, 'new'))
            yield response
            assert response.result == None

            assert file(join(tmp_dir, 'new')).read() == 'Hello\n'

            g.main_quit()

        tasks.Task(run())
        g.main()
        shutil.rmtree(tmp_dir)
Ejemplo n.º 28
0
	def testSu(self):	
		def run():
			maker = su.SuProxyMaker(
				'Need to become root to test this module.')
			yield maker.blocker
			root = maker.get_root()

			response = root.spawnvpe(os.P_NOWAIT, 'false', ['false'])
			yield response
			pid = response.result
			assert pid
			response = root.waitpid(pid, 0)
			yield response
			(pid, status) = response.result
			exitstatus = os.WEXITSTATUS(status)
			assert exitstatus != 0

			response = root.spawnvpe(os.P_WAIT, 'true', ['true'])
			yield response
			assert response.result == 0

			response = root.getuid()
			yield response
			assert response.result == 0

			response = root.setuid(os.getuid())
			yield response
			assert response.result is None

			response = root.getuid()
			yield response
			assert response.result == os.getuid()

			root.finish()
			g.main_quit()

		tasks.Task(run())
		g.main()
Ejemplo n.º 29
0
	def testFileRead(self):
		tmp_file = tempfile.mktemp(suffix = '-roxlib-test')
		s = file(tmp_file, 'w')
		s.write('Hello\n')
		s.close()
		root = self.master.root
		def run():
			response = root.open(tmp_file)
			yield response
			stream = response.result

			response = root.read(stream, 5)
			yield response
			assert "Hello" == response.result

			response = root.close(stream)
			yield response
			assert response.result is None

			g.main_quit()
		tasks.Task(run())
		g.main()
		os.unlink(tmp_file)
Ejemplo n.º 30
0
    def testSu(self):
        def run():
            maker = su.SuProxyMaker('Need to become root to test this module.')
            yield maker.blocker
            root = maker.get_root()

            response = root.spawnvpe(os.P_NOWAIT, 'false', ['false'])
            yield response
            pid = response.result
            assert pid
            response = root.waitpid(pid, 0)
            yield response
            (pid, status) = response.result
            exitstatus = os.WEXITSTATUS(status)
            assert exitstatus != 0

            response = root.spawnvpe(os.P_WAIT, 'true', ['true'])
            yield response
            assert response.result == 0

            response = root.getuid()
            yield response
            assert response.result == 0

            response = root.setuid(os.getuid())
            yield response
            assert response.result is None

            response = root.getuid()
            yield response
            assert response.result == os.getuid()

            root.finish()
            g.main_quit()

        tasks.Task(run())
        g.main()
Ejemplo n.º 31
0
def manage_session(test_mode):
    log.init()

    if not mydbus.dbus_version_ok and not xxmlrpc.working:
        rox.alert(problem_msg)

    set_up_environment()
    session.init()
    children.init()
    session_dbus.init()  # Start even if DBus is too old, for session bus
    xml_settings = settings.init()

    if mydbus.dbus_version_ok:
        service = dbus.service.BusName(constants.session_service,
                                       bus=session_dbus.get_session_bus())
        SessionObject3x(service)

    # This is like the D-BUS service, except using XML-RPC-over-X
    xml_service = xxmlrpc.XXMLRPCServer(constants.session_service)
    xml_service.add_object('/Session', XMLSessionObject())
    xml_service.add_object('/Settings', xml_settings)

    try:
        if test_mode:
            print "Test mode!"
            print "Started", os.system(
                "(/bin/echo hi >&2; sleep 4; date >&2)&")
            print "OK"
        else:
            try:
                wm.start()
            except:
                rox.report_exception()

        g.main()
    finally:
        session_dbus.destroy()
Ejemplo n.º 32
0
    def testFileWrite(self):
        tmp_dir = tempfile.mktemp('-roxlib-test')
        os.mkdir(tmp_dir)
        root = self.master.root
        tmp_file = join(tmp_dir, 'new')

        def run():
            response = root.open(tmp_file, 'w')
            yield response
            stream = response.result

            assert os.path.isfile(tmp_file)

            response = root.write(stream, 'Hello\n')
            yield response
            assert response.result == None

            response = root.close(stream)
            yield response
            assert response.result is None

            assert file(tmp_file).read() == 'Hello\n'

            response = root.close(stream)
            yield response
            try:
                response.result
                assert 0, 'Expected an exception!'
            except KeyError:
                pass

            g.main_quit()

        tasks.Task(run())
        g.main()
        shutil.rmtree(tmp_dir)
Ejemplo n.º 33
0
	def testInputBlocker(self):
		readable, writeable = os.pipe()
		def run():
			ib = tasks.InputBlocker(readable)
			tb = tasks.TimeoutBlocker(0.2)
			yield ib, tb
			assert not ib.happened
			assert tb.happened
			os.write(writeable, "!")

			tb = tasks.TimeoutBlocker(0.2)
			yield ib, tb
			assert ib.happened
			assert not tb.happened

			assert os.read(readable, 1) == '!'
			os.close(writeable)
			ib = tasks.InputBlocker(readable)
			yield ib
			assert ib.happened

			g.main_quit()
		tasks.Task(run())
		g.main()
Ejemplo n.º 34
0
from rox import xxmlrpc, g, tasks
try:
    proxy = xxmlrpc.XXMLProxy(memo_service)

    # Check to make sure it really is running...
    def check():
        call = proxy.get_object('/').get_pid()
        yield call, tasks.TimeoutBlocker(2)
        if call.happened:
            pid = call.get_response()
            rox.alert('Memo is already running (PID = %d)!' % pid)
            os._exit(1)
        g.main_quit()

    tasks.Task(check())
    g.main()
    print "Possible existing copy of Memo is not responding"
except xxmlrpc.NoSuchService:
    pass  # Good
server = xxmlrpc.XXMLRPCServer(memo_service)


class MemoObject:
    allowed_methods = ['get_pid']

    def get_pid(self):
        return os.getpid()


server.add_object('/', MemoObject())
Ejemplo n.º 35
0
	def testIdleBlocker(self):
		def run():
			yield None
			g.main_quit()
		tasks.Task(run())
		g.main()
Ejemplo n.º 36
0
def show_exception(type, value, tb, auto_details = False):
	"""Display this exception in an error box. The user has the options
	of ignoring the error, quitting the application and examining the
	exception in more detail. See also rox.report_exception()."""

	QUIT = 1
	DETAILS = 2
	SAVE = 3
	
	brief = ''.join(traceback.format_exception_only(type, value))

	toplevel_ref()
	box = g.MessageDialog(None, 0, g.MESSAGE_ERROR, g.BUTTONS_NONE, brief)
	
	if not auto_details:
		button = ButtonMixed(g.STOCK_ZOOM_IN, _('_Details'))
		button.set_flags(g.CAN_DEFAULT)
		button.show()
		box.add_action_widget(button, DETAILS)

	box.add_button(g.STOCK_HELP, g.RESPONSE_HELP)
	box.add_button(g.STOCK_OK, g.RESPONSE_OK)
	box.set_default_response(g.RESPONSE_OK)

	box.set_position(g.WIN_POS_CENTER)
	box.set_title(_('Error'))
	reply = []
	def response(box, resp):
		reply.append(resp)
		g.main_quit()
	box.connect('response', response)
	box.show()

	if tb:
		bug_report = 'Traceback (most recent call last):\n' + \
			     ''.join(traceback.format_stack(tb.tb_frame.f_back) +
				     traceback.format_tb(tb) +
				     traceback.format_exception_only(type, value))
	else:
		bug_report = 'No stack trace.'

	while 1:
		if auto_details:
			resp = DETAILS
			auto_details = False
		else:
			g.main()
			resp = reply.pop()
		if resp == int(g.RESPONSE_OK) or resp == int(g.RESPONSE_DELETE_EVENT):
			break
		if resp == SAVE:
			global savebox
			if savebox:
				savebox.destroy()
			def destroy(box):
				global savebox	# For pychecker
				savebox = None
			from saving import StringSaver
			savebox = StringSaver(bug_report, 'BugReport')
			savebox.connect('destroy', destroy)
			savebox.show()
			continue
		if resp == QUIT:
			sys.exit(1)
		elif resp == int(g.RESPONSE_HELP):
			_show_debug_help()
			continue
		assert resp == DETAILS
		box.set_response_sensitive(DETAILS, False)

		button = ButtonMixed(g.STOCK_SAVE, _('_Bug Report'))
		button.set_flags(g.CAN_DEFAULT)
		button.show()
		box.add_action_widget(button, SAVE)
		box.action_area.set_child_secondary(button, True)

		button = ButtonMixed(g.STOCK_QUIT, _('Forced Quit'))
		button.set_flags(g.CAN_DEFAULT)
		button.show()
		box.add_action_widget(button, QUIT)
		box.action_area.set_child_secondary(button, True)

		if tb:
			ee = ExceptionExplorer(tb)
			box.vbox.pack_start(ee)
			ee.show()
		else:
			no_trace = g.Label('No traceback object!')
			box.vbox.pack_start(no_trace)
			no_trace.show()
	box.destroy()
	toplevel_unref()