status-mobile/test/appium/support/message_reliability_report.py

62 lines
2.6 KiB
Python

def write_data_to_file(sent: int, received: int):
with open('messages.txt', 'a') as f:
f.write("['sent': %s, 'received': %s]" % (sent, received))
def create_one_to_one_chat_report(one_to_one_chat_data: dict):
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
user_a = one_to_one_chat_data['user_a']
user_b = one_to_one_chat_data['user_b']
fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(15, 7))
time_1 = sorted(user_a['message_time'])
ax.plot([i / 60 for i in time_1], [user_a['message_time'][i] for i in time_1],
'o-', color='#0c0fea', label='user_a')
time_2 = sorted(user_b['message_time'])
ax.plot([i / 60 for i in time_2], [user_b['message_time'][i] for i in time_2],
'o-', color='#f61e06', label='user_b')
sent_messages = user_a['sent_messages'] + user_b['sent_messages']
received_messages = len(user_a['message_time']) + len(user_b['message_time'])
title = "User A: sent messages: {}, received messages: {}" \
"\nUser B: sent messages: {}, received messages: {}".format(user_a['sent_messages'],
len(user_a['message_time']),
user_b['sent_messages'],
len(user_b['message_time']))
if sent_messages:
title += "\nReceived messages: {}%".format(
round((len(user_a['message_time']) + len(user_b['message_time'])) / sent_messages * 100, ndigits=2))
plt.title(title)
plt.xlabel('chat session duration, minutes')
plt.ylabel('time to receive a message, seconds')
plt.legend()
fig.savefig('chart.png')
write_data_to_file(sent=sent_messages, received=received_messages)
def create_public_chat_report(public_chat_data: dict):
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
sent_messages = public_chat_data['sent_messages']
message_time = public_chat_data['message_time']
fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(15, 7))
sorted_time = sorted(message_time)
ax.plot([i / 60 for i in sorted_time], [message_time[i] for i in sorted_time], 'o-', color='#0c0fea')
title = "Sent messages: {}\nReceived messages: {}".format(sent_messages, len(message_time))
plt.title(title)
plt.xlabel('chat session duration, minutes')
plt.ylabel('time to receive a message, seconds')
plt.legend()
fig.savefig('chart.png')
write_data_to_file(sent=sent_messages, received=len(message_time))