o ^$@sbdZddlZddlZddlZddlmZmZddlmZGdddej Z e dkr/e dSdS)z1Define set of helper functions for the dataframe.N) SeriesHelperInfluxDBClient)ConnectionErrorcsTeZdZdZefddZddZddZdd Zd d Z d d Z ddZ Z S)TestSeriesHelperz!Define the SeriesHelper for test.cs:tt|tdddddt_Gdddt}|t_dS) z1Set up an instance of the TestSerisHelper object.ZhostiZusernameZpasswordZdatabasec@eZdZdZGdddZdS)z3TestSeriesHelper.setUpClass..MySeriesHelperz&Define a subset SeriesHelper instance.c@*eZdZdZejZdZddgZdZdZ dS)z8TestSeriesHelper.setUpClass..MySeriesHelper.Metaz0Define metadata for the TestSeriesHelper object.events.stats.{server_name}time server_nameTN __name__ __module__ __qualname____doc__rclient series_namefields bulk_size autocommitrrG/usr/lib/python3/dist-packages/influxdb/tests/influxdb08/helper_test.pyMetarNr rrrrrrrrMySeriesHelperrN)superr setUpClassrrrr)clsr __class__rrrs zTestSeriesHelper.setUpClasscCsxGdddt}t}|ddd||j_|ddd|ddd|ddd||j|dd d||jd S) z?Test that write_points called after the right number of events.c@r)z9TestSeriesHelper.test_auto_commit..AutoCommitTestz7Define an instance of SeriesHelper for AutoCommit test.c@s*eZdZdZdZddgZdZeZdZ dS)z>TestSeriesHelper.test_auto_commit..AutoCommitTest.Metaz&Define metadata AutoCommitTest object.rr r r TN) r rrrrrrrrrrrrrr/srNrrrrrAutoCommitTest,rr" us.east-1r r is N)rmockZ MagicMockZ_clientZ write_pointsZ assertFalseZcalled assertTrue)selfr"Zfake_write_pointsrrrtest_auto_commit*s      z!TestSeriesHelper.test_auto_commitcstjdddtjdddtjdddtjdddddgddgddgddggddd gd gtj|tfd d DoMtfd d Ddtj|tjgddS)8Test JSON conversion when there is only one series name.r#r$r%r&r'r(events.stats.us.east-1r r Zpointsnamecolumnscg|]}|vqSrr.0el expectationrr Pz9TestSeriesHelper.testSingleSeriesName..cr2rrr3rcvdrrr8Qr9zTInvalid JSON body of time series returned from _json_body_ for one series name: {0}.*Resetting helper did not empty datapoints.NrrZ _json_body_r*allformatZ_reset_Z assertEqualr+rr7r;rtestSingleSeriesNameBs0  z%TestSeriesHelper.testSingleSeriesNamecstjdddtjdddtjdddtjdd dddggd d d gd d dggdd d gd ddggdd d gd ddggdd d gd gtj|tfddDobtfddDdtj|tjgddS)r-r#r$r%z fr.paris-10r&zlu.luxr'z uk.londonr(zevents.stats.lu.luxr r r/zevents.stats.uk.londonzevents.stats.fr.paris-10r.cr2rrr3r6rrr8nr9z;TestSeriesHelper.testSeveralSeriesNames..cr2rrr3r:rrr8or9zYInvalid JSON body of time series returned from _json_body_ for several series names: {0}.r<Nr=r@rrArtestSeveralSeriesNamesZsB  z'TestSeriesHelper.testSeveralSeriesNamescCspGdddt}Gdddt}Gdddt}Gdddt}||||fD]}|jt|fid d d q&d S) zTest errors in invalid helpers.c@seZdZdZdS)z8TestSeriesHelper.testInvalidHelpers..MissingMetaz0Define SeriesHelper object for MissingMeta test.N)r rrrrrrr MissingMeta{srDc@r)z:TestSeriesHelper.testInvalidHelpers..MissingClientz2Define SeriesHelper object for MissingClient test.c@s eZdZdZdZddgZdZdS)z?TestSeriesHelper.testInvalidHelpers..MissingClient.Metaz)Define metadata for MissingClient object.rr r TN)r rrrrrrrrrrrs rNrrrrr MissingClientrrEc@r)z>TestSeriesHelper.testInvalidHelpers..MissingSeriesNamez2Define SeriesHelper object for MissingSeries test.c@seZdZdZddgZdS)zCTestSeriesHelper.testInvalidHelpers..MissingSeriesName.Metaz-Define metadata for MissingSeriesName object.r r N)r rrrrrrrrrs rNrrrrrMissingSeriesNamerrFc@r)z:TestSeriesHelper.testInvalidHelpers..MissingFieldsz+Define SeriesHelper for MissingFields test.c@seZdZdZdZdS)z?TestSeriesHelper.testInvalidHelpers..MissingFields.Metaz)Define metadata for MissingFields object.rN)r rrrrrrrrrsrNrrrrr MissingFieldsrrGr$r#r r N)r assertRaisesAttributeError)r+rDrErFrGrrrrtestInvalidHelpersys z#TestSeriesHelper.testInvalidHelpersc CsGdddt}tjdd(}td|t|dddWd n1s*wYWd n1s9wY|t|d d |d t t |d }| |t dd|Ddd S)z&Test warning for an invalid bulk size.c@r)z?TestSeriesHelper.testWarnBulkSizeZero..WarnBulkSizeZeroz.Define SeriesHelper for WarnBulkSizeZero test.c@r)zDTestSeriesHelper.testWarnBulkSizeZero..WarnBulkSizeZero.Metaz,Define metadata for WarnBulkSizeZero object.rr r rTNr rrrrrrrNrrrrrWarnBulkSizeZerorrLTrecordalwaysr$r#rHNI{0} call should have generated one warning.Actual generated warnings: {1} zIDefinition of bulk_size in WarnBulkSizeZero forced to 1, was less than 1.cs|] }|jjdVqdSrNmessageargsr4wrrr z8TestSeriesHelper.testWarnBulkSizeZero..z.Warning message did not contain "forced to 1".)rwarningscatch_warnings simplefilterrIrassertGreaterEquallenr?joinmapstrassertInlist)r+rL rec_warnings expected_msgrrrtestWarnBulkSizeZeros,  z%TestSeriesHelper.testWarnBulkSizeZeroc CsGdddt}tjdd}td|dddWd n1s$wY|t|d d |d tt |d }| |t dd|Ddd S)z6Test warning for a set bulk size but autocommit False.c@r)zGTestSeriesHelper.testWarnBulkSizeNoEffect..WarnBulkSizeNoEffectz4Define SeriesHelper for WarnBulkSizeNoEffect object.c@s$eZdZdZdZddgZdZdZdS)zLTestSeriesHelper.testWarnBulkSizeNoEffect..WarnBulkSizeNoEffect.Metaz0Define metadata for WarnBulkSizeNoEffect object.rr r r FN)r rrrrrrrrrrrrs rNrrrrrWarnBulkSizeNoEffectrriTrMrOr$r#rHNrPrQrRzZDefinition of bulk_size in WarnBulkSizeNoEffect has no affect because autocommit is false.csrSrTrUrXrrrrZr[z.z1Warning message did not contain the expected_msg.) rr\r]r^r_r`r?rarbrcrdre)r+rirfrgrrrtestWarnBulkSizeNoEffects& z)TestSeriesHelper.testWarnBulkSizeNoEffect) r rrr classmethodrr,rBrCrKrhrj __classcell__rrr rr s'#r__main__) rZunittestr\r)Zinfluxdb.influxdb08rrZrequests.exceptionsrZTestCaserr mainrrrrs X