diff options
| author | Christian Hopps <chopps@labn.net> | 2021-09-03 08:47:30 -0400 |
|---|---|---|
| committer | Christian Hopps <chopps@labn.net> | 2021-09-04 09:04:47 -0400 |
| commit | a53c08bc131c02f4a20931d7aa9f974194ab16e7 (patch) | |
| tree | 2c48ef7c757706d47ac5b999bf667166bc1c2e1a /tests/topotests/lib/mcast-tester.py | |
| parent | 4953ca977f3a5de8109ee6353ad07f816ca1774c (diff) | |
tests: cleanup: rerun changed files through black
Signed-off-by: Christian Hopps <chopps@labn.net>
Diffstat (limited to 'tests/topotests/lib/mcast-tester.py')
| -rwxr-xr-x | tests/topotests/lib/mcast-tester.py | 36 |
1 files changed, 18 insertions, 18 deletions
diff --git a/tests/topotests/lib/mcast-tester.py b/tests/topotests/lib/mcast-tester.py index a594c4c88d..30beccb787 100755 --- a/tests/topotests/lib/mcast-tester.py +++ b/tests/topotests/lib/mcast-tester.py @@ -35,12 +35,11 @@ import time # def interface_name_to_index(name): "Gets the interface index using its name. Returns None on failure." - interfaces = json.loads( - subprocess.check_output('ip -j link show', shell=True)) + interfaces = json.loads(subprocess.check_output("ip -j link show", shell=True)) for interface in interfaces: - if interface['ifname'] == name: - return interface['ifindex'] + if interface["ifname"] == name: + return interface["ifindex"] return None @@ -60,13 +59,12 @@ def multicast_join(sock, ifindex, group, port): # Main code. # parser = argparse.ArgumentParser(description="Multicast RX utility") -parser.add_argument('group', help='Multicast IP') -parser.add_argument('interface', help='Interface name') -parser.add_argument('--socket', help='Point to topotest UNIX socket') +parser.add_argument("group", help="Multicast IP") +parser.add_argument("interface", help="Interface name") +parser.add_argument("--socket", help="Point to topotest UNIX socket") parser.add_argument( - '--send', - help='Transmit instead of join with interval', - type=float, default=0) + "--send", help="Transmit instead of join with interval", type=float, default=0 +) args = parser.parse_args() ttl = 16 @@ -75,7 +73,7 @@ port = 1000 # Get interface index/validate. ifindex = interface_name_to_index(args.interface) if ifindex is None: - sys.stderr.write('Interface {} does not exists\n'.format(args.interface)) + sys.stderr.write("Interface {} does not exists\n".format(args.interface)) sys.exit(1) # We need root privileges to set up multicast. @@ -102,17 +100,18 @@ msock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) if args.send > 0: # Prepare multicast bit in that interface. msock.setsockopt( - socket.SOL_SOCKET, 25, - struct.pack("%ds" % len(args.interface), - args.interface.encode('utf-8'))) + socket.SOL_SOCKET, + 25, + struct.pack("%ds" % len(args.interface), args.interface.encode("utf-8")), + ) # Set packets TTL. - msock.setsockopt( - socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, struct.pack("b", ttl)) + msock.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, struct.pack("b", ttl)) # Block to ensure packet send. msock.setblocking(True) else: multicast_join(msock, ifindex, args.group, port) + def should_exit(): if not toposock: # If we are sending then we have slept @@ -122,12 +121,13 @@ def should_exit(): else: try: data = toposock.recv(1) - if data == b'': - print(' -> Connection closed') + if data == b"": + print(" -> Connection closed") return True except BlockingIOError: return False + counter = 0 while not should_exit(): if args.send > 0: |
