Skip to content

Commit e7baad9

Browse files
committed
Make generate_configs response streaming optional
1 parent ce3b35b commit e7baad9

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/server.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,10 @@ def log_stream():
100100
except queue.Empty:
101101
pass
102102

103-
return Response(stream_with_context(log_stream()), mimetype='text/plain')
103+
if request.args.get("stream_response", "").lower() in ["1","true"]:
104+
return Response(stream_with_context(log_stream()), mimetype='text/plain')
105+
else:
106+
return Response(''.join(log_stream()), mimetype='text/plain')
104107

105108
@app.route("/maps", methods=['GET'])
106109
def maps():

tests/permissions_tests.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -79,9 +79,6 @@ def __run_config_generator(self, generator_config):
7979
url = "/generate_configs?tenant=default&use_cached_project_metadata=true"
8080
response = self.app.post(url)
8181

82-
# Wait for full response
83-
data = response.get_data(as_text=True)
84-
8582
if response.status_code != 200:
8683
print(response.text)
8784
self.assertEqual(response.status_code, 200)

0 commit comments

Comments
 (0)