def test_wrongserial(mock_getmodel):
"""Test a failed called based on an inaccurate MacBook serial number.
:return: True
"""
mock_getmodel.return_value = get_failedcall()
serial = "C02Z20D2LVDS"
with pytest.raises(AttributeError, match="Serial number lookup error."):
get_model(serial)
def test_getmodel(mock_getmodel):
"""Test a successful called based on a live accurate MacBook serial number.
:return: True
"""
mock_getmodel.return_value = get_successfulcall()
serial = "C02Z20D2LVDT"
assert (get_model(serial)) == "MacBook Pro (15-inch, 2019)"