Esempio n. 1
0
class Shell(object):
    def __init__(self):
        self.subject = Subject()

    def add_run_observer(self, observer):
        self.subject.add_observer(observer)
    
    def remove_run_observer(self, observer):
        self.subject.remove_observer(self, observer)

    def check_open(self, args, input_string=None):
        self.subject.notify(' '.join(args))
        return check_open(args, input_string)
Esempio n. 2
0
class SubjectTestCase(unittest.TestCase):
    def setUp(self):
        self.count1 = 0
        self.count2 = 0
        self.count3 = 0
        self.subject = Subject()

    def event1(self):
        self.count1 += 1

    def event2(self):
        self.count2 += 1

    def remove_during_event(self):
        self.count3 += 1
        self.subject.remove_observer(self.remove_during_event)

    def test_simple(self):
        self.subject.add_observer(self.event1)
        self.subject.notify()
        self.assertEqual(1, self.count1)
        self.subject.remove_observer(self.event1)
        self.subject.notify()
        self.assertEqual(1, self.count1)

        self.subject.add_observer(self.event1)
        self.subject.add_observer(self.event2)
        self.subject.notify()
        self.assertEqual(2, self.count1)
        self.assertEqual(1, self.count2)
    
    def test_multiple(self):
        self.subject.add_observer(self.event1)
        self.subject.add_observer(self.event1)
        self.subject.notify()
        self.assertEqual(2, self.count1)

    def test_remove_during_event(self):
        self.subject.add_observer(self.event1)
        self.subject.add_observer(self.remove_during_event)
        self.subject.add_observer(self.event2)
        self.subject.notify()
        self.assertEqual(1, self.count1)
        self.assertEqual(1, self.count2)
        self.assertEqual(1, self.count3)
        self.subject.notify()
        self.assertEqual(2, self.count1)
        self.assertEqual(2, self.count2)
        self.assertEqual(1, self.count3)
Esempio n. 3
0
class GDBDebugger(object):

    def __init__(self, gdb_stub):
        self.gdb_stub = gdb_stub
        self.gdb_stub.execute_sync('set breakpoint pending on')
        self.exit_subject = Subject()
        gdb_stub.add_event_observer(self.exit_subject.notify)

    def add_symbol_file(self, symbol_file, sections):
        out = StringIO()
        out.write('add-symbol-file /home/pag/Code/drk/src/core/%s ' % (symbol_file))
        out.write(sections['.text'])
        del sections['.text']
        for section in sections.iteritems():
            out.write(' -s %s %s' % (section[0], section[1]))
        out.write(' -readnow')
        self.gdb_stub.execute_sync(out.getvalue())

    def set_main_symbol_file(self, symbol_file):
        self.gdb_stub.execute_sync('file %s' % symbol_file)

    def add_breakpoint(self, symbol):
        self.gdb_stub.execute_sync('break %s' % symbol)

    def clear_symbols(self):
        self.gdb_stub.execute_sync('file')

    def clear_breakpoints(self):
        self.gdb_stub.execute('delete')

    def attach_tcp(self, host, port):
        self.gdb_stub.execute_sync('target remote %s:%s' % (host, str(port)))

    def detach(self):
        self.clear_breakpoints()
        self.clear_symbols()
        self.gdb_stub.execute_sync('detach')

    def add_exit_observer(self, observer):
        self.exit_subject.add_observer(observer)

    def remove_exit_observer(self, observer):
        self.exit_subject.remove_observer(observer)

    def pause(self):
        self.gdb_stub.pause()

    def resume(self):
        self.gdb_stub.execute('continue')