def run(self, **kwargs): input_var = input( 'Are you sure you wish to create fake data? y/n: ') if input_var == 'y' or input_var == 'Y': fdc = FakeDataFactory(db) fdc.generate_data(**kwargs) fdc.db.session.commit() fdc.increment_sequence_ids() print('Fake data created')
def setUp(self): db.create_all() self.datafactory = FakeDataFactory(1234)
class FactoryTest(MainTestCase): def setUp(self): db.create_all() self.datafactory = FakeDataFactory(1234) def tearDown(self): db.session.remove() db.drop_all() def test_authorfactory(self): author = AuthorFactory() self.assertIsInstance(author, Author) self.assertIsNotNone(author.id) self.assertIsNotNone(author.name) self.assertIsNotNone(author.givenname) self.assertIsNotNone(author.familyname) self.assertIsNotNone(author.url) self.assertIsNotNone(author.email) author2 = AuthorFactory() self.assertNotEqual(author.id, author2.id) def test_userfactory(self): user = UserFactory() self.assertIsInstance(user, User) self.assertIsNotNone(user.id) self.assertIsNotNone(user.email) self.assertIsNotNone(user.active) def test_entryfactory(self): entry = EntryFactory() self.assertIsInstance(entry, Entry) self.assertIsNotNone(entry.id) self.assertIsNotNone(entry.title) self.assertIsNotNone(entry.guid) self.assertIsNotNone(entry.content) self.assertIsInstance(entry.published, datetime) def test_subscriptionfactory(self): sub = SubscriptionFactory() self.assertIsInstance(sub, Subscription) self.assertIsNotNone(sub.user_id) self.assertIsNotNone(sub.author_id) self.assertIsNotNone(sub.active) self.assertIsInstance(sub.created_on, datetime) def test_feedfactory(self): feed = FeedFactory() self.assertIsInstance(feed, Feed) self.assertIsNotNone(feed.id) self.assertIsNotNone(feed.topic) self.assertIsNotNone(feed.hub) self.assertIsNotNone(feed.title) self.assertLess(len(feed.description), 501) self.assertIsNotNone(feed.lease_seconds) self.assertIsNotNone(feed.sub_time) self.assertIsNotNone(feed.unique_url) self.assertIsNotNone(feed.secret) self.assertIsNotNone(feed.status) def test_datafactory_creates_users(self): users = self.datafactory.create_users(10) self.assertEqual(len(users), 10) def test_datafactory_creates_feeds(self): feeds = self.datafactory.create_feeds(10) self.assertEqual(len(feeds), 10) def test_datafactory_creates_authors(self): authors = self.datafactory.create_authors(10) self.assertEqual(len(authors), 10) def test_datafactory_creates_entries(self): entries = self.datafactory.create_entries(10) self.assertEqual(len(entries), 10) def test_datafactory_creates_subscriptions(self): users = self.datafactory.create_users(2) authors = self.datafactory.create_authors(10) subs = self.datafactory.create_subscriptions(users, authors, 20, 5) self.assertGreater(len(subs), 0) self.assertIsNotNone(users[0].subscriptions)