✘✘ GRAYBYTE WORDPRESS FILE MANAGER ✘✘

3

���h�9�@s�dZd
Z
yddlZWnek
r0


ddlZYnXddlmZ
ddlmZ
ddlm	Z	
ddlm
Z

ddlmZ
dd	lmZ
dd
lm
Z

ddlmZ
ddlmZ
dd
lmZ
dZGdd�dej�ZGdd�dej�Zedkr�ej�
dS)z�Unittest that directly tests the output of the pure-Python protocol
compiler.  See //google/protobuf/internal/reflection_test.py for a test which
further ensures that we can use Python protocol message objects as we expect.
z#robinson@google.com (Will Robinson)�N)
�test_bad_identifiers_pb2)
�unittest_custom_options_pb2)
�unittest_import_pb2)
�unittest_import_public_pb2)
�unittest_mset_pb2)
�unittest_mset_wire_format_pb2)
� unittest_no_generic_services_pb2)
�unittest_pb2)
�service)
�symbol_databasei c@s�eZ
dZd
d�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zd#d$�Zd%d&�Zd'd(�Zd)d*�Zd+d,�Zd-S).�
GeneratorTestc
Cs(d
}
tj
}|j|jj|jj|
j�
dS)N�optional_nested_message)r	�TestAllTypes�assertEqual�
NestedMessage�
DESCRIPTOR�fields_by_nameZmessage_type)�self�
field_nameZ
proto_type�r�$/usr/lib/python3.6/generator_test.py�testNestedMessageDescriptor@s





z)GeneratorTest.testNestedMessageDescriptorc
Cs�|jd
t
j�
|jdt
j�
|jdt
j�
t
j�}
|jd|
j�
|jdt
jj�
|jd|
j�
|jdt
jj�
|jd|
j�
|jdt
jj�
dS)N����
��)	rr	ZFOREIGN_FOOZFOREIGN_BARZFOREIGN_BAZr�FOOZBARZBAZ)r�protorrr�	testEnumsGs







zGeneratorTest.testEnumsc
s�tj
�}
d
d���f
dd�}|j||
j�
�

|j|
jdk�

|j||
j�
�

|j|
jdk�

|j�|
j�
�

|j||
j�
�

|j|
jdk�

|j||
j�
�

|j|
jdk�

|j�|
j�
�

|j	d|
j
�
dS)Nc

Ss||kS)
Nr)
�valrrr�isnan\sz5GeneratorTest.testExtremeDefaultValues.<locals>.isnanc

s�|�
o�|d
�
S)Nr
r)
r!)
r"rr�isinf_sz5GeneratorTest.testExtremeDefaultValues.<locals>.isinfr
z? ? ?? ?? ??? ??/ ??-)r	ZTestExtremeDefaultValues�
assertTrueZ
inf_doubleZneg_inf_doubleZ
nan_doubleZ	inf_floatZ
neg_inf_floatZ	nan_floatrZcpp_trigraph)r�messager#r)
r"r�testExtremeDefaultValuesWs









z&GeneratorTest.testExtremeDefaultValuesc
s>tj
j}
d
d
d
dd��t�f
dd�|
jD�
�
}|j�|�
dS)NFT)Zoptional_int32Zrepeated_int32r
Z
default_int32c
s"g|]}
|
j�kr|
j|
j
f�qSr)�nameZhas_default_value)�.0�
f)
�expected_has_default_by_namerr�
<listcomp>{s

z6GeneratorTest.testHasDefaultValues.<locals>.<listcomp>)r	rr�dict�fieldsr)r�descZhas_default_by_namer)
r*r�testHasDefaultValuesps






z"GeneratorTest.testHasDefaultValuesc

Cs.|jt
jjt
jj�
|jt
jjjt
jj�
dS)
N)rr	�optional_int32_extension�containing_type�TestAllExtensionsr�TestRequired�single)
rrrr�'testContainingTypeBehaviorForExtensions�s





z5GeneratorTest.testContainingTypeBehaviorForExtensionsc

Cs*|jt
jjd�
|jt
jjjt
jj�
dS)
N)rr	r0Zextension_scoper3r4r)
rrrr�testExtensionScope�s




z GeneratorTest.testExtensionScopec
CsB|jt
jj�

|jt
jjj�

t
jj}
|
jd
}|j|j�

dS)N�
a)r$r	r0Zis_extensionr3r4rr)rZmessage_descriptorZnon_extension_descriptorrrr�testIsExtension�s





zGeneratorTest.testIsExtensionc
Cstj
�}
|j|
jj�j�

dS)
N)rZTestMessageSetr$r�
GetOptionsZmessage_set_wire_format)rrrrr�testOptions�s

zGeneratorTest.testOptionsc
Cs*tj
�}
|
jjd
j�}|j|dk	�

dS)NZAnEnum)rZTestMessageWithCustomOptionsr�enum_types_by_namer9r$)rrZenum_optionsrrr�testMessageWithCustomOptions�s


z*GeneratorTest.testMessageWithCustomOptionsc

Cs\|jt
tjjj�
t
tjjjtjjjtjjjg�
�
|jtj	jjg�
|jtjjjjg�
dS)
N)
r�setr	rrZnested_typesr�
OptionalGroup�
RepeatedGroup�TestEmptyMessage)
rrrr�testNestedTypes�s







zGeneratorTest.testNestedTypesc

Cst|jt
jjjdk�

|jt
jjjdk�

|jt
jjjjt
jj�
|jt
jjjjt
jj�
|jt
jjjjt
jj�
dS)
N)	r$r	r@rr1rrrr?)
rrrr�testContainingType�s

















z GeneratorTest.testContainingTypec

Cs*|jt
jjdk�

|jt
jjt
jj�
dS)
N)r$r	�_FOREIGNENUMr1r�_TESTALLTYPES_NESTEDENUMrr)
rrrr�"testContainingTypeInEnumDescriptor�s



z0GeneratorTest.testContainingTypeInEnumDescriptorc
Cs||jt
jjjjd
�
t
jjj}
|j|
jjd
�
|jtjjjjd�
|jt
j	jjd
�
|jt
j
jjd
�
|jtjjjd�
dS)N�protobuf_unittestZprotobuf_unittest_import)rr	rr�file�packagerrZ
ImportMessagerCrDZ_IMPORTENUM)rr.rrr�testPackage�s 

















zGeneratorTest.testPackagec

CsJ|jt
jjjg�
|jt
jjjd
tfg
�
|jt
jjjdddtfg�
dS)	Nr�*�+�/�i
)rJrK)rLrM)rr	rrZextension_rangesr2�
MAX_EXTENSIONZTestMultipleExtensionRanges)
rrrr�testExtensionRange�s







z GeneratorTest.testExtensionRangec

Csr|jt
jjd
�
|jt
jjd�
|jt
jjdk�

|jt
jjtjg
�
|jtjjt	jg
�
|jtjj
t	jg
�
dS)Nzgoogle/protobuf/unittest.protorF)rr	rr'rH�assertFalseZ
serialized_pbZdependenciesrrZpublic_dependencies)
rrrr�testFileDescriptor�s















z GeneratorTest.testFileDescriptorc

CsR|jt
td
��

|jt
td��

|jt
td��

t
td�rN|jttjtj��

dS)NZTestMessagerZtest_extension�TestService)r$�hasattrrrP�
issubclassrRr
ZService)
rrrr�testNoGenericServices�s






z#GeneratorTest.testNoGenericServicesc
Cs6tj
}
|jtj|
jtjj�
|jtjj|
jk�

dS)
N)r	rrZ
_TESTALLTYPESZmessage_types_by_namer'rPZ_TESTALLTYPES_NESTEDMESSAGE)r�	file_typerrr�testMessageTypesByName�s





z$GeneratorTest.testMessageTypesByNamec
Cs"tj
}
|jtj|
jtjj�
dS)
N)r	rrrCr;r')rrVrrr�testEnumTypesByName�s



z!GeneratorTest.testEnumTypesByNamec
Cs"tj
}
|jtj|
jtjj�
dS)
N)r	rrZmy_extension_stringZextensions_by_namer')rrVrrr�testExtensionsByName
s



z"GeneratorTest.testExtensionsByNamec
CsDtj
�}
|jd
|
jj�
tj�}|jd
|j�
|jtjtjk�

dS)Nr
)	r	rrZoptional_public_import_message�
erZPublicImportMessager$r)rZall_type_protoZpublic_import_protorrr�testPublicImports

s



zGeneratorTest.testPublicImportsc
Cs\tj
�}
|j|
jtjd
�
|j|
jtjd�
|j|
jtjd�
|j|
jtjd�
dS)NZfooZbarZbazZqux)rZTestBadIdentifiersrZ
Extensionsr%Z
descriptorZ
reflectionr
)rr%rrr�testBadIdentifiers
s







z GeneratorTest.testBadIdentifiersc
Cs�tj
j}
|jd
t|
j�
�
|jd|
jdj�
|jd|
jdj�
|j|
|
jdj	�
|j|
jd|
j
d�
tddddg�
}|j|tdd	�|
jdjD�
�
�
xDt
|
jj��
D]2\}}||kr�|j|
jd|j�
q�|j|j�

q�WdS)
NrZoneof_fieldr
Zoneof_uint32Zoneof_nested_messageZoneof_stringZoneof_bytesc
Ssg|]
}
|
j�qSr)
r')r(Zfieldrrrr+-
sz+GeneratorTest.testOneof.<locals>.<listcomp>)r	rrr�lenZoneofsr'�indexZassertIsr1Zoneofs_by_namer=r-�listr�itemsZcontaining_oneofZassertIsNone)rr.Znested_namesrZ
field_descrrr�	testOneof"
s














zGeneratorTest.testOneofN)�__name__�
__module__�__qualname__rr r&r/r5r6r8r:r<rArBrErIrOrQrUrWrXrYr[r\rarrrrr>s,
rc@s(eZ
dZd
Zdd�Zdd�Zdd�ZdS)	�SymbolDatabaseRegistrationTestz?Checks that messages, enums and files are correctly registered.c

Cs�|jt
jtj�jd
�
�
|jt
jjtj�jd�
�
|jt�
�
tj�jd�

WdQRX|jt
jj	tj�jd�
�
|jt
jj
tj�jd�
�
dS)Nzprotobuf_unittest.TestAllTypesz,protobuf_unittest.TestAllTypes.NestedMessagezprotobuf_unittest.NestedMessagez,protobuf_unittest.TestAllTypes.OptionalGroupz,protobuf_unittest.TestAllTypes.RepeatedGroup)rr	rr�DefaultZ	GetSymbolrZassertRaises�KeyErrorr>r?)
rrrr�
testGetSymbol8
s"
















z,SymbolDatabaseRegistrationTest.testGetSymbolc

Cs8|jd
t
j�jjd
�
j�
|jdt
j�jjd�
j�
dS)Nzprotobuf_unittest.ForeignEnumz)protobuf_unittest.TestAllTypes.NestedEnum)rrrf�poolZFindEnumTypeByNameZ	full_name)
rrrrr K
s










z(SymbolDatabaseRegistrationTest.testEnumsc

Cs|jd
t
j�jjd
�
j�
dS)Nzgoogle/protobuf/unittest.proto)rrrfriZFindFileByNamer')
rrrr�testFindFileByNameU
s




z1SymbolDatabaseRegistrationTest.testFindFileByNameN)rbrcrd�__doc__rhr rjrrrrre5
s

re�__main__)rk�
__author__Z	unittest2Zunittest�ImportErrorZgoogle.protobuf.internalrZgoogle.protobufrrrrrrr	r
rrNZTestCaserrerb�mainrrrr�<module>'s*











x&