def testAnonymousProfileIsFresh(self): driver = WebDriver() driver.get("http://localhost:%d/simpleTest.html" % WEB_SERVER_PORT) timestamp = time.mktime(datetime.datetime.now().timetuple()) + 100 cookie = { "name": "foo", "value": "bar", "expires": str(int(timestamp)) + "000", "domain": "localhost", "path": "/" } driver.add_cookie(utils.convert_cookie_to_json(cookie)) self.assertEquals(cookie, driver.get_cookies()[0]) driver.quit() driver = WebDriver() self.assertEquals([], driver.get_cookies()) driver.quit()
def main(): usage = 'usage: %s [options] <send|receive|delete> <id>' parser = OptionParser(usage=usage) parser.set_defaults() parser.add_option('-t', '--tags-file', dest='tagsfile', action='store', type='string', help='File to read tags from') (options, args) = parser.parse_args() if len(args) != 2: parser.error('Need to specify action and message id') if options.tagsfile is None: parser.error('Need to specify tags file') block_size = 8 max_unique_blocks = 2**16 tasks_per_message = 3 if args[0] == 'send': parser.error('Send action not implemented. Use photo donation tool.') elif args[0] == 'receive': driver = WebDriver() tags = [] for line in open(options.tagsfile, 'r'): tags.append(line.strip()) tag_pairs = [(a, b) for a in tags for b in tags if a < b] tasks = map(lambda pair: WebTagPairFlickrTask(driver, pair), tag_pairs) vector_provider = NullVectorProvider() message_layer = MessageLayer(vector_provider, block_size, max_unique_blocks, tasks, tasks_per_message, timestamper) data = message_layer.receive(args[1]) sys.stdout.write(data) elif args[0] == 'delete': parser.error("Delete action not implemented. You probably don't own the photos anyway.") else: parser.error('Invalid action')
def download(address, snippets): driver = WebDriver() tasks = [] for snippet in snippets: tasks.append(snippet.execute(driver)) vector_provider = NullVectorProvider() message_layer = MessageLayer(vector_provider, common.BLOCK_SIZE, common.MAX_UNIQUE_BLOCKS, tasks, common.TASKS_PER_MESSAGE, Timestamper(), mac=True) try: data = message_layer.receive(address) except MessageLayerError: pass driver.close() return data
def setUp(self): self.server = Popen("dev_appserver.py . --port=80", shell=True) self.browser = WebDriver()
def setup_module(module): webserver = SimpleWebServer() webserver.start() FirefoxCookieTest.webserver = webserver FirefoxCookieTest.driver = WebDriver()
def setUp(self): self._driver = WebDriver()
def main(): usage = 'usage: %s [options] <send|recieve> <id>' parser = OptionParser(usage=usage) parser.add_option('-u', '--username', dest='username', action='store', type='string', help='Twitter username') parser.add_option('-p', '--password', dest='password', action='store', type='string', help='Twitter password') parser.add_option('-f', '--file', dest='filename', action='store', type='string', help='File to send') parser.add_option('-n', '--num-tweets', dest='num_vectors', action='store', type='int', help='Number of tweets to send') parser.add_option('-r', '--send-ratio', dest='send_ratio', action='store', type='float', help='Ratio between data to send and total data length') parser.add_option('-d', '--directory', dest='directory', action='store', type='string', help='Directory to read tweets from') (options, args) = parser.parse_args() if len(args) != 2: parser.error('Need to specify action and message id') block_size = 8 max_unique_blocks = 2**8 tasks_per_message = 3 if args[0] == 'send': if options.username is None or options.password is None: parser.error( 'Must specify Twitter username and password to send messages.') if options.directory is None: parser.error( 'Must specify a directory with tweets (*.tweet) in it, to embed and upload.' ) vector_provider = DirectoryVectorProvider(BlatantVector, options.directory, '.tweet') if options.filename is None: print 'Enter message and press <Ctrl-D>' data = sys.stdin.read() else: data = open(options.filename, 'r').read() driver = WebDriver() twitter_login(driver, options.username, options.password) tasks = [WebTwitterTask(driver, options.username, BlatantVector)] message_layer = MessageLayer(vector_provider, block_size, max_unique_blocks, tasks, tasks_per_message, timestamper) if options.num_vectors is not None: message_layer.send(args[1], data, num_vectors=options.num_vectors) elif options.send_ratio is not None: message_layer.send(args[1], data, send_ratio=options.send_ratio) else: message_layer.send(args[1], data) elif args[0] == 'receive': if options.username is None: parser.error('Must specify username to search') vector_provider = NullVectorProvider() driver = WebDriver() tasks = [WebTwitterTask(driver, options.username, BlatantVector)] message_layer = MessageLayer(vector_provider, block_size, max_unique_blocks, tasks, tasks_per_message, timestamper) data = message_layer.receive(args[1]) sys.stdout.write(data) elif args[0] == 'delete': parser.error( "Deletion not supported through Web interface yet. Use direct client." ) else: parser.error('Invalid action')
def testNamedProfile(self): profile = FirefoxProfile("example") driver = WebDriver(profile) driver.get("http://localhost:%d/simpleTest.html" % WEB_SERVER_PORT) self.assertEquals("Hello WebDriver", driver.get_title()) driver.quit()
def testAnonymousProfileExample(self): driver = WebDriver() driver.get("http://localhost:%d/simpleTest.html" % WEB_SERVER_PORT) self.assertEquals("Hello WebDriver", driver.get_title()) driver.quit()
def testDoubleClose(self): self.webdriver = WebDriver() self.webdriver.close() self.webdriver.close() self.webdriver.quit()
def testLaunchAndCloseBrowser(self): self.webdriver = WebDriver() self.webdriver.quit()