def test_get_group_data_Returns_correct_value_if_field_and_id_are_valid(
self, mock_read_groups_file):
crud = CRUD()
mock_read_groups_file.return_value = self.groups_data
self.assertEqual(crud.get_group_data('1', 'name'), 'default')
""""""
pass
def test_get_group_data_Returns_false_for_invalid_field(
self, mock_read_groups_file):
crud = CRUD()
mock_read_groups_file.return_value = self.groups_data
self.assertFalse(crud.get_group_data('1', 'INVALID'))
""""""
pass
def test_get_group_data_Returns_false_for_invalid_id(
self, mock_read_groups_file):
""""""
mock_read_groups_file.return_value = self.groups_data
crud = CRUD()
self.assertFalse(crud.get_group_data('3', "name"))
pass
class RENEGE:
"""Class pour realiser le filtrage du spam en utilisant vocabular.json file et
CRUD et EmalAnalyze classes"""
def __init__(self):
self.email_file = "800-mails.json"
self.crud = CRUD()
self.e_mail = EmailAnalyzer()
def calculate_user_trust(self, user_id):
#extracting json data
date_of_first_seen_message = self.crud.get_user_data(user_id, "Date_of_first_seen_message")
date_of_last_seen_message = self.crud.get_user_data(user_id, "Date_of_last_seen_message")
n_ham = self.crud.get_user_data(user_id, "HamN")
n_spam = self.crud.get_user_data(user_id, "SpamN")
groups = self.crud.get_user_data(user_id, "Groups")
#calculate the sum of trust values of all groups
sum_trust = 0
for group in groups :
group_id = self.crud.get_group_id(group)
sum_trust += self.crud.get_group_data(group_id, 'Trust')
#now that we have all the needed vars, calculate trust1, trust2 and trust
trust1 = (date_of_last_seen_message * n_ham) / (date_of_first_seen_message * (n_ham + n_spam))
trust2 = sum_trust / len(groups)
trust = (trust1 + trust2) / 2
if trust2 < 50:
trust = trust2
if trust1 > 100:
trust = 100
#before returning a value, check if trust is between 0 and 100
if trust < 0:
trust = 0
elif trust > 100:
trust = 100
return trust
def classify_emails(self):
'''
fonction deja implemente
Description: fonction pour commencer l'analyse des e-mails.
Sortie: bool, 'True' pour succes, 'False' dans le cas de failure.
'''
try:
self.process_email(self.get_email())
return True
except Exception as e:
raise Exception
return False
def process_email(self, new_emails):
'''
Description: fonction pour analyser chaque nouvel e-mail dans le
dictionare. Elle gere l'ajout des nouveux utilisateurs et/ou modification
de l'information existante sur les utilisateurs et groupes.
Sortie: bool, 'True' pour succes, 'False' dans le cas de failure.
'''
print(type(new_emails))
emails = new_emails["dataset"]
for email in emails:
email_adr = email['mail']['From']
date = email['mail']['Date']
spam = email['mail']['Spam']
spam = spam == 'true' and True or False
if self.crud.get_user_id(email_adr):
self.update_user_info(email_adr, date, spam)
else:
self.add_user_info(email_adr, date)
return True
def update_user_info(self, new_user_email, new_user_date, new_email_spam):
'''
Description: fonction pour modifier l'information de utilisateur (date de dernier message arrive,
numero de spam/ham, trust level, etc).
Sortie: bool, 'True' pour succes, 'False' dans le cas de failure.
'''
users_dict = self.crud.read_users_file()
for key in users_dict:
user = users_dict.get(key)
if user.get("name") == new_user_email:
self.crud.update_users(key, 'Date_of_first_seen_message', new_user_date)
if new_email_spam:
spamN = user.get('SpamN')
spamN += 1
trust = user.get('Trust')
if trust > 0:
trust -= 1
self.crud.update_users(key, 'SpamN', spamN)
self.crud.update_users(key, 'Trust', trust)
else:
hamN = user.get('HamN')
hamN += 1
trust = user.get('Trust')
if trust < 100:
trust += 1
self.crud.update_users(key, 'HamN', hamN)
self.crud.update_users(key, 'HamN', trust)
break
return True
def add_user_info(self, user_email, user_date):
self.crud.add_new_user(user_email, user_date)
def update_group_info(self, user_group_list):
'''
Description: fonction pour modifier l'information de groupe dans lequel
l'utilisater est present (trust level, etc).
Sortie: bool, 'True' pour succes, 'False' dans le cas de failure.
'''
groups = self.crud.read_groups_file()
for group in groups:
if group['List_of_members'] == user_group_list:
group['Trust'] = self.crud.get_user_data(user_group_list[1], 'Trust')
return True
return False
def get_user_email_list(self):
'''
Description: fonction pour creer le liste des e-mails (noms)
des utilisateurs uniques.
Sortie: liste des uniques e-mails des utilisateurs
'''
emails = self.get_email()
list_of_email_addresses = []
for mail in emails['dataset']:
list_of_email_addresses.append((mail['mail']['From']))
return list_of_email_addresses
def get_email(self):
'''
Description: fonction pour lire le ficher json avec les mails et extraire les
donees necessaire.
Sortie: dictionare de e-mails formate selon le JSON.
'''
with open(self.email_file) as file:
emails = json.load(file)
return emails
class RENEGE:
"""Class pour realiser le filtrage du spam en utilisant vocabular.json file et
CRUD et EmalAnalyze classes"""
def __init__(self):
self.email_file = "800-mails.json"
self.crud = CRUD()
self.e_mail = EmailAnalyzer()
def calculate_user_trust(self, user_id):
#extracting json data
date_of_first_seen_message = self.crud.get_user_data(user_id, "Date_of_first_seen_message")
date_of_last_seen_message = self.crud.get_user_data(user_id, "Date_of_last_seen_message")
n_ham = self.crud.get_user_data(user_id, "HamN")
n_spam = self.crud.get_user_data(user_id, "SpamN")
groups = self.crud.get_user_data(user_id, "Groups")
#calculate the sum of trust values of all groups
sum_trust = 0
for group in groups :
group_id = self.crud.get_group_id(group)
sum_trust += self.crud.get_group_data(group_id, 'Trust')
#now that we have all the needed vars, calculate trust1, trust2 and trust
trust1 = (date_of_last_seen_message * n_ham) / (date_of_first_seen_message * (n_ham + n_spam))
trust2 = sum_trust / len(groups)
trust = (trust1 + trust2) / 2
if trust2 < 50:
trust = trust2
if trust1 > 100:
trust = 100
#before returning a value, check if trust is between 0 and 100
if trust < 0:
trust = 0
elif trust > 100:
trust = 100
return trust
def classify_emails(self, calculation_mode, is_log_est, is_log_combo):
'''
fonction deja implemente
Description: fonction pour commencer l'analyse des e-mails.
Sortie: bool, 'True' pour succes, 'False' dans le cas de failure.
'''
try:
self.process_email(self.get_email(), calculation_mode, is_log_est, is_log_combo)
return True
except Exception:
raise Exception
def process_email(self, new_emails, calculation_mode, is_log_est, is_log_combo):
'''
Description: fonction pour analyser chaque nouvel e-mail dans le
dictionare. Elle gere l'ajout des nouveux utilisateurs et/ou modification
de l'information existante sur les utilisateurs et groupes.
Sortie: bool, 'True' pour succes, 'False' dans le cas de failure.
'''
print(type(new_emails))
emails = new_emails["dataset"]
for email in emails:
email_adr = email['mail']['From']
date = email['mail']['Date']
spam = email['mail']['Spam'] == 'true'
subject = email['mail']['Subject']
body = email['mail']['Body']
user_id = self.crud.get_user_id(email_adr)
if user_id:
self.update_user_info(email_adr, date, spam)
else:
self.add_user_info(email_adr, date)
# trust de l'utilisateur
trust = self.crud.get_user_data(user_id, "Trust")
# moyenne du trust de tous les groupes
user_group = self.crud.get_user_data(user_id, "Groups")
sum_trust = 0
groups = self.crud.read_groups_file()
for group in groups :
if group['List_of_members'] == user_group:
sum_trust += group['Trust']
avg_group_trust = sum_trust / len(groups)
# difference de jours entre last_seen_msg et first_seen_msg
user_activity = self.substract_dates(self.crud.get_user_data(email_adr, "Date_of_last_seen_message"),
self.crud.get_user_data(email_adr, "Date_of_first_seen_message"))
if(calculation_mode == 1):
spam = self.e_mail.is_spam_function_one(spam, user_activity, trust, avg_group_trust)
elif(calculation_mode == 2):
spam = self.e_mail.is_spam_function_two(spam, trust, avg_group_trust)
elif(calculation_mode == 0):
spam = self.e_mail.is_spam(subject, body, is_log_est, is_log_combo, 0)
# mettre a jour l'utilisateur avec la nouvelle valeur de spam
self.update_user_info(email_adr, date, spam)
return True
def substract_dates(self, last_seen_msg, first_seen_msg):
# calcul de difference de jours entre le dernier et le premier message
last_seen_msg = str(last_seen_msg).split("-")
first_seen_msg = str(first_seen_msg).split("-")
user_activity = date(int(last_seen_msg[0]), int(last_seen_msg[1]), int(last_seen_msg[2]))
- date(int(first_seen_msg[0]), int(first_seen_msg[1]), int(first_seen_msg[2]))
# convertir la soustraction en int
user_activity = int((str(user_activity).split(" "))[0])
return user_activity
class TestCRUD(unittest.TestCase):
def setUp(self):
self.crud = CRUD()
# c'est un exemple de données "mock" à utiliser comme "return value" de read_users_file
self.users_data = {
"1": {
"name": "*****@*****.**",
"Trust": 100,
"SpamN": 0,
"HamN": 20,
"Date_of_first_seen_message": 1596848266.0,
"Date_of_last_seen_message": 1596848266.0,
"Groups": ["default"],
},
"2": {
"name": "*****@*****.**",
"Trust": 65.45454,
"SpamN": 171,
"HamN": 324,
"Date_of_first_seen_message": 1596855166.0,
"Date_of_last_seen_message": 1596855166.0,
"Groups": ["default"],
},
}
# c'est un exemple de données "mock" à utiliser comme "return value" de read_groups_file
self.groups_data = {
"1": {
"name": "default",
"Trust": 50,
"List_of_members": ["*****@*****.**", "*****@*****.**"],
},
"2": {
"name": "friends",
"Trust": 90,
"List_of_members": ["*****@*****.**"],
},
}
def tearDown(self):
pass
@patch("crud.CRUD.read_users_file")
def test_add_new_user_returns_false_if_username_and_date_are_not_given(
self, mock_read_users_file
):
"""Description: il faut utiliser les mocks des fonctions "read_users_file",
pour vérifier que le nom d'utilisateur est donné à la fonction
"""
mock_read_users_file.return_value = self.users_data
add_new_user_return_value_two_none = self.crud.add_new_user(None, None)
self.assertFalse(add_new_user_return_value_two_none)
@patch("crud.CRUD.read_users_file")
def test_add_new_user_returns_false_if_only_date_is_given(
self, mock_read_users_file
):
"""Description: il faut utiliser les mocks des fonctions "read_users_file",
pour vérifier que le nom d'utilisateur est donné à la fonction
"""
mock_read_users_file.return_value = self.users_data
add_new_user_return_value_one_none = self.crud.add_new_user(None, '2020-01-01')
self.assertFalse(add_new_user_return_value_one_none)
@patch("crud.CRUD.read_users_file")
def test_add_new_user_returns_false_if_only_name_is_given(
self, mock_read_users_file
):
"""Description: il faut utiliser les mocks des fonctions "read_users_file",
pour vérifier que le nom d'utilisateur est donné à la fonction
"""
mock_read_users_file.return_value = self.users_data
add_new_user_return_value_one_none = self.crud.add_new_user('*****@*****.**', None)
self.assertFalse(add_new_user_return_value_one_none)
@patch("crud.CRUD.read_users_file")
def test_add_new_user_returns_true_if_name_and_date_are_given(
self, mock_read_users_file
):
"""Description: il faut utiliser les mocks des fonctions "read_users_file",
pour vérifier que le nom d'utilisateur est donné à la fonction
"""
mock_read_users_file.return_value = self.users_data
add_new_user_return_value_correct_vals = self.crud.add_new_user('*****@*****.**', '2020-01-01')
self.assertTrue(add_new_user_return_value_correct_vals)
@patch("crud.CRUD.read_groups_file")
def test_add_new_group_returns_false_if_name_is_too_short(
self, mock_read_groups_file
):
"""Description: il faut utiliser les mocks des fonctions "read_groups_file",pour tester que
l'information a ajouter pour le groupe a étée formée correctement
"""
mock_read_groups_file.return_value = self.groups_data
add_new_group_no_name = self.crud.add_new_group("", 50, ['dummyUser1', 'dummyUser2'])
self.assertFalse(add_new_group_no_name)
@patch("crud.CRUD.read_groups_file")
def test_add_new_group_returns_false_if_name_is_too_long(
self, mock_read_groups_file
):
"""Description: il faut utiliser les mocks des fonctions "read_groups_file",pour tester que
l'information a ajouter pour le groupe a étée formée correctement
"""
dummy_long_name = ("a" * 65)
mock_read_groups_file.return_value = self.groups_data
add_new_group_long_name = self.crud.add_new_group(dummy_long_name, 50, ['dummyUser1', 'dummyUser2'])
self.assertFalse(add_new_group_long_name)
@patch("crud.CRUD.read_groups_file")
def test_add_new_group_returns_false_if_trust_is_more_than_100(
self, mock_read_groups_file
):
"""Description: il faut utiliser les mocks des fonctions "read_groups_file",pour tester que
l'information a ajouter pour le groupe a étée formée correctement
"""
mock_read_groups_file.return_value = self.groups_data
add_new_group_tooBigTrust = self.crud.add_new_group("dummyName", 101, ['dummyUser1', 'dummyUser2'])
self.assertFalse(add_new_group_tooBigTrust)
@patch("crud.CRUD.read_groups_file")
def test_add_new_group_returns_false_if_trust_is_less_than_0(
self, mock_read_groups_file
):
"""Description: il faut utiliser les mocks des fonctions "read_groups_file",pour tester que
l'information a ajouter pour le groupe a étée formée correctement
"""
mock_read_groups_file.return_value = self.groups_data
add_new_group_too_small_trust = self.crud.add_new_group("dummyName", -1, ['dummyUser1', 'dummyUser2'])
self.assertFalse(add_new_group_too_small_trust)
@patch("crud.CRUD.read_groups_file")
def test_add_new_group_returns_false_if_member_list_is_not_a_list(
self, mock_read_groups_file
):
"""Description: il faut utiliser les mocks des fonctions "read_groups_file",pour tester que
l'information a ajouter pour le groupe a étée formée correctement
"""
mock_read_groups_file.return_value = self.groups_data
add_new_group_not_good_member_list = self.crud.add_new_group("dummyName", 50, 'not a list')
self.assertFalse(add_new_group_not_good_member_list)
@patch("crud.CRUD.read_groups_file")
def test_add_new_group_returns_true_if_values_are_correct(
self, mock_read_groups_file
):
"""Description: il faut utiliser les mocks des fonctions "read_groups_file",pour tester que
l'information a ajouter pour le groupe a étée formée correctement
"""
mock_read_groups_file.return_value = self.groups_data
add_new_group_correct_vals = self.crud.add_new_group("dummyName", 50, ['dummyUser1', 'dummyUser2'])
self.assertTrue(add_new_group_correct_vals)
@patch("crud.CRUD.read_users_file")
def test_get_user_data_Returns_false_for_invalid_id(self, mock_read_users_file):
"""Description: il faut utiliser le mock de fonction "read_groups_file",
(ou selon votre realisation) pour tester que false (ou bien une excepton)
est returnee par la fonction si ID non-existant est utilisée
il faut utiliser ".assertEqual()" ou ".assertFalse()"
"""
dummyBadId = 10
mock_read_users_file.return_value = self.users_data
self.crud.number_of_users = 2
get_user_data_return_value = self.crud.get_user_data(dummyBadId, 'name')
self.assertIsNone(get_user_data_return_value)
@patch("crud.CRUD.read_users_file")
def test_get_user_data_Returns_false_for_invalid_field(self, mock_read_users_file):
"""Description: il faut utiliser le mock de fonction "read_groups_file",
(ou selon votre realisation) pour tester que false (ou bien une excepton)
est returnee par la fonction si champ non-existant est utilisée
il faut utiliser ".assertEqual()" ou ".assertFalse()"
"""
mock_read_users_file.return_value = self.users_data
self.crud.number_of_users = 2
get_user_data_return_value = self.crud.get_user_data(1, 'badField')
self.assertIsNone(get_user_data_return_value)
@patch("crud.CRUD.read_users_file")
def test_get_user_data_Returns_correct_value_if_field_and_id_are_valid(
self, mock_read_users_file
):
"""Description: il faut utiliser le mock de fonction "read_groups_file",
(ou selon votre realisation) pour tester que une bonne valeur est fournie
si champ est id valide sont utilisee
il faut utiliser ".assertEqual()"
"""
mock_read_users_file.return_value = self.users_data
self.crud.number_of_users = 2
get_user_data_return_value = self.crud.get_user_data(1, 'name')
self.assertEqual(get_user_data_return_value, "*****@*****.**")
@patch("crud.CRUD.read_groups_file")
def test_get_group_data_Returns_false_for_invalid_id(self, mock_read_groups_file):
""""""
dummyBadId = 10
mock_read_groups_file.return_value = self.groups_data
self.crud.number_of_groups = 2
get_group_return_value = self.crud.get_group_data(dummyBadId, 'name')
self.assertIsNone(get_group_return_value)
@patch("crud.CRUD.read_groups_file")
def test_get_group_data_Returns_false_for_invalid_field(
self, mock_read_groups_file
):
""""""
mock_read_groups_file.return_value = self.groups_data
self.crud.number_of_groups = 2
get_group_data_return_value = self.crud.get_group_data(1, 'badField')
self.assertIsNone(get_group_data_return_value)
@patch("crud.CRUD.read_groups_file")
def test_get_group_data_Returns_correct_value_if_field_and_id_are_valid(
self, mock_read_groups_file
):
""""""
mock_read_groups_file.return_value = self.groups_data
self.crud.number_of_groups = 2
get_group_data_return_value = self.crud.get_group_data(1, 'name')
self.assertEqual(get_group_data_return_value, 'default')
@patch("crud.CRUD.read_users_file")
def test_get_user_id_Returns_None_for_invalid_user_name(
self, mock_read_users_file
):
mock_read_users_file.return_value = self.users_data
get_user_id_return_value = self.crud.get_user_id("dummyName")
self.assertIsNone(get_user_id_return_value)
@patch("crud.CRUD.read_users_file")
def test_get_user_id_Returns_id_for_valid_user_name(self, mock_read_users_file):
mock_read_users_file.return_value = self.users_data
get_user_id_return_value = self.crud.get_user_id("*****@*****.**")
self.assertEqual('1', get_user_id_return_value)
@patch("crud.CRUD.read_groups_file")
def test_get_group_id_Returns_false_for_invalid_group_name(
self, mock_read_groups_file
):
mock_read_groups_file.return_value = self.groups_data
get_group_id_return_value = self.crud.get_group_id("dummyName")
self.assertIsNone(get_group_id_return_value)
@patch("crud.CRUD.read_groups_file")
def test_get_group_id_Returns_id_for_valid_group_name(self, mock_read_groups_file):
mock_read_groups_file.return_value = self.groups_data
get_group_id_return_value = self.crud.get_group_id("default")
self.assertEqual(get_group_id_return_value, '1')
@patch("crud.CRUD.read_users_file")
def test_update_users_Returns_false_for_invalid_id(
self, mock_read_users_file
):
mock_read_users_file.return_value = self.users_data
update_users_ret_val = self.crud.update_users('10', 'name', 'dummyName')
self.assertFalse(update_users_ret_val)
@patch("crud.CRUD.read_users_file")
def test_update_users_Returns_false_for_invalid_field(
self, mock_read_users_file
):
mock_read_users_file.return_value = self.users_data
update_users_ret_val = self.crud.update_users('1', 'invalid', 'dummyInvalid')
self.assertFalse(update_users_ret_val)
@patch("crud.CRUD.modify_users_file")
@patch("crud.CRUD.read_users_file")
def test_update_users_Passes_correct_data_to_modify_users_file(
self, mock_read_users_file, mock_modify_users_file
):
"""Il faut utiliser les mocks pour 'read_users_file' et 'modify_users_file'
(ou selon votre realisation)
Il faut utiliser ".assert_called_once_with(expected_data)"
"""
mock_read_users_file.return_value = self.users_data
new_data = self.users_data
new_data['1']['name'] = 'dummyName'
self.crud.update_users('1', 'name', 'dummyName')
mock_modify_users_file.assert_called_once_with(new_data)
@patch("crud.CRUD.read_groups_file")
def test_update_groups_Returns_false_for_invalid_id(
self, mock_read_groups_file
):
"""Il faut utiliser les mocks pour 'read_groups_file' et 'modify_groups_file'
(ou selon votre realisation)
"""
mock_read_groups_file.return_value = self.groups_data
update_groups_ret_val = self.crud.update_groups('10', 'name', 'dummyName')
self.assertFalse(update_groups_ret_val)
@patch("crud.CRUD.read_groups_file")
def test_update_groups_Returns_false_for_invalid_field(
self, mock_read_groups_file
):
mock_read_groups_file.return_value = self.groups_data
update_groups_ret_val = self.crud.update_groups('1', 'invalidField', 'dummyName')
self.assertFalse(update_groups_ret_val)
@patch("crud.CRUD.modify_groups_file")
@patch("crud.CRUD.read_groups_file")
def test_update_groups_Passes_correct_data_to_modify_groups_file(
self, mock_read_groups_file, mock_modify_groups_file
):
"""Il faut utiliser les mocks pour 'read_groups_file' et 'modify_groups_file'
(ou selon votre realisation)
Il faut utiliser ".assert_called_once_with(expected_data)"
"""
mock_read_groups_file.return_value = self.groups_data
new_data = self.groups_data
new_data['1']['name'] = 'dummyName'
self.crud.update_groups('1', 'name', 'dummyName')
mock_modify_groups_file.assert_called_once_with(new_data)
@patch("crud.CRUD.read_users_file")
def test_remove_user_Returns_false_for_invalid_id(
self, mock_read_users_file
):
mock_read_users_file.return_value = self.users_data
remove_user_ret_val = self.crud.remove_user('10')
self.assertFalse(remove_user_ret_val)
@patch("crud.CRUD.modify_users_file")
@patch("crud.CRUD.read_users_file")
def test_remove_user_Passes_correct_value_to_modify_users_file(
self, mock_read_users_file, mock_modify_users_file
):
mock_read_users_file.return_value = self.users_data
new_data = {"2": {
"name": "*****@*****.**",
"Trust": 65.45454,
"SpamN": 171,
"HamN": 324,
"Date_of_first_seen_message": 1596855166.0,
"Date_of_last_seen_message": 1596855166.0,
"Groups": ["default"],
}, }
self.crud.remove_user('1')
mock_modify_users_file.assert_called_once_with(new_data)
@patch("crud.CRUD.read_users_file")
def test_remove_user_group_Returns_false_for_invalid_id(
self, mock_read_users_file,
):
mock_read_users_file.return_value = self.users_data
remove_user_group_ret_val = self.crud.remove_user_group('10', 'default')
self.assertFalse(remove_user_group_ret_val)
@patch("crud.CRUD.read_users_file")
def test_remove_user_group_Returns_false_for_invalid_group(
self, mock_read_users_file
):
mock_read_users_file.return_value = self.users_data
remove_user_group_ret_val = self.crud.remove_user_group('1', 'notValidGroup')
self.assertFalse(remove_user_group_ret_val)
@patch("crud.CRUD.modify_users_file")
@patch("crud.CRUD.read_users_file")
def test_remove_user_group_Passes_correct_value_to_modify_users_file(
self, mock_read_users_file, mock_modify_users_file
):
mock_read_users_file.return_value = self.users_data
new_data = {
"1": {
"name": "*****@*****.**",
"Trust": 100,
"SpamN": 0,
"HamN": 20,
"Date_of_first_seen_message": 1596848266.0,
"Date_of_last_seen_message": 1596848266.0,
"Groups": [],
},
"2": {
"name": "*****@*****.**",
"Trust": 65.45454,
"SpamN": 171,
"HamN": 324,
"Date_of_first_seen_message": 1596855166.0,
"Date_of_last_seen_message": 1596855166.0,
"Groups": ["default"],
},
}
self.crud.remove_user_group('1', 'default')
mock_modify_users_file.assert_called_once_with(new_data)
@patch("crud.CRUD.read_groups_file")
def test_remove_group_Returns_false_for_invalid_id(
self, mock_read_groups_file
):
mock_read_groups_file.return_value = self.groups_data
remove_group_ret_val = self.crud.remove_group('10')
self.assertFalse(remove_group_ret_val)
@patch("crud.CRUD.modify_groups_file")
@patch("crud.CRUD.read_groups_file")
def test_remove_group_Passes_correct_value_to_modify_groups_file(
self, mock_read_groups_file, mock_modify_groups_file
):
mock_read_groups_file.return_value = self.groups_data
new_data = {"2": {
"name": "friends",
"Trust": 90,
"List_of_members": ["*****@*****.**"],
}, }
self.crud.remove_group('1')
mock_modify_groups_file.assert_called_once_with(new_data)
@patch("crud.CRUD.read_groups_file")
def test_remove_group_member_Returns_false_for_invalid_id(
self, mock_read_groups_file
):
mock_read_groups_file.return_value = self.groups_data
remove_group_member_ret_val = self.crud.remove_group_member('10', '*****@*****.**')
self.assertFalse(remove_group_member_ret_val)
@patch("crud.CRUD.read_groups_file")
def test_remove_group_member_Returns_false_for_invalid_group_member(
self, mock_read_groups_file
):
mock_read_groups_file.return_value = self.groups_data
remove_group_member_ret_val = self.crud.remove_group_member('1', 'invalidMember')
self.assertFalse(remove_group_member_ret_val)
@patch("crud.CRUD.modify_groups_file")
@patch("crud.CRUD.read_groups_file")
def test_remove_group_member_Passes_correct_value_to_modify_groups_file(
self, mock_read_groups_file, mock_modify_groups_file
):
mock_read_groups_file.return_value = self.groups_data
self.crud.remove_group_member('1', '*****@*****.**')
new_data = {
"1": {
"name": "default",
"Trust": 50,
"List_of_members": ["*****@*****.**"],
},
"2": {
"name": "friends",
"Trust": 90,
"List_of_members": ["*****@*****.**"],
},
}
mock_modify_groups_file.assert_called_once_with(new_data)
