def runTest(self): # drop_site should raise an exception when dropping a nonexistent site self.assertRaises(rapidrecdb.SiteNotFoundError, rapidrecdb.drop_site, self.sitename) self.assert_(self.sitename not in rapidrecdb.get_site_list(), "get_site_list should not include the not-yet-existent test site") rapidrecdb.add_site(self.sitename) self.assert_(self.sitename in rapidrecdb.get_site_list(), "get_site_list should contain the newly-created test site") # add_site should raise an exception when creating a site that already exists self.assertRaises(rapidrecdb.DuplicateSiteError, rapidrecdb.add_site, self.sitename) rapidrecdb.drop_site(self.sitename) self.assert_(self.sitename not in rapidrecdb.get_site_list(), "get_site_list should no longer include the test site after it was dropped")
def setUp(self): self.sitename = 'test' if self.sitename in rapidrecdb.get_site_list(): rapidrecdb.drop_site(self.sitename) rapidrecdb.add_site(self.sitename) self.hostname = '%s.localhost' % self.sitename #self.hostname = '%s.rapidrec.com' % self.sitename self.conn = CachedHTTPConnection(self.hostname)
import socket realsocket = socket.socket def socketwrap(family=socket.AF_INET, type=socket.SOCK_STREAM, proto=0): sockobj = realsocket(family, type, proto) sockobj.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1) return sockobj # socket.socket = socketwrap sitename = "test" if sitename in rapidrecdb.get_site_list(): rapidrecdb.drop_site(sitename) rapidrecdb.add_site(sitename) hostname = "%s.localhost" % sitename # self.hostname = '%s.rapidrec.com' % self.sitename numUsers = 10000 numItems = 10000 def randUsername(): return "user%s" % randint(1, numUsers) def randItemname():