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)