def test_download_big_or_not_img(self):
message_form = MessageForm(self.driver)
message_form.open_img_window()
message_form.choose_img_button()
message_form.choose_img(self.BIG_IMG_PATH)
message_form.download_img()
error_message = message_form.get_error_message()
self.assertEqual(error_message, self.ERROR_SIZE)
self.driver.refresh()
message_form.open_img_window()
message_form.choose_img_button()
message_form.choose_img(self.NOT_IMG_PATH)
message_form.download_img()
error_message = message_form.get_error_message()
self.assertEqual(error_message, self.ERROR_TYPE)
def test_download_unselected_img(self):
message_form = MessageForm(self.driver)
message_form.open_img_window()
message_form.download_img()
error_message = message_form.get_error_message()
self.assertEqual(error_message, self.ERROR_TYPE)