Beispiel #1
0
 def get_app(self):
     port = self.get_http_port()
     base_url = 'http://localhost:%s' % port
     self.spark = Spark(base_url=base_url)
     return tornado.web.Application([
         (spark.proxy_root + '.*', SparkHandler, {'spark': self.spark}),
         (FakeReplaceHandler.handler_root, FakeReplaceHandler),
         (FakeVerbatimHandler.handler_root, FakeVerbatimHandler),
     ])
Beispiel #2
0
# -*- coding: utf-8 -*-
import pytest
import six
import tornado
import tornado.httpclient
import tornado.testing
import tornado.web
from bs4 import BeautifulSoup
from jupyter_spark.handlers import SparkHandler
from jupyter_spark.spark import BEAUTIFULSOUP_BUILDER, Spark

PROXY_PREFIX = "/proxy/application_1234556789012_3456"
spark = Spark(base_url='http://localhost:8888')


class FakeHandler(tornado.web.RequestHandler):

    def get(self):
        self.set_header('Content-Type', self.CONTENT_TYPE)
        self.write(self.RESPONSE)


class FakeReplaceHandler(FakeHandler):
    handler_root = '/backend/replace'
    RESPONSE = six.b('<img src="/image.png" />')
    REPLACED = six.b('<img src="/spark/image.png"/>')
    CONTENT_TYPE = 'text/html'


class FakeVerbatimHandler(FakeHandler):
    handler_root = '/backend/verbatim'