a Wa@s<ddlmZddlZddlmZdZGdddejjZ dS))StringION)trafficz&testdata/traffic-sample-very-short.txtc@s,eZdZddZddZddZddZd S) TrafficEmulatorTestscCst|_dSN)r TrafficModelmodelselfr =/usr/lib/python3/dist-packages/samba/tests/emulate/traffic.pysetUpszTrafficEmulatorTests.setUpcCs|`dSr)rrr r r tearDown!szTrafficEmulatorTests.tearDownc CsRt}tt}tj|gdd\}}}}|||gdgddgddgdggddgd gd gd gdgd gd }gdgddgdgddgdgd}d|_dd|j D} dd|j D} | || | || t }| |t} |d| |dd| j D} dd| j D} | || | || dS)NZinclude)Zdns_mode)dns:0rrldap:3)rrrwait:0cldap:3rpc_netlogon:29)rrrr-ldap:2r kerberos:) rr)rrrrrrrr)rr)rrrrrrrr)rrrrr r ZNetlogonr r r rr) r r r r r r r r r r r r r r r r r r r r asubschemaSubentry,dsServiceName,namingContexts,defaultNamingContext,schemaNamingContext,configurationNamingContext,rootDomainNamingContext,supportedControl,supportedLDAPVersion,supportedLDAPPolicies,supportedSASLMechanisms,dnsHostName,ldapServiceName,serverName,supportedCapabilitiesr r r 2zDC,DCr Zcnr r r r )rrrrrrcSsi|]\}}|t|qSr sorted.0kvr r r SzGTrafficEmulatorTests.test_parse_ngrams_dns_included..cSsi|]\}}|t|qSr r'r)r r r r-Tr.rcSsi|]\}}|t|qSr r'r)r r r r-`r.cSsi|]\}}|t|qSr r'r)r r r r-ar.)rropen TEST_FILEingest_summariescloselearnmaxDiffngramsitems query_details assertEqualrsaveseekload) r rf conversationsintervalduration dns_countsexpected_ngramsexpected_query_detailsr5detailsmodel2r r r test_parse_ngrams_dns_included$sZ        z3TrafficEmulatorTests.test_parse_ngrams_dns_includedc Cs:tt}t|g\}}}}||j||ddgdgdgddgdgdgdgdgdgd }gd d gd gd d gdgd}d|_dd|jj D}dd|jj D} | ||| || t }|j |t} |d| |dd| j D}dd| j D} | ||| || dS)Nrrrrrrr) r)rrrrrrrrrrr!r"r#r$r )rrrrrr&cSsi|]\}}|t|qSr r'r)r r r r-r.z:TrafficEmulatorTests.test_parse_ngrams..cSsi|]\}}|t|qSr r'r)r r r r-r.rcSsi|]\}}|t|qSr r'r)r r r r-r.cSsi|]\}}|t|qSr r'r)r r r r-r.)r/r0rr1r2rr3r4r5r6r7r8rr9rr:r;) r r<r=r>r?r@rArBr5rCrDr r r test_parse_ngramsesP         z&TrafficEmulatorTests.test_parse_ngramsN)__name__ __module__ __qualname__r r rErFr r r r rsAr) Z samba.compatrZ samba.testsZsambaZ samba.emulaterr0ZtestsZTestCaserr r r r s