1from UM.Message import Message
2
3
4def test_addAction():
5    message = Message()
6    message.addAction(action_id = "blarg", name = "zomg", icon = "NO ICON", description="SuperAwesomeMessage")
7
8    assert len(message.getActions()) == 1
9
10
11def test_gettersAndSetters():
12    message = Message(text = "OMG", title="YAY", image_caption="DERP", image_source= "HERP", option_text="FOO", option_state= False)
13    message.setMaxProgress(200)
14    assert message.getText() == "OMG"
15    assert message.getTitle() == "YAY"
16    assert message.getImageCaption() == "DERP"
17    assert message.getImageSource() == "HERP"
18    assert message.getOptionText() == "FOO"
19    assert message.getMaxProgress() == 200
20    assert message.getOptionState() == False
21
22    message.setTitle("whoo")
23    assert message.getTitle() == "whoo"
24
25
26def test_dismissable():
27    # In certain conditions the message is always set to dismissable (even if we asked for a non dimissiable message)
28    message = Message(lifetime=0, dismissable=False)
29    assert message.isDismissable()
30
31
32