def test_missing_static_dir(self):
fake_dir = os.path.join(self.TEST_DIR,
'/this/directory/does/not/exist')
with self.assertRaises(SystemExit) as cm:
make_app(fake_dir)
assert cm.code == -1 # pylint: disable=no-member
def test_permalinks_work(self):
db = InMemoryDemoDatabase()
app = make_app(build_dir=self.TEST_DIR, demo_db=db)
predictor = CountingPredictor()
app.predictors = {"counting": predictor}
app.testing = True
client = app.test_client()
def post(endpoint: str, data: JsonDict) -> Response:
return client.post(endpoint,
content_type="application/json",
data=json.dumps(data))
data = {"some": "input"}
response = post("/predict/counting", data=data)
assert response.status_code == 200
result = json.loads(response.get_data())
slug = result.get("slug")
assert slug is not None
response = post("/permadata", data={"slug": "not the right slug"})
assert response.status_code == 400
response = post("/permadata", data={"slug": slug})
assert response.status_code == 200
result2 = json.loads(response.get_data())
assert set(
result2.keys()) == {"modelName", "requestData", "responseData"}
assert result2["modelName"] == "counting"
assert result2["requestData"] == data
assert result2["responseData"] == result
def test_permalinks_work(self):
db = InMemoryDemoDatabase()
app = make_app(build_dir=self.TEST_DIR, demo_db=db)
predictor = CountingPredictor()
app.predictors = {"counting": predictor}
app.testing = True
client = app.test_client()
def post(endpoint: str, data: JsonDict) -> Response:
return client.post(endpoint, content_type="application/json", data=json.dumps(data))
data = {"some": "input"}
response = post("/predict/counting", data=data)
assert response.status_code == 200
result = json.loads(response.get_data())
slug = result.get("slug")
assert slug is not None
response = post("/permadata", data={"slug": "not the right slug"})
assert response.status_code == 400
response = post("/permadata", data={"slug": slug})
assert response.status_code == 200
result2 = json.loads(response.get_data())
assert set(result2.keys()) == {"modelName", "requestData", "responseData"}
assert result2["modelName"] == "counting"
assert result2["requestData"] == data
assert result2["responseData"] == result
def test_disable_caching(self):
import allennlp.service.server_flask as server
server.CACHE_SIZE = 0
predictor = CountingPredictor()
app = server.make_app(build_dir=self.TEST_DIR)
app.predictors = {"counting": predictor}
app.testing = True
client = app.test_client()
data = {"input1": "this is input 1", "input2": 10}
key = json.dumps(data)
assert not predictor.calls
for i in range(5):
response = client.post("/predict/counting",
content_type="application/json",
data=json.dumps(data))
assert response.status_code == 200
assert json.loads(response.get_data()) == data
# cache is disabled, so call count should keep incrementing
assert predictor.calls[key] == i + 1
assert len(predictor.calls) == 1
def test_disable_caching(self):
import allennlp.service.server_flask as server
server.CACHE_SIZE = 0
predictor = CountingPredictor()
app = server.make_app(build_dir=self.TEST_DIR)
app.predictors = {"counting": predictor}
app.testing = True
client = app.test_client()
data = {"input1": "this is input 1", "input2": 10}
key = json.dumps(data)
assert not predictor.calls
for i in range(5):
response = client.post("/predict/counting",
content_type="application/json",
data=json.dumps(data))
assert response.status_code == 200
assert json.loads(response.get_data()) == data
# cache is disabled, so call count should keep incrementing
assert predictor.calls[key] == i + 1
assert len(predictor.calls) == 1
def setUp(self):
super().setUp()
# Create index.html in TEST_DIR
pathlib.Path(os.path.join(self.TEST_DIR, 'index.html')).touch()
if self.client is None:
self.app = make_app(build_dir=self.TEST_DIR)
self.app.predictors = PREDICTORS
self.app.testing = True
self.client = self.app.test_client()
def setUp(self):
super().setUp()
# Create index.html in TEST_DIR
(self.TEST_DIR / 'index.html').touch() # pylint: disable=no-member
if self.client is None:
self.app = make_app(build_dir=self.TEST_DIR)
self.app.predictors = PREDICTORS
self.app.testing = True
self.client = self.app.test_client()
def setUp(self):
super().setUp()
# Create index.html in TEST_DIR
pathlib.Path(os.path.join(self.TEST_DIR, 'index.html')).touch()
if self.client is None:
self.app = make_app(build_dir=self.TEST_DIR)
self.app.predictors = PREDICTORS
self.app.testing = True
self.client = self.app.test_client()
def test_permalinks_fail_gracefully_with_no_database(self):
app = make_app(build_dir=self.TEST_DIR)
predictor = CountingPredictor()
app.predictors = {"counting": predictor}
app.testing = True
client = app.test_client()
# Make a prediction, no permalinks.
data = {"some": "input"}
response = client.post("/predict/counting", content_type="application/json", data=json.dumps(data))
assert response.status_code == 200
# With permalinks not enabled, the result shouldn't contain a slug.
result = json.loads(response.get_data())
assert "slug" not in result
# With permalinks not enabled, a post to the /permadata endpoint should be a 400.
response = self.client.post("/permadata", data="""{"slug": "someslug"}""")
assert response.status_code == 400
def test_permalinks_fail_gracefully_with_no_database(self):
app = make_app(build_dir=self.TEST_DIR)
predictor = CountingPredictor()
app.predictors = {"counting": predictor}
app.testing = True
client = app.test_client()
# Make a prediction, no permalinks.
data = {"some": "input"}
response = client.post("/predict/counting", content_type="application/json", data=json.dumps(data))
assert response.status_code == 200
# With permalinks not enabled, the result shouldn't contain a slug.
result = json.loads(response.get_data())
assert "slug" not in result
# With permalinks not enabled, a post to the /permadata endpoint should be a 400.
response = self.client.post("/permadata", data="""{"slug": "someslug"}""")
assert response.status_code == 400
def test_missing_static_dir(self):
fake_dir = self.TEST_DIR / 'this' / 'directory' / 'does' / 'not' / 'exist'
with self.assertRaises(SystemExit) as cm:
make_app(fake_dir)
assert cm.code == -1 # pylint: disable=no-member
def test_missing_static_dir(self):
fake_dir = os.path.join(self.TEST_DIR, '/this/directory/does/not/exist')
with self.assertRaises(SystemExit) as cm:
make_app(fake_dir)
assert cm.code == -1 # pylint: disable=no-member
