def test_get_public_articles(self): # Start with 0 self.assertEqual(len(self.user1.get_public_articles()), 0) # Create Tuto ! article = ArticleFactory() article.authors.add(self.user1.user) article.sha_public = 'whatever' article.save() # Should be 1 articles = self.user1.get_public_articles() self.assertEqual(len(articles), 1) self.assertEqual(article, articles[0])
def test_get_public_articles(self): # Start with 0 self.assertEqual(len(self.user1.get_public_articles()), 0) # Create Tuto ! article = ArticleFactory() article.authors.add(self.user1.user) article.sha_public = 'whatever' article.save() # Should be 1 articles = self.user1.get_public_articles() self.assertEqual(len(articles), 1) self.assertEqual(article, articles[0])
def load_articles(cli, size, fake): """ Load articles """ articles = [] nb_arts = size * 10 percent_arts_in_validation = 0.4 percent_arts_with_validator = 0.2 percent_arts_public = 0.3 cli.stdout.write(u"Nombres d'articles à créer : {}".format(nb_arts)) cli.stdout.write(u"Nombres d'articles en validations : {}" .format(str(int(nb_arts * percent_arts_in_validation)))) cli.stdout.write(u"Nombres d'articles réservé en validations : {}" .format(str(int(nb_arts * percent_arts_with_validator)))) cli.stdout.write(u"Nombres d'articles publiés : {}" .format(str(int(nb_arts * percent_arts_public)))) tps1 = time.time() nb_users = User.objects.count() if nb_users == 0: cli.stdout.write(u"Il n'y a aucun membre actuellement." u"Vous devez rajouter les membre dans vos fixtures (member)") else: nb_sub_categories = SubCategory.objects.count() if nb_sub_categories == 0: cli.stdout.write(u"Il n'y a aucune catégories actuellement." u"Vous devez rajouter les membre dans vos fixtures (category_content)") else: sub_categories = list(SubCategory.objects.all()) profiles = list(Profile.objects.all()) perms = list(Permission.objects.filter(codename__startswith='change_').all()) staffs = list(User.objects.filter(groups__permissions__in=perms).all()) nb_staffs = len(staffs) if nb_staffs == 0: cli.stdout.write(u"Il n'y a aucun staff actuellement." u"Vous devez rajouter les staffs dans vos fixtures (staff)") else: for i in range(0, nb_arts): art = ArticleFactory(title=fake.text(max_nb_chars=80), description=fake.sentence(nb_words=15, variable_nb_words=True)) art.authors.add(profiles[i % nb_users].user) art.subcategory.add(sub_categories[random.randint(0, nb_sub_categories - 1)]) articles.append(art) if i < int(nb_arts * percent_arts_with_validator): validator = staffs[random.randint(0, nb_staffs - 1)] valid = AValidation(article=art, version=art.sha_draft, date_proposition=datetime.now(), date_reserve=datetime.now(), validator=validator, status="PENDING_V") valid.save() art.sha_validation = art.sha_draft art.save() elif i < int(nb_arts * (percent_arts_in_validation + percent_arts_with_validator)): valid = AValidation(article=art, version=art.sha_draft, date_proposition=datetime.now()) valid.save() art.sha_validation = art.sha_draft art.save() elif i < int(nb_arts * (percent_arts_in_validation + percent_arts_with_validator + percent_arts_public)): mep_art(art, art.sha_draft) valid = AValidation(article=art, version=art.sha_draft, date_proposition=datetime.now(), date_reserve=datetime.now(), validator=validator, status="ACCEPT", comment_validator=fake.text(max_nb_chars=200), date_validation=datetime.now()) valid.save() art.sha_public = art.sha_draft art.pubdate = datetime.now() art.save() sys.stdout.write(" Article {}/{} \r".format(i + 1, nb_arts)) sys.stdout.flush() tps2 = time.time() cli.stdout.write(u"\nFait en {} sec".format(tps2 - tps1))