def test_delete_story(): story_id = story_list.save_story("delete book", "delete theme") stories = story_list.get_stories() found = False for story in stories: if "delete" in story['book']: found = True assert found story_list.delete_story(story_id) stories = story_list.get_stories() found = False for story in stories: if "delete" in story['book']: found = True assert not found
def get_story_list(): stories = story_list.get_stories() for story in stories: book = story['book'] theme = story['theme'] output = template('story_list.tpl', stories=stories) return output
def test_get_story_by_key(): stories = story_list.get_stories() key1 = stories[0]['book'].split(' ')[0] assert type(key1) is str story = story_list.get_story_by_key(key1) assert key1 in story[0]['book'] assert key1 in story[0]['theme']
def test_get_stories(): story_list.save_story("Test book 1", "Test theme 1") stories = story_list.get_stories() assert type(stories) is list for story in stories: for item in ['_id', 'book', 'theme']: assert type(story[item]) is str assert story['book'] == 'Test book 1' assert story['theme'] == 'Test theme 1'
def test_save_story(): story_list.save_story("save book", "save theme") stories = story_list.get_stories() assert type(stories) is list found = False for story in stories: assert 'book' in story if story['book'] == "save book": found = True assert found
def teardown_module(): stories = story_list.get_stories() for story_id in [s['_id'] for s in stories]: story_list.delete_story(story_id)
def setup_module(): for story_id in [s['_id'] for s in story_list.get_stories()]: story_list.delete_story(story_id)
def test_get_story(): stories = story_list.get_stories() story_id = stories[0]['_id'] story_get = story_list.get_story(story_id) assert stories[0]['book'] == story_get['book'] assert stories[0]['theme'] == story_get['theme']