def testStoryExists(self):
story = self._getFakeStory()
db = StoryDatabase()
db.createDatabase(self.TEST_DB_NAME)
db.addStory(story)
saved_story = db.getStory(str(story['stories_id']))
self.assertTrue(db.storyExists(str(story['stories_id'])))
self.assertFalse(db.storyExists('43223535'))
db.deleteDatabase(self.TEST_DB_NAME)
def testAddStory(self):
story = self._getFakeStory()
db = StoryDatabase()
db.createDatabase(self.TEST_DB_NAME)
worked = db.addStory(story)
self.assertTrue(worked)
worked = db.addStory(story)
self.assertFalse(worked)
saved_story = db.getStory(str(story['stories_id']))
self.assertEquals(saved_story['_id'], str(story['stories_id']))
self.assertEquals(saved_story['story_sentences_count'], 2)
db.deleteDatabase(self.TEST_DB_NAME)
def testGetMaxStoryId(self):
story1 = self._getFakeStory()
story1['stories_id'] = "1000"
story2 = self._getFakeStory()
story1['stories_id'] = "2000"
db = StoryDatabase()
db.createDatabase(self.TEST_DB_NAME)
db._db.save(mediacloud.examples.getAllExampleViews())
self.assertEquals(db.getMaxStoryId(),0)
db.addStory(story1)
db.addStory(story2)
self.assertEquals(db.getMaxStoryId(),2000)
db.deleteDatabase(self.TEST_DB_NAME)
def testCreateMaxIdView(self):
db = StoryDatabase()
db.createDatabase(self.TEST_DB_NAME)
db._db.save(mediacloud.examples.getAllExampleViews())
self.assertEquals(db.getMaxStoryId(),0)
db.deleteDatabase(self.TEST_DB_NAME)
def testManageDatabase(self):
db = StoryDatabase()
db.createDatabase(self.TEST_DB_NAME)
db.deleteDatabase(self.TEST_DB_NAME)
