def testWaitForFlusherAndGarbageCollect(self, deleteMetricMock, getAllCustomMetricsMock, _timeSleepMock):
oldFlusherMetricName = delete_companies._DATA_PATH_FLUSHER_METRIC_PREFIX + "0"
flusherMetricName = delete_companies._DATA_PATH_FLUSHER_METRIC_PREFIX + "1"
getAllCustomMetricsResultGroups = [
[{"name": "XIGNITE.FOOBAR.CLOSINGPRICE"}],
[{"name": "XIGNITE.FOOBAR.CLOSINGPRICE"}, {"name": oldFlusherMetricName}],
[{"name": "XIGNITE.FOOBAR.CLOSINGPRICE"}, {"name": flusherMetricName}],
]
# Patch getAllCustomMetrics to return all negatives and positives
getAllCustomMetricsMock.side_effect = iter(getAllCustomMetricsResultGroups)
# Execute
delete_companies._waitForFlusherAndGarbageCollect(
engineServer="host", engineApiKey="apikey", flusherMetricName=flusherMetricName
)
# Validate
self.assertEqual(getAllCustomMetricsMock.call_count, len(getAllCustomMetricsResultGroups))
expectedDeleteMetricCalls = [
mock.call(host="host", apiKey="apikey", metricName=oldFlusherMetricName),
mock.call(host="host", apiKey="apikey", metricName=flusherMetricName),
]
self.assertSequenceEqual(deleteMetricMock.mock_calls, expectedDeleteMetricCalls)
def testWaitForFlusherAndGarbageCollect(self, deleteMetricMock,
getAllCustomMetricsMock,
_timeSleepMock):
oldFlusherMetricName = (
delete_companies._DATA_PATH_FLUSHER_METRIC_PREFIX + "0")
flusherMetricName = (
delete_companies._DATA_PATH_FLUSHER_METRIC_PREFIX + "1")
getAllCustomMetricsResultGroups = [
[{
"name": "XIGNITE.FOOBAR.CLOSINGPRICE"
}],
[{
"name": "XIGNITE.FOOBAR.CLOSINGPRICE"
}, {
"name": oldFlusherMetricName
}],
[{
"name": "XIGNITE.FOOBAR.CLOSINGPRICE"
}, {
"name": flusherMetricName
}],
]
# Patch getAllCustomMetrics to return all negatives and positives
getAllCustomMetricsMock.side_effect = iter(
getAllCustomMetricsResultGroups)
# Execute
delete_companies._waitForFlusherAndGarbageCollect(
engineServer="host",
engineApiKey="apikey",
flusherMetricName=flusherMetricName)
# Validate
self.assertEqual(getAllCustomMetricsMock.call_count,
len(getAllCustomMetricsResultGroups))
expectedDeleteMetricCalls = [
mock.call(host="host",
apiKey="apikey",
metricName=oldFlusherMetricName),
mock.call(host="host",
apiKey="apikey",
metricName=flusherMetricName)
]
self.assertSequenceEqual(deleteMetricMock.mock_calls,
expectedDeleteMetricCalls)
def testWaitForFlusherAndGarbageCollectWithFlusherNotFound(self, getAllCustomMetricsMock, timeMock, sleepMock):
time_test_utils.configureTimeAndSleepMocks(timeMock, sleepMock)
flusherMetricName = delete_companies._DATA_PATH_FLUSHER_METRIC_PREFIX + "1"
# Patch getAllCustomMetrics to return all negatives and positives
getAllCustomMetricsMock.return_value = [{"name": "XIGNITE.FOOBAR.CLOSINGPRICE"}]
# Execute and validate
with self.assertRaises(delete_companies.FlusherMetricNotFound):
delete_companies._waitForFlusherAndGarbageCollect(
engineServer="host", engineApiKey="apikey", flusherMetricName=flusherMetricName
)
def testWaitForFlusherAndGarbageCollectWithFlusherNotFound(
self, getAllCustomMetricsMock, timeMock, sleepMock):
time_test_utils.configureTimeAndSleepMocks(timeMock, sleepMock)
flusherMetricName = (
delete_companies._DATA_PATH_FLUSHER_METRIC_PREFIX + "1")
# Patch getAllCustomMetrics to return all negatives and positives
getAllCustomMetricsMock.return_value = [{
"name":
"XIGNITE.FOOBAR.CLOSINGPRICE"
}]
# Execute and validate
with self.assertRaises(delete_companies.FlusherMetricNotFound):
delete_companies._waitForFlusherAndGarbageCollect(
engineServer="host",
engineApiKey="apikey",
flusherMetricName=flusherMetricName)