# -*- coding: utf-8 -*-
from model.contact import Contact
from model.group import Group
import random
from fixture.orm import ORMGroup
from fixture.orm import ORMContact


def test_add_contact_to_group(app, db, orm, check_ui):
    app.open_home_page()
    if len(db.get_contact_list()) == 0:
        app.contact.create(Contact(first_name="Vasia"))
    if len(db.get_group_list()) == 0:
        app.group.create(Group(name="test"))
    old_contacts = orm.get_contacts_in_group()
    contact = random.choice(orm.get_contacts_not_in_group)
    app.contact.add_contact_to_group_by_id(contact.id)
    app.open_home_page()
    old_contacts.append(contact)
    new_contacts = orm.get_contacts_in_group()
    assert sorted(old_contacts, key=Contact.id_or_max) == sorted(new_contacts, key=Contact.id_or_max)



Testing started at 2:43 PM ...
C:\Users\PStoiko\PycharmProjects\python_training_2\env\Scripts\python.exe "C:\Program Files\JetBrains\PyCharm Community Edition 2018.2.4\helpers\pycharm\_jb_pytest_runner.py" --path C:/Users/PStoiko/PycharmProjects/python_training_2/test/test_add_contact_to_group.py
Launching pytest with arguments C:/Users/PStoiko/PycharmProjects/python_training_2/test/test_add_contact_to_group.py in C:\Users\PStoiko\PycharmProjects\python_training_2\test

============================= test session starts =============================
platform win32 -- Python 3.7.1, pytest-3.10.0, py-1.7.0, pluggy-0.8.0
rootdir: C:\Users\PStoiko\PycharmProjects\python_training_2\test, inifile:
test_add_contact_to_group.py:None (test_add_contact_to_group.py)
ImportError while importing test module 'C:\Users\PStoiko\PycharmProjects\python_training_2\test\test_add_contact_to_group.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
test_add_contact_to_group.py:5: in <module>
    from fixture.orm import ORMGroup
E   ImportError: cannot import name 'ORMGroup' from 'fixture.orm' (C:\Users\PStoiko\PycharmProjects\python_training_2\fixture\orm.py)
collected 0 items / 1 errors

=================================== ERRORS ====================================
________________ ERROR collecting test_add_contact_to_group.py ________________
ImportError while importing test module 'C:\Users\PStoiko\PycharmProjects\python_training_2\test\test_add_contact_to_group.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
test_add_contact_to_group.py:5: in <module>
    from fixture.orm import ORMGroup
E   ImportError: cannot import name 'ORMGroup' from 'fixture.orm' (C:\Users\PStoiko\PycharmProjects\python_training_2\fixture\orm.py)
!!!!!!!!!!!!!!!!!!! Interrupted: 1 errors during collection !!!!!!!!!!!!!!!!!!!
=========================== 1 error in 0.20 seconds ===========================
Process finished with exit code 0