Refactor tests

Author: AttilaGombosER

tests/receiverIntegrationTest.py

@@ -22,13 +22,12 @@ def setUp(self):
 
     def test_receives_message(self):
         # Given
-        group = GROUP.hello()
         context = Context()
-        radio = context.socket(RADIO)
-        radio.connect(group.url)
         messages = []
 
-        with DishReceiver(context) as receiver:
+        with DishReceiver(context) as receiver, context.socket(RADIO) as radio:
+            group = GROUP.hello()
+            radio.connect(group.url)
             receiver.register(lambda message: messages.append(message))
             receiver.start(group)
 

tests/senderIntegrationTest.py

@@ -24,14 +24,13 @@ def setUp(self):
 
     def test_sends_message(self):
         # Given
-        group = GROUP.hello()
         context = Context()
-        dish = context.socket(DISH)
-        dish.bind(group.url)
-        dish.join(group.name)
         messages = []
 
-        with RadioSender(context) as sender:
+        with RadioSender(context) as sender, context.socket(DISH) as dish:
+            group = GROUP.hello()
+            dish.bind(group.url)
+            dish.join(group.name)
             sender.start(group)
             Thread(target=lambda: messages.append(dish.recv_json())).start()