mirror of
https://github.com/leanprover/lean4.git
synced 2026-03-17 18:34:06 +00:00
08eb78a5b2
This PR sets up the new integrated test/bench suite. It then migrates all benchmarks and some related tests to the new suite. There's also some documentation and some linting. For now, a lot of the old tests are left alone so this PR doesn't become even larger than it already is. Eventually, all tests should be migrated to the new suite though so there isn't a confusing mix of two systems.
64 lines
1.5 KiB
Lean4
64 lines
1.5 KiB
Lean4
import Std.Internal.Async
|
||
import Std.Internal.UV
|
||
import Std.Net.Addr
|
||
|
||
open Std.Internal.IO Async
|
||
open Std.Net
|
||
|
||
def assertBEq [BEq α] [ToString α] (actual expected : α) : IO Unit := do
|
||
unless actual == expected do
|
||
throw <| IO.userError <|
|
||
s!"expected '{expected}', got '{actual}'"
|
||
|
||
/-- Mike is another client. -/
|
||
def runMike (client: TCP.Socket.Client) : Async Unit := do
|
||
let message ← client.recv? 1024
|
||
assertBEq (String.fromUTF8? =<< message) none
|
||
|
||
/-- Joe is another client. -/
|
||
def runJoe (client: TCP.Socket.Client) : Async Unit := do
|
||
let message ← client.recv? 1024
|
||
assertBEq (String.fromUTF8? =<< message) none
|
||
|
||
/-- Robert is the server. -/
|
||
def runRobert (server: TCP.Socket.Server) : Async Unit := do
|
||
discard <| server.accept
|
||
discard <| server.accept
|
||
|
||
def clientServer : IO Unit := do
|
||
let addr := SocketAddressV4.mk (.ofParts 127 0 0 1) 8083
|
||
|
||
let server ← TCP.Socket.Server.mk
|
||
server.bind addr
|
||
server.listen 128
|
||
|
||
assertBEq (← server.getSockName).port 8083
|
||
|
||
let joe ← TCP.Socket.Client.mk
|
||
let task ← joe.connect addr |>.toBaseIO
|
||
task.block
|
||
|
||
assertBEq (← joe.getPeerName).port 8083
|
||
|
||
joe.noDelay
|
||
|
||
let mike ← TCP.Socket.Client.mk
|
||
let task ← mike.connect addr |>.toBaseIO
|
||
task.block
|
||
|
||
assertBEq (← mike.getPeerName).port 8083
|
||
|
||
mike.noDelay
|
||
|
||
let serverTask ← (runRobert server).toIO
|
||
|
||
let joeTask ← (runJoe joe).toIO
|
||
let mikeTask ← (runMike mike).toIO
|
||
|
||
serverTask.block
|
||
|
||
joeTask.block
|
||
mikeTask.block
|
||
|
||
#eval clientServer
|