def METHOD_NAME(make_stubber, error_code):
    lambda_client = boto3.client('lambda')
    lambda_stubber = make_stubber(lambda_client)
    wrapper = LambdaWrapper(lambda_client, None)
    func_name = 'test-func_name'
    package = 'test-package'
    update_status = 'InProgress'
    lambda_stubber.stub_update_function_code(
        func_name, update_status, package=package, error_code=error_code)
    if error_code is None:
        got_response = wrapper.update_function_code(func_name, package)
        assert got_response['LastUpdateStatus'] == update_status
    else:
        with pytest.raises(ClientError) as exc_info:
            wrapper.update_function_code(func_name, package)
        assert exc_info.value.response['Error']['Code'] == error_code