示例#1
0
文件： app.py 项目： pweids/moviebattle 
def add_movie(link: str) -> Movie:
    res = get_movie(link)
    if isinstance(res, str):
        raise ValueError(f"Error: Cannot add movie: {res}")
    else:
        save_movie(res)
        movies.add(res)
        return res
示例#2
0
def test_save_movie():
    dbf = 'test.db'
    try:
        m = get_movie.get_movie(NO_COUNTRY)
        db.save_movie(m, dbf)

        ms = db.get_movies(dbf)

        assert m in ms
    finally:
        os.remove(dbf)
示例#3
0
def test_update():
    dbf = 'test2.db'
    try:
        m = get_movie.get_movie(NO_COUNTRY)
        db.save_movie(m, dbf)
        m.viewed = True
        db.update_viewed(m, dbf)
        
        ms = db.get_movies(dbf)

        assert m in ms
    finally:
        if os.path.exists(dbf):
            os.remove(dbf)
示例#4
0
def test_get_movie_2_dirs():
    m = get_movie.get_movie(NO_COUNTRY)
    assert len(m.directors) == 2
示例#5
0
def test_get_movie():
    m = get_movie.get_movie(BAD_WORDS)
    assert isinstance(m, Movie)
示例#6
0
def test_get_tv():
    res = get_movie.get_movie(ARRESTED_DEVELOPMENT)
    assert isinstance(res, str)
    print(res)
    assert 'TV' in res
示例#7
0
def test_get_bad_link():
    res = get_movie.get_movie("https://www.imdb.com/name/hm0000867/")
    assert isinstance(res, str)
    assert 'valid' in res