cnr-ibba / IMAGE-InjectTool

    # mock subprocess.run an raise Exception. Read it and update submission

    # message using helpers.upload_cryoweb

    @patch('cryoweb.helpers.send_message_to_websocket')

    @patch('asyncio.get_event_loop')

    def test_upload_cryoweb_errors(self, asyncio_mock,

                                   send_message_to_websocket_mock):

        """Testing errors in uploading cryoweb data"""

        tmp = asyncio_mock.return_value

        tmp.run_until_complete = Mock()

        with patch('subprocess.run') as runMock:

            runMock.side_effect = Exception("Test upload failed")

            self.assertFalse(upload_cryoweb(self.submission.id))

            # reload submission

            self.submission = Submission.objects.get(pk=1)

            self.assertEqual(

                self.submission.status,

                ERROR)

            self.assertIn(

                "Test upload failed",

                self.submission.message)

            self.assertEqual(asyncio_mock.call_count, 1)

            self.assertEqual(tmp.run_until_complete.call_count, 1)

            self.assertEqual(send_message_to_websocket_mock.call_count, 1)

            send_message_to_websocket_mock.assert_called_with(

                {'message': 'Error',

                 'notification_message': 'Error in importing data: Test '

                                         'upload failed'}, 1)

class CryowebImport(CryoWebMixin, BaseTestCase, TestCase):

                                    'animal_unkn': 3, 'sample_unkn': 1,

                                    'animal_issues': 0, 'sample_issues': 0}}, 1)

    @patch('cryoweb.helpers.send_message_to_websocket')

    @patch('asyncio.get_event_loop')

    @patch("cryoweb.helpers.check_UID", side_effect=Exception("Test message"))

    def test_cryoweb_import_errors(self, my_check, asyncio_mock,

                                   send_message_to_websocket_mock):

        """Testing importing with data into UID with errors"""

        tmp = asyncio_mock.return_value

        tmp.run_until_complete = Mock()

        self.assertFalse(cryoweb_import(self.submission))

        self.assertTrue(my_check.called)

        # reload submission

        self.submission = Submission.objects.get(pk=1)

        self.assertEqual(

            self.submission.status,

            ERROR)

        self.assertIn(

            "Test message",

            self.submission.message)

        self.assertEqual(asyncio_mock.call_count, 1)

        self.assertEqual(tmp.run_until_complete.call_count, 1)

        self.assertEqual(send_message_to_websocket_mock.call_count, 1)

        send_message_to_websocket_mock.assert_called_with(

            {'message': 'Error',

             'notification_message': 'Error in importing data: Test message'},

            1)

class CryowebReload(CryoWebMixin, BaseTestCase, TestCase):