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)