66from hazelcast .connection import _Reader
77from hazelcast .errors import _ErrorsCodec
88from hazelcast .protocol import ErrorHolder
9- from hazelcast .protocol .builtin import CodecUtil , FixSizedTypesCodec , ByteArrayCodec , DataCodec , EntryListCodec , \
10- StringCodec , EntryListUUIDListIntegerCodec , EntryListUUIDLongCodec , ListMultiFrameCodec , ListIntegerCodec , \
11- ListLongCodec , ListUUIDCodec , MapCodec
9+ from hazelcast .protocol .builtin import (
10+ CodecUtil ,
11+ FixSizedTypesCodec ,
12+ ByteArrayCodec ,
13+ DataCodec ,
14+ EntryListCodec ,
15+ StringCodec ,
16+ EntryListUUIDListIntegerCodec ,
17+ EntryListUUIDLongCodec ,
18+ ListMultiFrameCodec ,
19+ ListIntegerCodec ,
20+ ListLongCodec ,
21+ ListUUIDCodec ,
22+ MapCodec ,
23+ )
1224from hazelcast .protocol .client_message import *
1325from hazelcast .protocol .codec import client_authentication_codec
1426from hazelcast .protocol .codec .custom .error_holder_codec import ErrorHolderCodec
@@ -144,9 +156,15 @@ def test_entry_list(self):
144156 EntryListCodec .encode_nullable (self .buf , None , StringCodec .encode , StringCodec .encode , True )
145157 message = self .write_and_decode ()
146158 message .next_frame () # initial frame
147- self .assertEqual (entries , EntryListCodec .decode (message , StringCodec .decode , StringCodec .decode ))
148- self .assertEqual (entries , EntryListCodec .decode_nullable (message , StringCodec .decode , StringCodec .decode ))
149- self .assertIsNone (EntryListCodec .decode_nullable (message , StringCodec .decode , StringCodec .decode ))
159+ self .assertEqual (
160+ entries , EntryListCodec .decode (message , StringCodec .decode , StringCodec .decode )
161+ )
162+ self .assertEqual (
163+ entries , EntryListCodec .decode_nullable (message , StringCodec .decode , StringCodec .decode )
164+ )
165+ self .assertIsNone (
166+ EntryListCodec .decode_nullable (message , StringCodec .decode , StringCodec .decode )
167+ )
150168
151169 def test_uuid_integer_list_entry_list (self ):
152170 self .mark_initial_frame_as_non_final ()
@@ -200,8 +218,12 @@ def test_list(self):
200218 self .assertEqual (l , ListMultiFrameCodec .decode (message , StringCodec .decode ))
201219 self .assertEqual (l , ListMultiFrameCodec .decode_nullable (message , StringCodec .decode ))
202220 self .assertIsNone (ListMultiFrameCodec .decode_nullable (message , StringCodec .decode ))
203- self .assertEqual (l , ListMultiFrameCodec .decode_contains_nullable (message , StringCodec .decode ))
204- self .assertEqual ([None ], ListMultiFrameCodec .decode_contains_nullable (message , StringCodec .decode ))
221+ self .assertEqual (
222+ l , ListMultiFrameCodec .decode_contains_nullable (message , StringCodec .decode )
223+ )
224+ self .assertEqual (
225+ [None ], ListMultiFrameCodec .decode_contains_nullable (message , StringCodec .decode )
226+ )
205227
206228 def test_uuid_list (self ):
207229 self .mark_initial_frame_as_non_final ()
@@ -223,7 +245,9 @@ def test_map(self):
223245 message = self .write_and_decode ()
224246 message .next_frame () # initial frame
225247 self .assertEqual (m , MapCodec .decode (message , StringCodec .decode , StringCodec .decode ))
226- self .assertEqual (m , MapCodec .decode_nullable (message , StringCodec .decode , StringCodec .decode ))
248+ self .assertEqual (
249+ m , MapCodec .decode_nullable (message , StringCodec .decode , StringCodec .decode )
250+ )
227251 self .assertIsNone (MapCodec .decode_nullable (message , StringCodec .decode , StringCodec .decode ))
228252
229253 def test_string (self ):
@@ -262,8 +286,9 @@ def setUp(self):
262286 self .builder = ClientMessageBuilder (lambda m : self .counter .increment ())
263287
264288 def test_unfragmented_message (self ):
265- request = client_authentication_codec .encode_request ("dev" , "user" , "pass" , uuid .uuid4 (),
266- "PYH" , 1 , "4.0" , "python" , [])
289+ request = client_authentication_codec .encode_request (
290+ "dev" , "user" , "pass" , uuid .uuid4 (), "PYH" , 1 , "4.0" , "python" , []
291+ )
267292 self .reader .read (request .buf )
268293 message = self .reader ._read_message ()
269294 self .builder .on_message (message )
0 commit comments