示例#1
0
文件: tests.py 项目: tangerine26/smyt
    def setUp(self):
        """
        Создание категорий и товаров для тестов.
        """
        cls = type(self)

        # Цикл создания категорий и принадлежащих к ним товаров, с указанием
        # цены
        for index_ in xrange(cls.CATEGORY_NUM):
            with atomic():    
                new_category = Category.objects.create(
                    name=cls.gen_category_name(index_ + 1))
                new_category.save()
                for index2_ in xrange(cls.PRODUCT_NUM_START + index_ * 2):
                    new_product_price = 90 + index2_ * 2 
                    new_product = Product(category=new_category,
                                          name=cls.gen_product_name(index2_),
                                          price=new_product_price)
                    new_product.save()
        
        # Для сверки
        self._products_num = Product.objects.count()
        self._products_more_100_num = Product.objects\
            .filter(price__gte=100).count()
        category_ids = [item.pk for item in Category.objects\
            .annotate(num_products_total=Count('product', distinct=True))\
            .filter(num_products_total__gt=10)]
        self._products_more_100_num_second = Product.objects\
            .filter(price__gte=100).filter(category_id__in=category_ids)\
            .count()
示例#2
0
    def run(self):
        cls = type(self)

        # Цикл создания категорий и принадлежащих к ним товаров, с указанием
        # цены
        for index_ in xrange(*cls.CATEGORY_TOTAL_RANGE):
            with atomic():    
                new_category = Category.objects.create(
                    name=cls.gen_category_name(index_))
                new_category.save()
                for index2_ in xrange(
                    1, random.choice(range(*cls.PRODUCT_TOTAL_RANGE))):

                    # Цена продукта
                    new_product_price = random.choice(range(*cls.PRICE_RANGE))
                    new_product = Product(category=new_category,
                                          name=cls.gen_product_name(index2_),
                                          price=new_product_price)
                    new_product.save()