Ejemplo n.º 1
0
    def test_distinct_values_of(self):
        user, new = User.objects.get_or_create(name="Sergio Mena")
        role_lst = []
        with BlockSave(Role):
            for i in range(1, 6, 1):
                role = Role(name="Musician%s" % i, usr=user)
                role.save()
                role_lst.append(role)

        user_dict_1 = Role.objects.filter(
            name="Musician1").distinct_values_of("usr_id")
        assert sum(user_dict_1.values()) == 1

        user_dict_2 = Role.objects.filter(
            usr_id=user.key).distinct_values_of("usr_id")
        assert sum(user_dict_2.values()) == 5

        Role.objects.filter(active=True).delete()

        with BlockSave(Role, query_dict={'active': True}):
            for i, r in enumerate(role_lst):
                if i == 3:
                    pass
                else:
                    r.active = True
                    r.save()

        user_dict_3 = Role.objects.filter(
            active=True).distinct_values_of("usr_id")
        assert sum(user_dict_3.values()) == 4

        new_user = User(name="Valnetin Hegg")
        new_user.blocking_save()
        role_lst[0].usr = new_user
        role_lst[0].blocking_save(query_dict={'usr': new_user})
        user_dict_4 = Role.objects.filter(
            active=True, usr_id=user.key).distinct_values_of("usr_id")
        assert sum(user_dict_4.values()) == 3

        with BlockDelete(Role):
            for r in role_lst:
                r.delete()
Ejemplo n.º 2
0
    def test_distinct_values_of(self):
        user, new = User.objects.get_or_create(name="Sergio Mena")
        role_lst = []
        with BlockSave(Role):
            for i in range(1, 6, 1):
                role = Role(name="Musician%s" % i, usr=user)
                role.save()
                role_lst.append(role)

        user_dict_1 = Role.objects.filter(name="Musician1").distinct_values_of("usr_id")
        assert sum(user_dict_1.values()) == 1

        user_dict_2 = Role.objects.filter(usr_id=user.key).distinct_values_of("usr_id")
        assert sum(user_dict_2.values()) == 5

        Role.objects.filter(active=True).delete()

        with BlockSave(Role, query_dict={'active': True}):
            for i, r in enumerate(role_lst):
                if i == 3:
                    pass
                else:
                    r.active = True
                    r.save()

        user_dict_3 = Role.objects.filter(active=True).distinct_values_of("usr_id")
        assert sum(user_dict_3.values()) == 4

        new_user = User(name="Valnetin Hegg")
        new_user.blocking_save()
        role_lst[0].usr = new_user
        role_lst[0].blocking_save(query_dict={'usr': new_user})
        user_dict_4 = Role.objects.filter(active=True, usr_id=user.key).distinct_values_of("usr_id")
        assert sum(user_dict_4.values()) == 3
        
        with BlockDelete(Role):
            for r in role_lst:
                r.delete()