Add timeout test

Author: Lynskylate

tests/snippets/stdlib_socket.py

@@ -138,23 +138,12 @@
 with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
 	pass
 
-connector = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-connection.settimeout(None)
-assert connection.getblocking() == True
-assert connection.gettimeout() == None
-
-connection.settimeout(0)
-assert connection.getblocking() == False
-assert connection.gettimeout() == 0
-
-connection.setblocking(True)
-assert connection.getblocking() == True
-assert connection.gettimeout() == None
-
-connection.setblocking(False)
-assert connection.getblocking() == False
-assert connection.gettimeout() == 0.0
-
-connection.settimeout(3)
-assert connection.gettimeout() == 3
-assert connection.getblocking() == True
+with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as listener:
+	listener.bind(("127.0.0.1", 0))
+	listener.listen(1)
+	connector = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+	connector.connect(("127.0.0.1", listener.getsockname()[1]))
+	(connection, addr) = listener.accept()
+	connection.settimeout(1.0)
+	with assertRaises(OSError):
+		connection.recv(len(MESSAGE_A))

vm/src/obj/objfloat.rs

@@ -46,15 +46,6 @@ impl IntoPyObject for f32 {
     }
 }
 
-impl TryFromObject for f64 {
-    fn try_from_object(vm: &VirtualMachine, obj: PyObjectRef) -> PyResult<Self> {
-        try_float(&obj, vm)?.map_or_else(
-            || Err(vm.new_type_error(format!("Expect float object, but get {}", obj.class().name))),
-            |val| Ok(val),
-        )
-    }
-}
-
 impl From<f64> for PyFloat {
     fn from(value: f64) -> Self {
         PyFloat { value }

vm/src/stdlib/socket.rs

@@ -2,7 +2,7 @@ use std::cell::RefCell;
 use std::io;
 use std::io::Read;
 use std::io::Write;
-use std::net::{Ipv4Addr, SocketAddr, TcpListener, TcpStream, ToSocketAddrs, UdpSocket };
+use std::net::{Ipv4Addr, SocketAddr, TcpListener, TcpStream, ToSocketAddrs, UdpSocket};
 use std::time::Duration;
 
 #[cfg(all(unix, not(target_os = "redox")))]
@@ -25,7 +25,6 @@ use crate::obj::objtype::PyClassRef;
 use crate::stdlib::os::convert_nix_error;
 use num_bigint::Sign;
 use num_traits::ToPrimitive;
-use itertools::Itertools;
 
 #[derive(Debug, Copy, Clone)]
 enum AddressFamily {
@@ -225,8 +224,11 @@ impl SocketRef {
                         Ok(mut sock_addrs) => {
                             if sock_addrs.len() == 0 {
                                 let error_type = vm.class("socket", "gaierror");
-                                return Err(vm.new_exception(error_type, "nodename nor servname provided, or not known".to_string()))
-                            }else{
+                                return Err(vm.new_exception(
+                                    error_type,
+                                    "nodename nor servname provided, or not known".to_string(),
+                                ));
+                            } else {
                                 sock_addrs.next().unwrap()
                             }
                         }
@@ -466,9 +468,10 @@ impl SocketRef {
                 let block = timeout > 0.0;
 
                 if let Some(conn) = self.con.borrow_mut().as_mut() {
-                    conn.setblocking(block).and_then(|_| {
-                        conn.settimeout(Duration::from_secs(timeout as u64))
-                    }).map_err(|err| vm.new_os_error(err.to_string())).map(|_| ())
+                    conn.setblocking(block)
+                        .and_then(|_| conn.settimeout(Duration::from_secs(timeout as u64)))
+                        .map_err(|err| vm.new_os_error(err.to_string()))
+                        .map(|_| ())
                 } else {
                     Ok(())
                 }