o ;i@s<UdZddlZddlmZddlmZmZmZmZm Z ddl m Z m Z edZ e jdde jd de jd de jd de jd de jd de jdde jdde jdde jdde jdde jdde jdde jdde jddiZeeeegeffe d<de deddfddZ!d de"e deede fddZ#dS)!zUtilities to generate random valid Protobuf messages for testing. This is based on https://github.com/yupingso/randomproto but customizable for Modal, with random seeds, and it supports oneofs, and Protobuf v4. N)Random)AnyCallableOptionalTypeVarUnion) DescriptorFieldDescriptorTcC |ddSNr normalvariaterandrP/home/ubuntu/.local/lib/python3.10/site-packages/modal/_utils/rand_pb_testing.py rcCr r rrrrrrrcCtj|ddddSNlittleTsignedint from_bytes randbytesrrrrrcCrNrTrrrrrrrr cCt|ddSNrrrrrrrrcCr#Nr"rrrrrrrr%cCrrrrrrrrr cCrr!rrrrrrr cCr#r$rrrrrrr%cCr#r&rrrrrrr%cCrrrrrrrrr cCrr!rrrrrrr cCs|ddgS)NTFchoicerrrrrscs&dfddttdDS)Nc3s|] }tjVqdSN)r(string printable).0_rrr s  z..333333?)joinranger expovariaterrrrrscCs|t|dS)Nr0)rrr3rrrrr!s_FIELD_RANDOM_GENERATORdescrreturncsdt}|jD]}|t|jdg}|dur||jq|jD]}|jdur.|j|vr.q!|jt j k}t |dr=|j }n|j t jk}|rnt||j} |rf|dd} t| D] } | } t| |j|qWq!t| |j|q!|jt jkrdd|jjDfdd} nt|j} |r|dd} t||j} t| D] } | | |qq!t||j| |q!dS)N is_repeatedrcSsg|]}|jqSr)number)r-xrrr Cr%z_fill..cs |Sr*r'r enum_valuesrrrDs z_fill..)setoneofsr(listfieldsaddnamecontaining_oneoftyper TYPE_MESSAGEhasattrr7labelLABEL_REPEATEDgetattrrandintr2_fill message_type TYPE_ENUM enum_typevaluesr4appendsetattr)msgr5r oneof_fieldsoneof oneof_fieldfield is_messager7 msg_fieldnumr.element generatorrr<rrL%sD              rLprotocCs(|durtd}|}t||j||S)zGenerate a pseudorandom protobuf message. ```python notest rand = random.Random(42) definition = rand_pb(api_pb2.Function, rand) ``` Nr)rrL DESCRIPTOR)r]rrSrrrrand_pbQs r_r*)$__doc__r+randomrtypingrrrrrgoogle.protobuf.descriptorrr r TYPE_DOUBLE TYPE_FLOAT TYPE_INT32 TYPE_INT64 TYPE_UINT32 TYPE_UINT64 TYPE_SINT32 TYPE_SINT64 TYPE_FIXED32 TYPE_FIXED64 TYPE_SFIXED32 TYPE_SFIXED64 TYPE_BOOL TYPE_STRING TYPE_BYTESr4dictr__annotations__rLrEr_rrrrs0                $,