fake_trx/udp_link.py: close socket in destructor

Previously it was required to call the UDPLink.shutdown() method
manually in order to close a socket. Let's do this automatically
using the destructor of UDPLink.

Change-Id: I59c3dc61ec58cd9effeb789947d28fd602ca91f4

1 files changed, 3 insertions, 3 deletions

diff --git a/src/target/fake_trx/udp_link.py b/src/target/fake_trx/udp_link.py
index 0afd1504..c0c41ff6 100644
--- a/src/target/fake_trx/udp_link.py
+++ b/src/target/fake_trx/udp_link.py
@@ -35,6 +35,9 @@ class UDPLink:
 		self.remote_addr = remote_addr
 		self.remote_port = remote_port
 
+	def __del__(self):
+		self.sock.close()
+
 	def loop(self):
 		r_event, w_event, x_event = select.select([self.sock], [], [])
 
@@ -43,9 +46,6 @@ class UDPLink:
 			data, addr = self.sock.recvfrom(128)
 			self.handle_rx(data.decode())
 
-	def shutdown(self):
-		self.sock.close();
-
 	def send(self, data):
 		if type(data) not in [bytearray, bytes]:
 			data = data.encode()