1*cc02d7e2SAndroid Build Coastguard Worker# Copyright 2022 gRPC authors. 2*cc02d7e2SAndroid Build Coastguard Worker# 3*cc02d7e2SAndroid Build Coastguard Worker# Licensed under the Apache License, Version 2.0 (the "License"); 4*cc02d7e2SAndroid Build Coastguard Worker# you may not use this file except in compliance with the License. 5*cc02d7e2SAndroid Build Coastguard Worker# You may obtain a copy of the License at 6*cc02d7e2SAndroid Build Coastguard Worker# 7*cc02d7e2SAndroid Build Coastguard Worker# http://www.apache.org/licenses/LICENSE-2.0 8*cc02d7e2SAndroid Build Coastguard Worker# 9*cc02d7e2SAndroid Build Coastguard Worker# Unless required by applicable law or agreed to in writing, software 10*cc02d7e2SAndroid Build Coastguard Worker# distributed under the License is distributed on an "AS IS" BASIS, 11*cc02d7e2SAndroid Build Coastguard Worker# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12*cc02d7e2SAndroid Build Coastguard Worker# See the License for the specific language governing permissions and 13*cc02d7e2SAndroid Build Coastguard Worker# limitations under the License. 14*cc02d7e2SAndroid Build Coastguard Worker 15*cc02d7e2SAndroid Build Coastguard Workerimport functions_framework 16*cc02d7e2SAndroid Build Coastguard Workerfrom google.cloud import pubsub_v1 17*cc02d7e2SAndroid Build Coastguard Worker 18*cc02d7e2SAndroid Build Coastguard Workerps_client = pubsub_v1.PublisherClient() 19*cc02d7e2SAndroid Build Coastguard Worker_PROJECT_ID = "grpc-testing" 20*cc02d7e2SAndroid Build Coastguard Worker_PUBSUB_TOPIC = "gcf-distribtest-topic" 21*cc02d7e2SAndroid Build Coastguard Worker 22*cc02d7e2SAndroid Build Coastguard Worker 23*cc02d7e2SAndroid Build Coastguard Worker@functions_framework.http 24*cc02d7e2SAndroid Build Coastguard Workerdef test_publish(request): 25*cc02d7e2SAndroid Build Coastguard Worker topic_path = ps_client.topic_path(_PROJECT_ID, _PUBSUB_TOPIC) 26*cc02d7e2SAndroid Build Coastguard Worker message = '{"function": "TEST"}' 27*cc02d7e2SAndroid Build Coastguard Worker message_bytes = message.encode("utf-8") 28*cc02d7e2SAndroid Build Coastguard Worker 29*cc02d7e2SAndroid Build Coastguard Worker for _ in range(100): 30*cc02d7e2SAndroid Build Coastguard Worker future = ps_client.publish(topic_path, data=message_bytes) 31*cc02d7e2SAndroid Build Coastguard Worker 32*cc02d7e2SAndroid Build Coastguard Worker return "ok", 200 33