Python EnvironmentStub.insert_users方法代码示例

# 需要导入模块: from trac.test import EnvironmentStub [as 别名]
# 或者: from trac.test.EnvironmentStub import insert_users [as 别名]
class CrashDumpWebUiTestCase(unittest.TestCase):
    def setUp(self):
        self.env = EnvironmentStub(enable=['trac.*', 'crashdump.*'])
        self.env.path = tempfile.mkdtemp()
        self.db_mgr = DatabaseManager(self.env)
        self.env.upgrade()
        #self.db = self.env.get_db_cnx()
        self.crashdump_module = CrashDumpModule(self.env)

    def tearDown(self):
        #self.db.close()
        self.env.shutdown()
        shutil.rmtree(self.env.path)

    def _create_ticket_with_change(self, old_props, new_props,
                                   author='anonymous'):
        """Create a ticket with `old_props` and apply properties
        in `new_props`.
        """
        t = Ticket(self.env)
        t.populate(old_props)
        t.insert()
        comment = new_props.pop('comment', None)
        t.populate(new_props)
        t.save_changes(author, comment=comment)
        return t

    def _insert_ticket(self, **kw):
        """Helper for inserting a ticket into the database"""
        ticket = Ticket(self.env)
        for k, v in kw.items():
            ticket[k] = v
        ticket.insert()
        with self.env.db_transaction as db:
            links = CrashDumpTicketLinks(self.env, ticket, db=db)
            if 'linked_crashes' in kw:
                links.crashes = kw['linked_crashes']
                links.save(author='anonymous', db=db)
            db.commit()
        return ticket, links

    def _insert_crashdump(self, **kw):
        """Helper for inserting a ticket into the database"""
        crash = CrashDump(env=self.env)
        for k, v in kw.items():
            crash[k] = v
        crash.insert()
        return crash

    def test_no_crash_id(self):
        req = MockRequest(self.env, authname='user', method='GET',
                          args={'without-crashid':'42'})
        self.assertRaises(ResourceNotFound,
                          self.crashdump_module.process_request, req)

    def test_non_existing_crash_id(self):
        req = MockRequest(self.env, authname='user', method='GET',
                          args={'crashid':'42'})
        self.assertRaises(ResourceNotFound,
                          self.crashdump_module.process_request, req)

    def test_action_view_crash(self):
        """Full name of reporter and owner are used in ticket properties."""
        self.env.insert_users([('user1', 'User One', ''),
                               ('user2', 'User Two', '')])
        crash = self._insert_crashdump(reporter='user1', owner='user2')
        req = MockRequest(self.env, authname='user', method='GET',
                          args={'crashid':crash.id, 'action': 'view'})
        tmpl, data, extra = self.crashdump_module.process_request(req)

        self.assertEqual(tmpl, 'report.html')


    def test_action_view_crash_child(self):
        """Full name of reporter and owner are used in ticket properties."""
        self.env.insert_users([('user1', 'User One', ''),
                               ('user2', 'User Two', '')])
        crash = self._insert_crashdump(reporter='user1', owner='user2')

        for param in ['sysinfo', 'sysinfo_ex', 'fast_protect_version_info', 'exception', 'memory_regions', 'modules', 'threads', 'memory_block', 'stackdump']:
            req = MockRequest(self.env, authname='user', method='GET',
                            args={'crashid':crash.id, 'action': 'view', 'params': [param] })
            tmpl, data, extra = self.crashdump_module.process_request(req)

            self.assertEqual(tmpl, param + '.html')

    def test_action_view_ticket_linked_crash(self):
        """Full name of reporter and owner are used in ticket properties."""
        self.env.insert_users([('user1', 'User One', ''),
                               ('user2', 'User Two', '')])
        crash = self._insert_crashdump(reporter='user1', owner='user2')
        tkt, tkt_links = self._insert_ticket(reporter='user1', owner='user2', linked_crashes='%i' % crash.id)

        req = MockRequest(self.env, authname='user', method='GET',
                          args={'crashid':crash.id, 'action': 'view'})
        tmpl, data, extra = self.crashdump_module.process_request(req)

        self.assertEqual(tmpl, 'report.html')
        self.assertEqual(crash.linked_tickets, [tkt.id])

#.........这里部分代码省略.........

# 需要导入模块: from trac.test import EnvironmentStub [as 别名]
# 或者: from trac.test.EnvironmentStub import insert_users [as 别名]
class test_commitupdater(unittest.TestCase):


    def _test_authenticated_session(self, username, fullname, email):
        """
        Verifies that a session cookie does not get used if the user is logged
        in, and that Trac expires the cookie.
        """
        req = MockRequest(self.env, authname=username)
        req.incookie['trac_session'] = '123456'
        session = Session(self.env, req)
        self.assertEqual(username, session.sid)
        session['email'] = email
        session['name'] = fullname
        session.save()

    def setUp(self):
        self.env = \
            EnvironmentStub(enable=['trac.attachment.LegacyAttachmentPolicy',
                                    'trac.perm.*',
                                    'trac.wiki.web_ui.ReadonlyWikiPolicy',
                                    'trac.ticket.*'])
        self.policy = ReadonlyWikiPolicy(self.env)
        store = perm.DefaultPermissionStore(self.env)


        self.perm_sys = perm.PermissionSystem(self.env)
        users = [('user1', 'User C', '[email protected]'),
                               ('user2', 'User A', '[email protected]'),
                               ('user3', 'User D', '[email protected]'),
                               ('user4', 'User B', '[email protected]')]
        self.env.insert_users(users)
        store.grant_permission('user1', 'TICKET_MODIFY')
        store.grant_permission('user2', 'TICKET_VIEW')
        store.grant_permission('user3', 'TICKET_MODIFY')
        store.grant_permission('user4', 'TICKET_MODIFY')

        for (username, fullname, email) in users:
            self._test_authenticated_session(username, fullname, email)

        self.repo = Mock(MockRepository, 'testrepo',
                    {'name': 'testrepo', 'id': 4321}, None)

        # Set all component objects to defaults
        config = self.env.config
        config.set("ticket","commit_ticket_update_commands.close","close closed closes fix fixed fixes")
        config.set("ticket","commit_ticket_update_commands.implements","implement implements implemented impl")
        config.set("ticket","commit_ticket_update_commands.invalidate","invalid invalidate invalidated invalidates")
        config.set("ticket","commit_ticket_update_commands.refs","addresses re references refs see")
        config.set("ticket","commit_ticket_update_commands.rejects","reject rejects rejected")
        config.set("ticket","commit_ticket_update_commands.worksforme","worksforme")
        config.set("ticket","commit_ticket_update_commands.alreadyimplemented","alreadyimplemented already_implemented")
        config.set("ticket","commit_ticket_update_commands.reopen","reopen reopens reopened")
        config.set("ticket","commit_ticket_update_commands.testready","testready test_ready ready_for_test rft")
        config.set("ticket","commit_ticket_update_allowed_domains","example.org mydomain.net")
        #config.set("ticket","commit_ticket_update_check_perms",False)

        self._add_component('component3', 'user3')

        self.ticket = Ticket(self.env)
        self.ticket.populate({
            'reporter': 'user1',
            'summary': 'the summary',
            'component': 'component3',
            'owner': 'user3',
            'status': 'new',
        })
        self.tkt_id = self.ticket.insert()

        self.ticket2 = Ticket(self.env)
        self.ticket2.populate({
            'reporter': 'user2',
            'summary': 'the summary',
            'component': 'component3',
            'owner': 'user2',
            'status': 'new',
        })
        self.tkt2_id = self.ticket2.insert()

        #for username, name, email in self.env.get_known_users():
         #   sys.stderr.write('known user %s, %s, %s\n' % (username, name, email))

        with self.env.db_transaction as db:
            db("INSERT INTO enum VALUES ('resolution', 'already_implemented', 6)")
            #db("INSERT INTO enum VALUES ('resolution', 'worksforme', 5)")

        self._committicketupdater = CommitTicketUpdater(self.env)

    def noop(self):
        pass


    def tearDown(self):
        self.env.reset_db()

    def _add_component(self, name='test', owner='owner1'):
        component = Component(self.env)
        component.name = name
        component.owner = owner
        component.insert()
#.........这里部分代码省略.........