def add_2_articles(self): self.add_2_categories() s = self.session() a1 = Article(u'This is content of article 1', Category.find_by_name(s, u'c1'), u'Title of c1') a2 = Article(u'This is content of article 2', Category.find_by_name(s, u'c2'), u'Title of c2') s.add_all([a1, a2]) s.commit()
def add_test_data(): from ljn.Model import Category s = session_maker() if not len(Category.all(s)): s.add(Category(u'c1')) s.add(Category(u'c2')) if not len(Article.all(s)): s.add(Article('this is content 1', Category.find_by_name(s, u'c1'), u'title of a1')) s.add(Article('this is content 2', Category.find_by_name(s, u'c1'), u'title of a2')) s.add(Article('this is content 3', Category.find_by_name(s, u'c2'), u'title of a3')) article = Category.find_by_name(s, u'c1').articles[0] if not len(article.new_words): w = Word('is') article.new_words.append(ArticleNewWord(article, w, 'is')) s.commit()
def testCategory(self): s = self.session() s.add(Category(u'c1')) self.assertEqual(1, len(s.query(Category).all())) c = Category.find_by_name(s, u'c1') self.assertEqual(c.name, u'c1') self.assertEqual(len(c.articles), 0) s.commit() s.add(Article(u'ac', c, u'at')) s.commit() self.assertIsNotNone(c.create_date) self.assertEqual(len(c.articles), 1) s.delete(c) s.commit() self.assertEqual(len(Article.all(s)), 0)