_SrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSK r SSK J r SSK J r SSK Jr SSK Jr SrSR#S \ "S 555r"S S \R&5r"S S\5r"SS\R,5r"SS\R,5r"SS\R,5r"SS\R,5r"SS\R,5r"SS\R,5r"SS\5rSrSrg)a4Test utilities for message testing. Includes module interface test to ensure that public parts of module are correctly declared in __all__. Includes message types that correspond to those defined in services_test.proto. Includes additional test utilities to make sure encoding/decoding libraries conform. N)range) message_types)messages)utiluрусскийc#T# UHn[R"U5S-v M g7f)N)sixint2byte).0values 7lib/third_party/apitools/base/protorpclite/test_util.py r1sF:%#,,u%-:s&(c&\rSrSrSrSrSrSrg)TestCase4c U"U0UD6 URSUR-5 g!UaNn[[R"U[ U555nUR USU<SU<S35 SnAgSnAff=f)a?Check that exception is raised and text matches regular expression. Args: exception: Exception type that is expected. regexp: String regular expression that is expected in error message. function: Callable to test. params: Parameters to forward to function. kwargs: Keyword arguments to forward to function. z$Expected exception %s was not raisedzExpected match "z ", found ""N)fail__name__boolrematchstr assertTrue)self exceptionregexpfunctionparamskwargserrrs rassertRaisesWithRegexpMatch$TestCase.assertRaisesWithRegexpMatch6s} N f ' ' II<(() * N&#c(34E OOEHK$M N N Ns&)A=AA88A=c[R"U5up4[R"U5upVURX55 URXF5 g)zuCheck that two HTTP headers are the same. Args: header1: Header value string 1. header2: header value string 2. N)cgi parse_header assertEqual)rheader1header2value1params1value2params2s rassertHeaderSameTestCase.assertHeaderSameNsD**73**73 ( *rc[U5n[U5n[5nU(aIUSnUS [[U55HnXdU:XdM XG O URU5 U(aMI/nUHn URSU -5 M UHn URSU -5 M U(a$UR SSR U5-5 gg)zCheck two iterators or iterables are equal independent of order. Similar to Python 2.7 assertItemsEqual. Named differently in order to avoid potential conflict. Args: iter1: An iterator or iterable. iter2: An iterator or iterable. rz( Item from iter1 not found in iter2: %rz( Item from iter2 not found in iter1: %rzCollections not equivalent:  N)listrlenappendrjoin) riter1iter2list1list2 unmatched1item1index error_messageitems rassertIterEqualTestCase.assertIterEqualZsU U V !HEas5z*%L( + !!%(e D  :TA CD  :TA C  II5ii ./ 0 rN)r __module__ __qualname____firstlineno__r$r0rA__static_attributes__rCrrrr4sN0 +"0rrc6\rSrSrSrSrSrSrSrSr Sr g ) ModuleInterfaceTestao Test to ensure module interface is carefully constructed. A module interface is the set of public objects listed in the module __all__ attribute. Modules that that are considered public should have this interface carefully declared. At all times, the __all__ attribute should have objects intended to be publically used and all other objects in the module should be considered unused. Protected attributes (those beginning with '_') and other imported modules should not be part of this set of variables. An exception is for variables that begin and end with '__' which are implicitly part of the interface (eg. __name__, __file__, __all__ itself, etc.). Modules that are imported in to the tested modules are an exception and may be left out of the __all__ definition. The test is done by checking the value of what would otherwise be a public name and not allowing it to be exported if it is an instance of a module. Modules that are explicitly exported are for the time being not permitted. To use this test class a module should define a new class that inherits first from ModuleInterfaceTest and then from test_util.TestCase. No other tests should be added to this test case, making the order of inheritance less important, but if setUp for some reason is overidden, it is important that ModuleInterfaceTest is first in the list so that its setUp method is invoked. Multiple inheritance is required so that ModuleInterfaceTest is not itself a test, and is not itself executed as one. The test class is expected to have the following class attributes defined: MODULE: A reference to the module that is being validated for interface correctness. Example: Module definition (hello.py): import sys __all__ = ['hello'] def _get_outputter(): return sys.stdout def hello(): _get_outputter().write('Hello ') Test definition: import unittest from protorpc import test_util import hello class ModuleInterfaceTest(test_util.ModuleInterfaceTest, test_util.TestCase): MODULE = hello class HelloTest(test_util.TestCase): ... Test 'hello' module ... if __name__ == '__main__': unittest.main() cv[US5(d(URS[U5R-5 gg)zSet up makes sure that MODULE and IMPORTED_MODULES is defined. This is a basic configuration test for the test itself so does not get it's own test case. MODULEz=You must define 'MODULE' on ModuleInterfaceTest sub-class %s.N)hasattrrtyperrs rsetUpModuleInterfaceTest.setUps9 tX&& IIT ++, -'rc/nURRH1n[URU5(aM URU5 M3 U(aUR SU-5 gg)z2Test that all attributes defined in __all__ exist.z(%s of __all__ are not defined in module.N)rL__all__rMr6rrmissing_attributes attributes r testAllExist ModuleInterfaceTest.testAllExists\,,I4;; 22")))4-  II@() * rcz/n[UR5HnURS5(aMX RR;dM6[ [ URU5[ R5(aMkUS:wdMsURU5 M U(aURSU-5 gg)z.>.D.D/35K''080@0@0F0F157K((191A1A1H1H268L''080@0@0F0F157K&&q/7/?/?/D/D046J''080@0@0G0G157L%%a.6.>.>.D.D/35K##J$&-13Jrrc^\rSrSr\R "\S5r\R "\SSS9rSr g)HasOptionalNestedMessageiSrprwTrxrCN) rrDrErFrrzrr{r|rGrCrrrrSs(  " "?A 6F++OQNOrrc\rSrSrSrSrSrSrSrSr Sr S r S r S r S rS rSrSrSrSrSrSrSrSrSrSrSrSrg)ProtoConformanceTestBaseiZu Protocol conformance test base class. Each supported protocol should implement two methods that support encoding and decoding of Message objects in that format: encode_message(message) - Serialize to encoding. encode_message(message, encoded_message) - Deserialize from encoding. Tests for the modules where these functions are implemented should extend this class in order to support basic behavioral expectations. This ensures that protocols correctly encode and decode message transparently to the caller. In order to support these test, the base class should also extend the TestCase class and implement the following class attributes which define the encoded version of certain protocol buffers: encoded_partial: encoded_full: encoded_repeated: encoded_nested: > encoded_repeated_nested: , ] > unexpected_tag_message: An encoded message that has an undefined tag or number in the stream. encoded_default_assigned: encoded_nested_empty: > encoded_invalid_enum: encoded_invalid_repeated_enum: c[5nUR[RURR U5 gN)ro assertRaisesrValidationErrorPROTOLIBencode_messagermessages rtestEncodeInvalidMessage1ProtoConformanceTestBase.testEncodeInvalidMessages// (22--66 Arc&URX5 g)a Compare two encoded protocol values. Can be overridden by sub-classes to special case comparison. For example, to eliminate white space from output that is not relevant to encoding. Args: expected_encoded: Expected string encoded value. actual_encoded: Actual string encoded value. N)r))rexpected_encodedactual_encodeds rCompareEncoded'ProtoConformanceTestBase.CompareEncodeds ):rcURR[U5U5nURX#5 UR XRR U55 gr)rdecode_messagerNr)rr)rencodedexpected_messagers r EncodeDecode%ProtoConformanceTestBase.EncodeDecodesJ--..t4D/EwO )3 G]]%A%A'%JKrcLURUR[55 gr)rencoded_empty_messagerrOs rtestEmptyMessage)ProtoConformanceTestBase.testEmptyMessages $44o6GHrc[5nSUlSUlSUlSUl[R R UlURURU5 g)z#Test message with a few values set.Gz?h]a stringN) rrrrrrrrrencoded_partialrs r testPartial$ProtoConformanceTestBase.testPartialsT!##+"*,77<< $..8rc[5nSUlSUlSUlSUlSUlSUlSUlSUl[RRUl URURU5 g ) zTest all types.rr$>_rT a stringя a bytesN)rrrrrrrrrrrrr encoded_fullrs rtestFull!ProtoConformanceTestBase.testFullst!##"++"!00,77<< $++W5rcR[5nSS/UlSS/UlSS/UlSS/UlS S /UlS S /UlS S/UlSS/Ul[RR[RR/Ul URURU5 g)zTest repeated fields.rgffffff@rg?rrrriTFranother stringrs another bytesN)rrrrrrrrrrrrrrencoded_repeatedrs r testRepeated%ProtoConformanceTestBase.testRepeateds!# $c{#Sk,b1 ,b1#Sk"E] 13DE13CD-88==-88==? $//9rc~[5nSUl[5nXlUR UR U5 g)zTest nested messages.rN)rortrvr{rencoded_nested)rnested_messagers r testNested#ProtoConformanceTestBase.testNesteds4&!,"$' $--w7rc[5nSUl[5nSUl[5nX/UlUR UR U5 g)zTest repeated nested messages.rrN)rortrvr|rencoded_repeated_nested)rnested_message1nested_message2rs rtestRepeatedNested+ProtoConformanceTestBase.testRepeatedNestedsI'/"-'/"3"$#2"D $66@rc^[5nSUlURURU5 g)z,Test that encoding str on StringField works.LatinN)rrrencoded_string_typesrs rtestStringTypes(ProtoConformanceTestBase.testStringTypess(!#& $33W=rc[5nUR[RSURR U5 g)z.Test that cannot encode uninitialized message.z7Message NestedMessage is missing required field a_valueN)ror$rrrr)rrqs rtestEncodeUninitialized0ProtoConformanceTestBase.testEncodeUninitializeds5 ? (()A)A*B)-)E)E)1  3rcURR[UR5nUR [5U5 UR URURR U55 g)z-Test decoding and encoding unexpected fields.N)rrrunexpected_tag_messager)r)rloaded_messages rtestUnexpectedField,ProtoConformanceTestBase.testUnexpectedField!sa55 T88: *N; 44--66~F HrcLURUR[55 g)z7Test that default is not sent when nothing is assigned.N)rrr~rOs rtestDoNotSendDefault-ProtoConformanceTestBase.testDoNotSendDefault-s $44jlCrc[5n[RRUlURURU5 g)z2Test that default is sent when explcitly assigned.N)r~rtrrencoded_default_assignedrs r!testSendDefaultExplicitlyAssigned:ProtoConformanceTestBase.testSendDefaultExplicitlyAssigned1s2,$,,44 $77Arcn[5n[5UlURURU5 gz%Test encoding a nested empty message.N)rrr{rencoded_nested_emptyrs rtestEncodingNestedEmptyMessage7ProtoConformanceTestBase.testEncodingNestedEmptyMessage9s**,(* $33W=rc[5n[5[5/UlURURU5 gr)rrr|rencoded_repeated_nested_emptyrs r&testEncodingRepeatedNestedEmptyMessage?ProtoConformanceTestBase.testEncodingRepeatedNestedEmptyMessage@s3*,#2#4o6G"H $< *--..w7 ;;WErc "SS[R5n[R"SSSSSSS 5nU"US 9nURR XRR U55nUR URU5 g ) z7Test that DateTimeFields are encoded/decoded correctly.c8\rSrSr\R "S5rSrg)BProtoConformanceTestBase.testDateTimeNoTimeZone..MyMessageifrprCNrrDrErFr DateTimeFieldr rGrCrr MyMessagerf!//2Errrpr $xr N)rMessagedatetimerrrr)r rrr rrs rtestDateTimeNoTimeZone/ProtoConformanceTestBase.testDateTimeNoTimeZonecsv 3(( 3!!$1b"b&A%(--.. }}33G<> .rc D"SS[R5n[R"SSSSSSS [R"S 55nU"US 9nUR R XR RU55nURURU5 g ) z$Test DateTimeFields with time zones.c8\rSrSr\R "S5rSrg)DProtoConformanceTestBase.testDateTimeWithTimeZone..MyMessageirrprCNrrCrrrr-rrrrr rprr!r"r#r$ir%N) rr&r'rTimeZoneOffsetrrrr)r r(s rtestDateTimeWithTimeZone1ProtoConformanceTestBase.testDateTimeWithTimeZoneos 3(( 3!!$1b"b&"&"5"5f"=?%(--.. }}33G<> .rrCN)rrDrErFrmrrrrrrrrrrrrrrrrr r rrr)r/rGrCrrrrZsYvA ;L I 9 6: 8 A> 3 HDB>GE = F / /rrc[R"[R[R5nURS5 UR 5SnUR 5 U$!UR 5 f=f)zFind an unused port to use in tests. Derived from Damon Kohlers example: http://code.activestate.com/recipes/531822-pick-unused-port ) localhostrrp)socketAF_INET SOCK_STREAMbind getsocknameclose)tempports rpick_unused_portr;}s^ ==);); rNs$    4/+ 8 F5:F FH0x  H0VF(&F(R5H$$5 Mx''M<!!< 4h&&4&"3h&&"3JOx//O`/v`/F  'r