예제 #1
0
파일: tests.py 프로젝트: JackyChou/SGRS
    def test_function_has_report_perm(self):
        first_user = UserCreationForm(
            data = dict(
                username = '******',
                password1 = '123456',
                password2 = '123456',
            )
        ).save()

        first_role = SGRSRole(
            name = 'first_role',
            description = 'first role for test',
            can_download = True,
        )
        first_role.save()

        first_perm = ReportPermission(
            name = 'first-perm',
            description = 'first perm',
            db_conf = 'default',
            SQL_conf = """
                SELECT count(*) FROM tablename
            """,
            filter_conf = '{}',
        )
        first_perm.save()

        second_perm = ReportPermission(
            name = 'second-perm',
            description = 'second perm',
            db_conf = 'default',
            SQL_conf = """
                SELECT count(*) FROM tablename
            """,
            filter_conf = '{}',
        )
        second_perm.save()

        first_role.report_permissions = [first_perm,]
        first_role.save()

        first_user_assignment = SGRSUserAssignment(user = first_user)
        first_user_assignment.save()
        first_user_assignment.roles = [first_role, ]
        first_user_assignment.save()

        self.assertEqual(first_user.has_report_perm(first_perm), True)
        self.assertEqual(first_user.has_report_perm('first-perm'), True)
        self.assertEqual(first_user.has_report_perm(second_perm), False)
        self.assertEqual(first_user.has_report_perm('second-perm'), False)
예제 #2
0
파일: tests.py 프로젝트: JackyChou/SGRS
    def test_function_get_all_roles(self):
        first_user = UserCreationForm(
            data = dict(
                username = '******',
                password1 = '123456',
                password2 = '123456',
            )
        ).save()

        second_user = UserCreationForm(
            data = dict(
                username = '******',
                password1 = '654321',
                password2 = '654321',
            )
        ).save()

        first_role = SGRSRole(
            name = 'first_role',
            description = 'first role for test',
            can_download = True,
        )
        first_role.save()

        second_role = SGRSRole(
            name = 'second_role',
            description = 'second role for test',
            can_download = True,
        )
        second_role.save()

        first_user_assignment = SGRSUserAssignment(user = first_user)
        first_user_assignment.save()
        first_user_assignment.roles = [second_role, first_role]
        first_user_assignment.save()

        second_user_assignment = SGRSUserAssignment(user = second_user)
        second_user_assignment.save()
        second_user_assignment.roles = [first_role, ]
        second_user_assignment.save()

        sort_key_function = lambda x:x.id
        self.assertEqual(
            sorted(first_user_assignment.roles.all(),key = sort_key_function),
            sorted(first_user.get_all_roles(),key = sort_key_function)
        )
        self.assertEqual(
            sorted(second_user_assignment.roles.all(),key = sort_key_function),
            sorted(second_user.get_all_roles(),key = sort_key_function)
        )
예제 #3
0
파일: tests.py 프로젝트: JackyChou/SGRS
    def test_function_get_all_report_combination_permissions(self):
        first_user = UserCreationForm(
            data = dict(
                username = '******',
                password1 = '123456',
                password2 = '123456',
            )
        ).save()

        first_role = SGRSRole(
            name = 'first_role',
            description = 'first role for test',
            can_download = True,
        )
        first_role.save()

        first_perm = ReportPermission(
            name = 'first-perm',
            description = 'first perm',
            db_conf = 'default',
            SQL_conf = """
                SELECT count(*) FROM tablename
            """,
            filter_conf = '{}',
        )
        first_perm.save()

        second_perm = ReportPermission(
            name = 'second-perm',
            description = 'second perm',
            db_conf = 'default',
            SQL_conf = """
                SELECT count(*) FROM tablename
            """,
            filter_conf = '{}',
        )
        second_perm.save()

        perm_comb = ReportPermissionCombination(
            name = 'first_perm_comb',
            description = 'first perm comb',
        )
        perm_comb.save()
        perm_comb.report_permissions = [first_perm, second_perm]
        perm_comb.save()

        first_role.report_permission_combinations = [perm_comb,]
        first_role.save()

        first_user_assignment = SGRSUserAssignment(user = first_user)
        first_user_assignment.save()
        first_user_assignment.roles = [first_role, ]
        first_user_assignment.save()

        sort_key_function = lambda x:x.id
        self.assertEqual(
            sorted(ReportPermissionCombination.objects.filter(
                    sgrsrole__in=[first_role, ]),
                key=sort_key_function),
            sorted(first_user.get_all_report_combination_permissions(),
                key=sort_key_function)
        )