def test_capture_call__clearLocalTemporary_when_exception(self):
capture = Capture()
capture._executeCaptureScript = MagicMock()
capture._executeCaptureScript.side_effect = Exception("")
capture._clearLocalTemporary = MagicMock()
random_uuid = 'random_uuid'
uuid.uuid4 = MagicMock(return_value=random_uuid)
try:
capture.capture("url", "language", 0, 1, 2)
except Exception as e:
print(e)
capture._clearLocalTemporary.assert_called_with(random_uuid)
def test_capture(self):
url = "url"
language = "language"
numberToCapture = "numberToCapture"
startTimeStamp = "startTimeStamp"
endTimeStamp = "endTimeStamp"
convertedCaptureItems = []
random_uuid = 'random_uuid'
uuid.uuid4 = MagicMock(return_value=random_uuid)
capture = Capture()
capture._executeCaptureScript = MagicMock(return_value={})
capture._convertToCaptureItems = MagicMock(
return_value=convertedCaptureItems)
capture._store = MagicMock(return_value={})
capture._clearLocalTemporary = MagicMock()
capture.capture(url, language, numberToCapture, startTimeStamp,
endTimeStamp)
capture._executeCaptureScript.assert_called_with(
url, language, numberToCapture, startTimeStamp, endTimeStamp,
random_uuid)
capture._store.assert_called_with(convertedCaptureItems)
capture._clearLocalTemporary.assert_called_with(random_uuid)