add unit test flag

MP1/client.py

@@ -40,7 +40,7 @@ def receive_message(sock):
     received_data = receive_exact(sock, message_length)
     return received_data.decode('utf-8')
 
-def grep_all_ips(ips, cmd):
+def grep_all_ips(ips, cmd, if_unit_test):
     match_count = []
 
     for i, ip in enumerate(ips):
@@ -48,7 +48,10 @@ def grep_all_ips(ips, cmd):
             client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
             client_socket.connect((ip, PORT))
 
-            full_cmd = cmd + f" ./machine.{i + 1}.log"
+            if if_unit_test:
+                full_cmd = cmd + f" ./machine.{i + 1}.test.log"
+            else:
+                full_cmd = cmd + f" ./machine.{i + 1}.log"
 
             send_message(client_socket, full_cmd)
             data = receive_message(client_socket)
@@ -69,8 +72,9 @@ def grep_all_ips(ips, cmd):
     return match_count
 
 #main
-cmd = "grep " + " ".join(sys.argv[1:])
-match_count = grep_all_ips(ALL_IPS, cmd)
+#if unit test flag exists, then use the test log files instead of the actual log files
+cmd = "grep " + " ".join([arg for arg in sys.argv[1:] if arg != "-t"])
+match_count = grep_all_ips(ALL_IPS, cmd, if_unit_test=[x for x in sys.argv].count("-t") > 0)
 
 print("=========================== SUMMARY ===========================")
 for i, count in enumerate(match_count):