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