def setUp(self): super(TestPermissionsLoading, self).setUp() self.api = Api() self.generator = generator.ObjectGenerator() self.control_id = factories.ControlFactory().id _, user = self.generator.generate_person(user_role="Creator") self.api.set_user(user)
def setUp(self): converters.TestCase.setUp(self) self.client.get("/login") self.api_helper = api_helper.Api() self.import_file("request_full_no_warnings.csv") self.request1 = Request.query.filter_by(slug="Request 1").one() self.request3 = Request.query.filter_by(slug="Request 3").one() self.generator = generator.ObjectGenerator() self.handlers = contributions.contributed_notifications()
def setUp(self): super(TestAssessmentDataHandlers, self).setUp() self.client.get("/login") self.api_helper = api_helper.Api() self.import_file("assessment_notifications.csv") self.asmt1 = Assessment.query.filter_by(slug="A 1").one() self.asmt3 = Assessment.query.filter_by(slug="A 3").one() self.generator = generator.ObjectGenerator() self.handlers = contributions.contributed_notifications()
def setUp(self): # pylint: disable=super-on-old-class # pylint seems to get confused, mro chain successfully resolves and returns # <type 'object'> as last entry. super(TestClonable, self).setUp() self.client.get("/login") self.generator = generator.Generator() self.object_generator = generator.ObjectGenerator()
def setUp(self): super(TestPermissionsLoading, self).setUp() self.api = Api() self.generator = generator.ObjectGenerator() _, self.user = self.generator.generate_person(user_role="Creator") _, self.user1 = self.generator.generate_person(user_role="Creator") self.api.set_user(self.user) self.user_id = self.user.id self.user1_id = self.user1.id
def setUp(self): self.clear_data() super(TestAuditActions, self).setUp() self.client.get("/login") self.gen = generator.ObjectGenerator() filename = "program_audit.csv" self.import_file(filename) self.assertEqual(2, Audit.query.count()) audit = Audit.query.filter(Audit.slug == "Aud-1").first() program = Program.query.filter(Program.slug == "prog-1").first() self.assertEqual(audit.program_id, program.id)
def setUp(self): """Set up for test cases.""" from ggrc.fulltext import listeners from ggrc.models.background_task import reindex_on_commit super(TestTaskqueueIndexing, self).setUp() listeners.reindex_on_commit = reindex_on_commit self.api = api_helper.Api() self.init_taskqueue() self.object_generator = generator.ObjectGenerator() self._bg_tasks = {}
def setUp(self): """Set up for test methods.""" super(TestBulkIssuesSync, self).setUp() self.api = Api() self.gen = generator.ObjectGenerator() self.role_people = { "Audit Captains": factories.PersonFactory(email="*****@*****.**"), "Creators": factories.PersonFactory(email="*****@*****.**"), "Assignees": factories.PersonFactory(email="*****@*****.**"), "Verifiers": factories.PersonFactory(email="*****@*****.**"), } self.issue_id = "42"
def __init__(self, user_id, acr, parent=None): """Set up objects for Assessment permission tests. Args: user_id: Id of user under which all operations will be run. acr: Instance of ACR that should be assigned for tested user. parent: Model name in scope of which objects should be set up. """ self.setup_program_scope(user_id, acr, parent) self.api = Api() self.objgen = generator.ObjectGenerator() self.objgen.api = self.api if user_id: user = all_models.Person.query.get(user_id) self.api.set_user(user)
def setUp(self): super(TestCADPermissions, self).setUp() self.api = Api() self.client.get("/login") self.definition_type = "control" self.generator = generator.ObjectGenerator() roles = {r.name: r for r in all_models.Role.query.all()} with factories.single_commit(): self.people = { "Creator": factories.PersonFactory(), "Reader": factories.PersonFactory(), "Editor": factories.PersonFactory(), "Administrator": factories.PersonFactory(), } for role_name in ["Creator", "Reader", "Editor", "Administrator"]: rbac_factories.UserRoleFactory(role=roles[role_name], person=self.people[role_name]) self.fake_cad = factories.CustomAttributeDefinitionFactory( definition_type=self.definition_type)
def __init__(self, user_id, acr, parent=None): """Set up objects for permission tests. Args: user_id: Id of user under which all operations will be run. acr: Instance of ACR that should be assigned for tested user. parent: Model name in scope of which objects should be set up. """ self.api = Api() self.objgen = generator.ObjectGenerator() self.objgen.api = self.api self.acr = acr self.user_id = user_id self.parent_name = parent self.document_id = None self.parent = None self.parent_id = None self.setup_models(self.parent_name) self.set_user(user_id)
def setUp(self): """setUp, nothing else to add.""" super(TestSyncServiceControl, self).setUp() self.api = api_helper.Api() self.generator = generator.ObjectGenerator() self.app_user_email = "*****@*****.**" self.ext_user_email = "*****@*****.**" self.ext_owner_email = "*****@*****.**" self.ext_compliance_email = "*****@*****.**" settings.EXTERNAL_APP_USER = self.app_user_email custom_headers = { 'X-GGRC-user': '******' % self.app_user_email, 'X-external-user': '******' % self.ext_user_email } self.api.headers.update(custom_headers) self.api.client.get("/login", headers=self.api.headers)
def setUp(self): super(TestIssueAuditMapping, self).setUp() self.generator = generator.ObjectGenerator(fail_no_json=False) control = factories.ControlFactory() revision = all_models.Revision.query.filter( all_models.Revision.resource_type == control.type, all_models.Revision.resource_id == control.id, ).first() with factories.single_commit(): self.audit = factories.AuditFactory() self.snapshot = factories.SnapshotFactory(parent=self.audit, revision=revision) self.other_audits = [factories.AuditFactory() for _ in range(2)] self.issue_mapped = factories.IssueFactory() self.issue_unmapped = factories.IssueFactory() self.issue_audit_mapping = factories.RelationshipFactory( source=self.audit, destination=self.issue_mapped, context=self.audit.context, )
def setUp(self): """Setup tests data""" super(TestIssueUnmap, self).setUp() self.generator = generator.ObjectGenerator(fail_no_json=False) # TODO: replace this hack with a special test util from ggrc.login import noop noop.login() # this is needed to pass the permission checks in automapper with factories.single_commit(): audit = factories.AuditFactory() self.audit_id = audit.id assessments = [ factories.AssessmentFactory(audit=audit) for _ in range(2) ] controls = [factories.ControlFactory() for _ in range(2)] snapshots = self._create_snapshots(audit, controls) self.snapshot_ids = [s.id for s in snapshots] issue = factories.IssueFactory() self.issue_id = issue.id factories.RelationshipFactory(source=audit, destination=assessments[0]) factories.RelationshipFactory(source=audit, destination=assessments[1]) factories.RelationshipFactory( source=assessments[0], destination=snapshots[0] ) factories.RelationshipFactory( source=assessments[0], destination=snapshots[1] ) factories.RelationshipFactory( source=assessments[1], destination=snapshots[1] ) self.unmap_rel_id1 = factories.RelationshipFactory( source=issue, destination=assessments[0] ).id self.unmap_rel_id2 = factories.RelationshipFactory( source=issue, destination=assessments[1] ).id
def setUp(self): super(TestAssessmentUpdates, self).setUp() self.api = ggrc.api_helper.Api() self.generator = generator.ObjectGenerator() _, program = self.generator.generate_object(all_models.Program) program_id = program.id _, audit = self.generator.generate_object( all_models.Audit, { "title": "Assessment Updates Audit", "program": { "id": program_id }, "status": "Planned" }, ) with freezegun.freeze_time("2015-04-01 17:13:15"): _, assessment = self.generator.generate_object( all_models.Assessment, { "title": "Assessment-Comment", "audit": { "id": audit.id, "type": "Audit" }, "audit_title": audit.title, "people_value": [], "default_people": { "verifiers": "Admin", "assignees": "Admin", }, "context": { "id": audit.context.id }, }) self.assessment_id = assessment.id self.assessment = all_models.Assessment.query.get(self.assessment_id)
def __init__(self, user_id, acr, parent=None): """Set up objects for Audit permission tests. Args: user_id: Id of user under which all operations will be run. acr: Instance of ACR that should be assigned for tested user. parent: Model name in scope of which objects should be set up. """ # pylint: disable=unused-argument self.setup_program_scope(user_id, acr, "Audit") self.api = Api() self.objgen = generator.ObjectGenerator() self.objgen.api = self.api self.admin_control_id = { name: id for id, name in access_control.role.get_custom_roles_for("Control").items() }["Admin"] if user_id: self.user_id = user_id user = all_models.Person.query.get(user_id) self.api.set_user(user)
def setUp(self): super(TestWithLastCommentAssessment, self).setUp() self.client.get("/login") self.api = Api() self.gen = generator.ObjectGenerator() self.asmnt_comments = defaultdict(dict) with factories.single_commit(): for _ in range(3): asmnt = factories.AssessmentFactory() for _ in range(3): comment = factories.CommentFactory( description=factories.random_str()) self.asmnt_comments[asmnt.id][ comment.id] = comment.description factories.RelationshipFactory(source=asmnt, destination=comment) query = all_models.Revision.query.filter_by(resource_type="Comment") revision_ids = [revision.id for revision in query] self.api.send_request(self.api.client.post, api_link="/admin/compute_attributes", data={"revision_ids": revision_ids})
def setUp(self): """Setup tests data""" super(TestIssueUnmap, self).setUp() self.generator = generator.ObjectGenerator(fail_no_json=False) with factories.single_commit(): audit = factories.AuditFactory() self.audit_id = audit.id assessments = [ factories.AssessmentFactory(audit=audit) for _ in range(2) ] objectives = [factories.ObjectiveFactory() for _ in range(2)] snapshots = self._create_snapshots(audit, objectives) self.snapshot_ids = [s.id for s in snapshots] issue = factories.IssueFactory() self.issue_id = issue.id factories.RelationshipFactory(source=audit, destination=assessments[0]) factories.RelationshipFactory(source=audit, destination=assessments[1]) factories.RelationshipFactory( source=assessments[0], destination=snapshots[0] ) factories.RelationshipFactory( source=assessments[0], destination=snapshots[1] ) factories.RelationshipFactory( source=assessments[1], destination=snapshots[1] ) self.unmap_rel_id1 = factories.RelationshipFactory( source=issue, destination=assessments[0] ).id self.unmap_rel_id2 = factories.RelationshipFactory( source=issue, destination=assessments[1] ).id
def setUp(self): super(TestBasicCsvImport, self).setUp() self.generator = generator.ObjectGenerator() self.api = api_helper.Api() self.client.get("/login")
# Copyright (C) 2017 Google Inc. # Licensed under http://www.apache.org/licenses/LICENSE-2.0 <see LICENSE file> """Integration tests for "preconditions_failed" CAV and CAable fields logic.""" from ggrc.models.assessment import Assessment from integration.ggrc import TestCase from integration.ggrc import generator from integration.ggrc.models import factories GENERATOR = generator.ObjectGenerator() # pylint: disable=too-many-instance-attributes class CustomAttributeMock(object): """Defines CustomAttributeDefinition and CustomAttributeValue objects""" # pylint: disable=too-many-arguments def __init__(self, attributable, attribute_type="Text", mandatory=False, dropdown_parameters=None, global_=False, value=None): self.attributable = attributable self.attribute_type = attribute_type self.mandatory = mandatory self.dropdown_parameters = dropdown_parameters self.attribute_value = value self.global_ = global_ self.definition = self.make_definition()
def setUp(self): super(TestReviewStatusUpdate, self).setUp() self.api = Api() self.api.client.get("/login") self.generator = generator.ObjectGenerator()
def setUp(self): super(TestReviewNotification, self).setUp() self.client.get("/login") self.generator = generator.ObjectGenerator() self.api = Api()
def setUp(self): super(TestAutomappings, self).setUp() self.gen = generator.ObjectGenerator() self.api = self.gen.api
def setUp(self): super(TestLastAssessmentDate, self).setUp() self.api = Api() self.generator = generator.ObjectGenerator() self.client.get("/login") person = models.Person.query.first() admin_control = models.AccessControlRole.query.filter_by( name="Admin", object_type="Control").first() admin_objective = models.AccessControlRole.query.filter_by( name="Admin", object_type="Objective").first() with factories.single_commit(): controls = [ factories.ControlFactory(slug="Control_{}".format(i), title="Control_{}".format(i)) for i in range(5) ] objectives = [ factories.ObjectiveFactory(slug="Objective_{}".format(i), title="Objective_{}".format(i)) for i in range(2) ] for obj in itertools.chain(controls, objectives): acr = admin_control if obj.type == "Control" else admin_objective factories.AccessControlList(object=obj, person=person, ac_role=acr) audit_0 = factories.AuditFactory(title="Audit_0") audit_1 = factories.AuditFactory(title="Audit_1") audit_0_snapshots = self._create_snapshots( audit_0, controls[:2] + objectives[:1]) audit_1_snapshots = self._create_snapshots( audit_1, controls[1:4] + objectives) assessment_0 = factories.AssessmentFactory(title="Assessment_0", audit=audit_0) assessment_1 = factories.AssessmentFactory(title="Assessment_1", audit=audit_1) factories.RelationshipFactory(source=assessment_0, destination=audit_0) factories.RelationshipFactory(source=audit_1, destination=assessment_1) # Audit 0 assessment mappings: factories.RelationshipFactory( source=assessment_0, destination=audit_0_snapshots[1], # snapshot of control_1 ) factories.RelationshipFactory( source=assessment_0, destination=audit_0_snapshots[2], # snapshot of objective_0 ) # Audit 1 assessment mappings: factories.RelationshipFactory( source=audit_1_snapshots[0], # snapshot of control_1 destination=assessment_1, ) factories.RelationshipFactory( source=assessment_1, destination=audit_1_snapshots[1], # snapshot of control_2 ) factories.RelationshipFactory( source=assessment_1, destination=audit_1_snapshots[3], # snapshot of objective_0 ) factories.RelationshipFactory( source=audit_1_snapshots[4], # snapshot of objective_1 destination=assessment_1, )
def setUp(self): self.clear_data() super(TestSnapshotIndexing, self).setUp() self.generator = generator.ObjectGenerator() self.client.get("/login")
def setUp(self): super(TestMetric, self).setUp() self.gen = generator.ObjectGenerator()
def setUp(self): super(TestAssignableNotificationUsingAPI, self).setUp() self.api_helper = api_helper.Api() self.objgen = generator.ObjectGenerator()
def setUp(self): super(TestCloneWorkflow, self).setUp() self.object_generator = generator.ObjectGenerator()
def setUp(self): super(TestCommentNotification, self).setUp() self.client.get("/login") self._fix_notification_init() self.generator = generator.ObjectGenerator()
def setUp(self): """Setup test case.""" super(TestCommentAdmin, self).setUp() self.response = self.client.get("/login") self.generator = generator.ObjectGenerator()