def test_assign(self):
""" PUT /groups/id/users assign users """
mock = MockUtils.create_mock_response(status_code=204, data=None)
employee1 = Employee(firstname=u"Steven", lastname=u"Gerrard")
employee2 = Employee(firstname=u"Gerrard", lastname=u"Lampard")
with patch('requests.Session.request', mock):
(objects, connection) = self.group.assign([employee1, employee2],
Employee)
method = MockUtils.get_mock_parameter(mock, 'method')
url = MockUtils.get_mock_parameter(mock, 'url')
headers = MockUtils.get_mock_parameter(mock, 'headers')
self.assertEqual(connection.response.status_code, 204)
self.assertEqual(
url,
'https://vsd:8443/api/v3_2/groups/' + self.group.id + '/users')
self.assertEqual(method, 'PUT')
self.assertEqual(
headers['Authorization'],
'XREST dXNlcjo1MWYzMTA0Mi1iMDQ3LTQ4Y2EtYTg4Yi02ODM2ODYwOGUzZGE=')
self.assertEqual(headers['X-Nuage-Organization'], 'enterprise')
self.assertEqual(headers['Content-Type'], 'application/json')
self.assertEqual(objects, [employee1, employee2])
self.assertEqual(self.group.employees, [employee1, employee2])
self.group.employees.flush()
def test_genealogic_types(self):
"""
"""
enterprise1 = Enterprise(id='4', name='enterprise')
employee = Employee()
employee.parent_object = enterprise1
self.assertEquals(employee.genealogic_types(), ['user', 'enterprise'])
def test_genealogy_contains_id(self):
"""
"""
enterprise1 = Enterprise(id='4', name='enterprise')
enterprise1.id = '1'
employee = Employee()
employee.id = '2'
employee.parent_object = enterprise1
self.assertTrue(employee.genealogy_contains_id('1'))
def test_genealogic_ids(self):
"""
"""
enterprise1 = Enterprise(id='4', name='enterprise')
enterprise1.id = '1'
employee = Employee()
employee.id = '2'
employee.parent_object = enterprise1
self.assertEquals(employee.genealogic_ids(), ['2', '1'])
def test_assign_with_commit(self):
""" PUT /groups/id/users assign users without commit """
mock = MockUtils.create_mock_response(status_code=204, data=None)
employee1 = Employee(firstname=u"Steven", lastname=u"Gerrard")
employee2 = Employee(firstname=u"Gerrard", lastname=u"Lampard")
with patch('requests.Session.request', mock):
(objects, connection) = self.group.assign([employee1, employee2],
Employee,
commit=False)
self.assertEqual(objects, [employee1, employee2])
self.assertEqual(self.group.employees, [])
''' creates a test database ''' from tests.models import db, Group, Company, Employee from faker import Faker ######################################## fake = Faker() db.drop_all() db.create_all() ######################################## GROUP1 = Group.create(abbr='HRM') GROUP2 = Group.create(abbr='R&D') COMPANY = Company.create(name=fake.company()) PERSON1 = Employee.create(email=fake.simple_profile()['mail'], company=COMPANY) PERSON2 = Employee.create(email=fake.simple_profile()['mail'], company=COMPANY) PERSON1.groups.append(GROUP1) PERSON2.groups.append(GROUP2) print(COMPANY)
