Denying WebSocket connection in WebSocketEndpoint.on_connect leads to Exception

[dingensundso]

I created a WebSocketEndpoint class in which I want to deny the connection in certain conditions.
When the WebSocket is closed in on_connect I receive the following exception:

  File "starlette/endpoints.py", line 83, in dispatch
    close_code = int(message.get("code", status.WS_1000_NORMAL_CLOSURE))
TypeError: int() argument must be a string, a bytes-like object or a real number, not 'NoneType'

So code is None when the server denies the connection instead of non-existant.

starlette/starlette/endpoints.py

Lines 72 to 87 in e086fc2

	await self.on_connect(websocket)
	close_code = status.WS_1000_NORMAL_CLOSURE
	try:
	while True:
	message = await websocket.receive()
	if message["type"] == "websocket.receive":
	data = await self.decode(websocket, message)
	await self.on_receive(websocket, data)
	elif message["type"] == "websocket.disconnect":
	close_code = int(message.get("code", status.WS_1000_NORMAL_CLOSURE))
	break
	except Exception as exc:
	close_code = status.WS_1011_INTERNAL_ERROR
	raise exc

Changing line 83 to the following should fix the issue:

close_code = int(message.get("code") or status.WS_1000_NORMAL_CLOSURE)               

[Kludex]

Created #1560 from this discussion. Thanks for bringing it up. :)