def test_proxy_check_upgrade_required_fail(self):
"""Verify proxy_check_upgrade_required throws expected exceptions."""
self._set_args({
"firmware": "test_firmware.dlp",
"nvsram": "test_nvsram.dlp"
})
firmware = NetAppESeriesFirmware()
firmware.firmware_version = lambda: b"08.42.50.00"
firmware.nvsram_version = lambda: b"not_nvsram_version"
with self.assertRaisesRegexp(
AnsibleFailJson,
"Failed to retrieve controller firmware information."):
with patch(self.REQUEST_FUNC, return_value=Exception()):
firmware.proxy_check_upgrade_required()
with self.assertRaisesRegexp(
AnsibleFailJson,
"Failed to retrieve storage system's NVSRAM version."):
with patch(self.REQUEST_FUNC,
side_effect=[(200, [{
"versionString": "08.42.50.00"
}]),
Exception()]):
firmware.is_firmware_bundled = lambda: True
firmware.proxy_check_upgrade_required()
with self.assertRaisesRegexp(
AnsibleFailJson,
"Failed to retrieve storage system's NVSRAM version."):
with patch(self.REQUEST_FUNC,
side_effect=[(200, ["08.42.50.00"]),
Exception()]):
firmware.is_firmware_bundled = lambda: False
firmware.proxy_check_upgrade_required()
with self.assertRaisesRegexp(AnsibleFailJson,
"Downgrades are not permitted."):
with patch(self.REQUEST_FUNC,
side_effect=[(200, [{
"versionString": "08.42.50.00"
}]), (200, ["nvsram_version"])]):
firmware.firmware_version = lambda: b"08.40.00.00"
firmware.nvsram_version = lambda: "nvsram_version"
firmware.is_firmware_bundled = lambda: True
firmware.proxy_check_upgrade_required()
with self.assertRaisesRegexp(AnsibleFailJson,
"Downgrades are not permitted."):
with patch(self.REQUEST_FUNC,
side_effect=[(200, ["08.42.50.00"]),
(200, ["nvsram_version"])]):
firmware.is_firmware_bundled = lambda: False
firmware.proxy_check_upgrade_required()
def test_proxy_wait_for_upgrade_pass(self):
"""Verify proxy_wait_for_upgrade."""
with patch(self.SLEEP_FUNC, return_value=None):
self._set_args({
"firmware": "test_firmware.dlp",
"nvsram": "expected_nvsram.dlp"
})
firmware = NetAppESeriesFirmware()
firmware.is_firmware_bundled = lambda: True
with patch(self.REQUEST_FUNC,
side_effect=[(200, {
"status": "not_done"
}), (200, {
"status": "complete"
})]):
firmware.proxy_wait_for_upgrade("1")
firmware.is_firmware_bundled = lambda: False
firmware.firmware_version = lambda: b"08.50.00.00"
firmware.nvsram_version = lambda: b"expected_nvsram"
with patch(self.REQUEST_FUNC,
side_effect=[(200, ["08.40.00.00"]),
(200, ["not_expected_nvsram"]),
(200, ["08.50.00.00"]),
(200, ["expected_nvsram"])]):
firmware.proxy_wait_for_upgrade("1")
def test_firmware_version(self):
"""Verify correct firmware version is returned."""
self._set_args({"firmware": "test.dlp", "nvsram": "test.dlp"})
firmware = NetAppESeriesFirmware()
firmware.is_firmware_bundled = lambda: True
with patch(builtin_path,
mock_open_with_iter(
read_data=self.BUNDLE_HEADER)) as mock_file:
self.assertEqual(firmware.firmware_version(), b"11.40.5")
def test_embedded_wait_for_upgrade_pass(self):
"""Verify controller reboot wait succeeds."""
self._set_args({"firmware": "test.dlp", "nvsram": "test.dlp"})
firmware = NetAppESeriesFirmware()
firmware.firmware_version = lambda: b"11.40.3R2"
firmware.nvsram_version = lambda: b"N280X-842834-D02"
with patch(self.SLEEP_FUNC, return_value=None):
with patch(self.REQUEST_FUNC,
return_value=(200, [{
"fwVersion": "08.42.30.05",
"nvsramVersion": "N280X-842834-D02",
"extendedSAData": {
"codeVersions": [{
"codeModule": "bundleDisplay",
"versionString": "11.40.3R2"
}]
}
}])):
firmware.embedded_wait_for_upgrade()
def test_embedded_upgrade_pass(self):
"""Verify embedded upgrade function."""
with patch(self.CREATE_MULTIPART_FORMDATA_FUNC, return_value=("", {})):
with patch(self.SLEEP_FUNC, return_value=None):
self._set_args({"firmware": "test.dlp", "nvsram": "test.dlp"})
firmware = NetAppESeriesFirmware()
with patch(self.REQUEST_FUNC, return_value=(200, "")):
with patch(
self.BASE_REQUEST_FUNC,
side_effect=[Exception(),
Exception(), (200, "")]):
firmware.embedded_upgrade()
self.assertTrue(firmware.upgrade_in_progress)
self._set_args({
"firmware": "test.dlp",
"nvsram": "test.dlp",
"wait_for_completion": True
})
firmware = NetAppESeriesFirmware()
firmware.firmware_version = lambda: b"11.40.3R2"
firmware.nvsram_version = lambda: b"N280X-842834-D02"
with patch(self.REQUEST_FUNC,
return_value=(200, [{
"fwVersion": "08.42.30.05",
"nvsramVersion": "N280X-842834-D02",
"extendedSAData": {
"codeVersions": [{
"codeModule":
"bundleDisplay",
"versionString":
"11.40.3R2"
}]
}
}])):
firmware.embedded_upgrade()
self.assertFalse(firmware.upgrade_in_progress)
def test_proxy_check_upgrade_required_pass(self):
"""Verify proxy_check_upgrade_required."""
self._set_args({
"firmware": "test_firmware.dlp",
"nvsram": "test_nvsram.dlp"
})
firmware = NetAppESeriesFirmware()
firmware.firmware_version = lambda: b"08.42.50.00"
firmware.nvsram_version = lambda: b"nvsram_version"
with patch(self.REQUEST_FUNC,
side_effect=[(200, [{
"versionString": "08.42.50.00"
}]), (200, ["nvsram_version"])]):
firmware.is_firmware_bundled = lambda: True
firmware.proxy_check_upgrade_required()
self.assertFalse(firmware.upgrade_required)
with patch(self.REQUEST_FUNC,
side_effect=[(200, ["08.42.50.00"]),
(200, ["nvsram_version"])]):
firmware.is_firmware_bundled = lambda: False
firmware.proxy_check_upgrade_required()
self.assertFalse(firmware.upgrade_required)
firmware.firmware_version = lambda: b"08.42.50.00"
firmware.nvsram_version = lambda: b"not_nvsram_version"
with patch(self.REQUEST_FUNC,
side_effect=[(200, [{
"versionString": "08.42.50.00"
}]), (200, ["nvsram_version"])]):
firmware.is_firmware_bundled = lambda: True
firmware.proxy_check_upgrade_required()
self.assertTrue(firmware.upgrade_required)
with patch(self.REQUEST_FUNC,
side_effect=[(200, ["08.42.50.00"]),
(200, ["nvsram_version"])]):
firmware.is_firmware_bundled = lambda: False
firmware.proxy_check_upgrade_required()
self.assertTrue(firmware.upgrade_required)
firmware.firmware_version = lambda: b"08.52.00.00"
firmware.nvsram_version = lambda: b"nvsram_version"
with patch(self.REQUEST_FUNC,
side_effect=[(200, [{
"versionString": "08.42.50.00"
}]), (200, ["nvsram_version"])]):
firmware.is_firmware_bundled = lambda: True
firmware.proxy_check_upgrade_required()
self.assertTrue(firmware.upgrade_required)
with patch(self.REQUEST_FUNC,
side_effect=[(200, ["08.42.50.00"]),
(200, ["nvsram_version"])]):
firmware.is_firmware_bundled = lambda: False
firmware.proxy_check_upgrade_required()
self.assertTrue(firmware.upgrade_required)
firmware.firmware_version = lambda: b"08.52.00.00"
firmware.nvsram_version = lambda: b"not_nvsram_version"
with patch(self.REQUEST_FUNC,
side_effect=[(200, [{
"versionString": "08.42.50.00"
}]), (200, ["nvsram_version"])]):
firmware.is_firmware_bundled = lambda: True
firmware.proxy_check_upgrade_required()
self.assertTrue(firmware.upgrade_required)
with patch(self.REQUEST_FUNC,
side_effect=[(200, ["08.42.50.00"]),
(200, ["nvsram_version"])]):
firmware.is_firmware_bundled = lambda: False
firmware.proxy_check_upgrade_required()
self.assertTrue(firmware.upgrade_required)