def test_add_phone():
"""Test-func for <add_phone> function"""
address_book = AddressBook()
name, phone = (
Name('Romanskyy Andrey').name, Phone('380675119811').phone)
record = Record()
record.add_contact_phonenumb(phone)
address_book.add_record(name, record.__dict__)
datacase_one = ('Zivenko Kos', '380675119812')
result_one = (
"\n\tThe contact <Zivenko Kos> was added to address book with phone <380675119812>."
)
datacase_two = ('Romanskyy Andrey', '380675119811')
result_two = (
"\n\tThe contact's data is already in address book."
)
datacase_three = ('Romanskyy Andrey', '380675119822')
result_three = (
"\n\tThe phone <380675119822> for contact <Romanskyy Andrey>\n\t\
was successfully added to address book."
)
assert add_phone(address_book, datacase_one) == result_one
assert add_phone(address_book, datacase_two) == result_two
assert add_phone(address_book, datacase_three) == result_three
def test_change_existing_phone():
"""Test-func for <change_existing_phone> method"""
address_book = AddressBook()
name, phone = (
Name('Romanskyy Andrey').name, Phone('380675119811').phone)
record = Record()
record.add_contact_phonenumb(phone)
address_book.add_record(name, record.__dict__)
data = ('Romanskyy Andrey', '380675119811')
assert change_existing_phone(address_book, data) == (
"\n\t<380675119811> for contact <Romanskyy Andrey>\n\twas changed for <380675119812>."
)
def test_show_all_contacts():
"""Test-func for <show_all_contacts> func"""
address_book = AddressBook()
name_one, phone_one = (
Name('Romanskyy Andrey').name, Phone('380675119811').phone)
record_one = Record()
record_one.add_contact_phonenumb(phone_one)
address_book.add_record(name_one, record_one.__dict__)
name_two, phone_two = (
Name('Romanskyy Andrey').name, Phone('380675119811').phone)
record_two = Record()
record_two.add_contact_phonenumb(phone_two)
address_book.add_record(name_two, record_two.__dict__)
# use <n>
assert show_all_contacts(address_book) == (
'\n\tContinue your work with commands.'
)
# use <y>
with pytest.raises(StopIteration) as pytest_si:
show_all_contacts(address_book)
assert pytest_si.type == StopIteration
def test_change_error():
"""Test-func for <change_existing_phone> method"""
address_book = AddressBook()
name, phone = (
Name('Romanskyy Andrey').name, Phone('380675119811').phone)
record = Record()
record.add_contact_phonenumb(phone)
address_book.add_record(name, record.__dict__)
data = ('Romanskyy Andrey', '380675119811')
assert change_existing_phone(address_book, data) == (
'\n\tEnter correct number.'
)
def test_exit():
"""Test-func for <exit> func"""
address_book = AddressBook()
name, phone = (
Name('Romanskyy Andrey').name, Phone('380675119811').phone)
record = Record()
record.add_contact_phonenumb(phone)
address_book.add_record(name, record.__dict__)
with pytest.raises(SystemExit) as pytest_exit:
save_and_exit(address_book)
assert pytest_exit.type == SystemExit
assert pytest_exit.value.code == (
"\n\tThe address book was saved to <data.bin> file.\n\tGood bye!"
)
def test_show_one_contact_data():
"""Test-func for <show_one_contact_data> function"""
address_book = AddressBook()
name, phone = (
Name('Romanskyy Andrey').name, Phone('380675119811').phone)
record = Record()
record.add_contact_phonenumb(phone)
address_book.add_record(name, record.__dict__)
result = '\n\tThe contact Romanskyy Andrey has\
\n\tbirthday info - None\
\n\tphone/s - 380675119811.\n'
wrong_result = None
assert show_one_contact_data(address_book) == result
assert show_one_contact_data(address_book) == wrong_result
def test_search_contacts():
"""Test-func for <search_contacts> function"""
address_book = AddressBook()
name, phone = (
Name('Romanskyy Andrey').name, Phone('380675119811').phone)
record = Record()
record.add_contact_phonenumb(phone)
address_book.add_record(name, record.__dict__)
contact_data = 'romans'
wrong_data = 'some'
result = '\n\tThe contact Romanskyy Andrey has\
\n\tbirthday info - None\
\n\tphone/s - 380675119811.\n'
assert address_book.search_contacts(contact_data) == result
assert address_book.search_contacts(wrong_data) != result
assert address_book.search_contacts(wrong_data) is None
def test_record():
"""Test-func for <Record> class"""
rec = Record('03.06.1978')
rec.add_contact_phonenumb('38(0676665544)')
rec.add_contact_phonenumb('38(0676)665--544)')
assert rec.birthday == '03.06.1978'
assert rec.phones == ['380676665544', '380676665544']
def test_handler_func():
"""Test-func for <handler_func> function"""
address_book = AddressBook()
name, phone = (
Name('Romanskyy Andrey').name, Phone('380675119811').phone)
record = Record()
record.add_contact_phonenumb(phone)
address_book.add_record(name, record.__dict__)
show_command = 'show all'
show_result = "\n\tYou have seen all contact's info."
data = ('Nata Bazan', '380675554433')
add_command = 'add'
add_result = '\n\tThe contact <Nata Bazan> was added to address book with phone <380675554433>.'
assert handler_func(show_command, address_book, data) == (
'\n\tContinue your work with commands.'
)
assert handler_func(show_command, address_book, data) == show_result
assert handler_func(show_command, address_book, data) == show_result
assert handler_func(add_command, address_book, data) == add_result
def add_phone(address_book, data):
"""Add phone to addressbook or
creates phone if not exists
Parameters
----------
data : tuple
Returns
-------
Info-string about condition anter function has worked
"""
name, phone = inner_parser(data)
if name not in address_book:
record = Record()
record.add_contact_phonenumb(phone)
user_input = input(
f"\n\tDo you want to enter birthday for contact <{name}>? \n\tEnter y|n:"
).casefold()
if user_input == "y":
raw_birthday = input(
"\n\tEnter contact's birthday in format <dd.mm.yyyy>: ")
record.birthday = raw_birthday
birthday_info = record.days_to_birthday()
print(birthday_info)
print('*' * 80)
address_book.add_record(name, record.__dict__)
result = (
f"\n\tThe contact <{name}> was added to address book with phone <{phone}>."
)
elif name in address_book and phone in address_book[name]["phones"]:
result = "\n\tThe contact's data is already in address book."
elif name in address_book and phone not in address_book[name]["phones"]:
address_book[name]["phones"].append(phone)
result = f"\n\tThe phone <{phone}> for contact <{name}>\n\t\
was successfully added to address book."
return result
def test_days_to_birthday():
"""Test-func for <days_to_birthday> method"""
rom = Record('03.06.1978')
assert rom.days_to_birthday() == (
"\n\tThere are 214 days left until the contact's next birthday"
)
def test_add_contact_phonenumber():
"""Test-func for <add_contact_phonenumber> method"""
record = Record()
record.add_contact_phonenumb('3(8067)511-98-11')
result = ['380675119811']
assert record.phones == result