Python Voting.Voteの例

コード例 #1

ファイル: test_Voting.py プロジェクト: Sergeileduc/KoalaBot

def test_vote_remove_role():
    vote = Voting.Vote(111, "Test Vote", 222, 333, db_manager)
    vote.add_role(777)
    vote.remove_role(777)
    assert 777 not in vote.target_roles
    in_db = db_manager.db_execute_select("SELECT * FROM VoteTargetRoles WHERE vote_id=? AND role_id=?", (111, 777))
    assert not in_db

コード例 #2

ファイル: test_Voting.py プロジェクト: Sergeileduc/KoalaBot

def test_vote_is_ready():
    vote = Voting.Vote(111, "Test Vote", 222, 333, db_manager)
    vote.add_option(Voting.Option(111, "head", "body"))
    assert not vote.is_ready()
    vote.add_option(Voting.Option(122, "head", "body"))
    assert vote.is_ready()