o QDie@sddlZddlmZdZdZdZdZ efddZefd d Zefd d Z efd dZ efddZ efddZ efddZ efddZddZddZddZddZdS)N)REGISTRYz2 %s%s = %s, expected %s. The values for %s are: %szC %s%s changed by %f, expected %f. Value before: %s Value after: %s zZ The change in value of %s%s didn't match the predicate. Value before: %s Value after: %s z7 %s%s was None after. Value before: %s Value after: %s c KsHt|fd|i|}t|t||||tt|f}||ks"J|dSz5Asserts that metric_name{**labels} == expected_value.registryN get_metricMETRIC_EQUALS_ERR_EXPLANATION format_labels format_vectorget_metrics_vectorexpected_value metric_namerlabelsvalue assert_errrT/var/www/Datamplify/venv/lib/python3.10/site-packages/django_prometheus/testutils.pyassert_metric_equal" rc Ks~t||fi|}t|fd|i|}|dus#Jt|t|||f||p'd}t|t|||||f}||ks=J|dS)zAsserts that metric_name{**labels} changed by expected_diff between the frozen registry and now. A frozen registry can be obtained by calling save_registry, typically at the beginning of a test case. rNget_metric_from_frozen_registryr METRIC_DIFF_ERR_NONE_EXPLANATIONrMETRIC_DIFF_ERR_EXPLANATION frozen_registry expected_diffr rr saved_value current_valuediffrrrrassert_metric_diff0$  r c Ks~t||fi|}t|fd|i|}|dus#Jt|t|||f||p'd}t|t|||||f}||ks=J|dS)zAsserts that metric_name{**labels} isn't changed by expected_diff between the frozen registry and now. A frozen registry can be obtained by calling save_registry, typically at the beginning of a test case. rNrrrrrrassert_metric_no_diffJr!r"c KsHt|fd|i|}t|t||||tt|f}||ks"J|dSrrr rrrassert_metric_not_equaldrr#cKspt||fi|}t|fd|i|}|dus#Jt|t|||f|||dus6Jt|t|||fdS)zAsserts that metric_name{**labels} changed according to a provided predicate function between the frozen registry and now. A frozen registry can be obtained by calling save_registry, typically at the beginning of a test case. rNT)rrrrMETRIC_COMPARE_ERR_EXPLANATION)r predicater rrrrrrrassert_metric_comparers  r&cCstt|S)zFreezes a registry. This lets a user test changes to a metric instead of testing the absolute value. A typical use case looks like: registry = save_registry() doStuff() assert_metric_diff(registry, 1, 'stuff_done_total') )copydeepcopylistcollect)rrrr save_registrysr+cKst||fi|S)zGets a single metric.)rr*)r rrrrrrsrcCst||S)aCReturns the values for all labels of a given metric. The result is returned as a list of (labels, value) tuples, where `labels` is a dict. This is quite a hack since it relies on the internal representation of the prometheus_client, and it should probably be provided as a function there instead. )&get_metric_vector_from_frozen_registryr*)r rrrrr s r cCs@g}|D]}|jD]}|d|kr||d|dfq q|S)z4Like get_metrics_vector, but from a frozen registry.r)samplesappend)r routputmetricsamplerrrr,s  r,cKsB|D]}|jD]}|d|kr|d|kr|dSqqdS)z,Gets a single metric from a frozen registry.rr-r.N)r/)r rrr2r3rrrrs rcCsdddd|DS)zFormat a set of labels to Prometheus representation. In: {'method': 'GET', 'port': '80'} Out: '{method="GET",port="80"}' z{{{}}},cSs g|] \}}|d|dqS)z=""r).0kvrrr s z!format_labels..)formatjoinitems)rrrrrs rcCsddd|DS)zgFormats a list of (labels, value) where labels is a dict into a human-readable representation.  cSs"g|] \}}t|d|qS)z = )r)r6rrrrrr9s"z!format_vector..)r;)vectorrrrr sr )r'prometheus_clientrrrr$rrr r"r#r&r+rr r,rrr rrrrs&