From dc5c7732d7d7f0cdb637f23fe44068e620fab06b Mon Sep 17 00:00:00 2001
From: Joachim Wiberg <troglobit@gmail.com>
Date: Fri, 10 Jan 2025 11:10:02 +0100
Subject: [PATCH] test: infamy: return None if port doesn't exist

Follow-up to e4535aa, allowing tests to be even simpler.

Signed-off-by: Joachim Wiberg <troglobit@gmail.com>
---
 test/infamy/transport.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/test/infamy/transport.py b/test/infamy/transport.py
index 922a9de2..2148073b 100644
--- a/test/infamy/transport.py
+++ b/test/infamy/transport.py
@@ -56,7 +56,9 @@ class Transport(ABC):
         pass
 
     def __getitem__(self, key):
-        return self.mapping[key]
+        if key in self.mapping:
+            return self.mapping[key]
+        return None
 
     def get_iface(self, name):
         """Fetch target dict for iface and extract param from JSON"""