1 # Copyright 2016 gRPC authors.
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
7 # http://www.apache.org/licenses/LICENSE-2.0
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
14 """Test of gRPC Python's application-layer API."""
22 from tests.unit import _from_grpc_import_star
25 class AllTest(unittest.TestCase):
28 expected_grpc_code_elements = (
30 'FutureCancelledError',
32 'ChannelConnectivity',
41 'AuthMetadataContext',
42 'AuthMetadataPluginCallback',
44 'ServerCertificateConfiguration',
46 'UnaryUnaryMultiCallable',
47 'UnaryStreamMultiCallable',
48 'StreamUnaryMultiCallable',
49 'StreamStreamMultiCallable',
50 'UnaryUnaryClientInterceptor',
51 'UnaryStreamClientInterceptor',
52 'StreamUnaryClientInterceptor',
53 'StreamStreamClientInterceptor',
62 'LocalConnectionType',
63 'local_channel_credentials',
64 'local_server_credentials',
65 'alts_channel_credentials',
66 'alts_server_credentials',
67 'unary_unary_rpc_method_handler',
68 'unary_stream_rpc_method_handler',
69 'stream_unary_rpc_method_handler',
71 'stream_stream_rpc_method_handler',
72 'method_handlers_generic_handler',
73 'ssl_channel_credentials',
74 'metadata_call_credentials',
75 'access_token_call_credentials',
76 'composite_call_credentials',
77 'composite_channel_credentials',
78 'compute_engine_channel_credentials',
79 'ssl_server_credentials',
80 'ssl_server_certificate_configuration',
81 'dynamic_ssl_server_credentials',
82 'channel_ready_future',
89 'protos_and_services',
90 'xds_channel_credentials',
91 'xds_server_credentials',
92 'insecure_server_credentials',
95 six.assertCountEqual(self, expected_grpc_code_elements,
96 _from_grpc_import_star.GRPC_ELEMENTS)
99 class ChannelConnectivityTest(unittest.TestCase):
101 def testChannelConnectivity(self):
102 self.assertSequenceEqual((
103 grpc.ChannelConnectivity.IDLE,
104 grpc.ChannelConnectivity.CONNECTING,
105 grpc.ChannelConnectivity.READY,
106 grpc.ChannelConnectivity.TRANSIENT_FAILURE,
107 grpc.ChannelConnectivity.SHUTDOWN,
108 ), tuple(grpc.ChannelConnectivity))
111 class ChannelTest(unittest.TestCase):
113 def test_secure_channel(self):
114 channel_credentials = grpc.ssl_channel_credentials()
115 channel = grpc.secure_channel('google.com:443', channel_credentials)
119 if __name__ == '__main__':
120 logging.basicConfig()
121 unittest.main(verbosity=2)