def test_unique_data3(self): msg1 = QueueReqMessage() msg2 = QueueReqMessage() msg1.params["ala"] = True msg2.params['ala'] = False self.assertEqual(False, msg2.params['ala']) self.assertEqual(True, msg1.params['ala'])
def test_routing_call_process(self): router = Router(MagicMock('application')) router.add_method('metoda', MagicMock(return_value=self.method)) message = QueueReqMessage() message.method = 'metoda' message.params = {'key': 'test'} router.find_method(message) self.method.process.assert_called_with(key='test')
def run_analysis(send): # msg = QueueReqMessage(id='1', params={'login': '******', 'password': '******'}, method='login') msg = QueueReqMessage() msg.id = 1 msg.method = 'thesismethods-run_analysis' msg.params = {'thesis_id': 7} send.on_recv(callback) send.send_string(str(msg))
def test_call_unknow_ext(self): """Test get text when no extension given""" method = GetText() msg = QueueReqMessage() msg.params = {} msg.params['file_path'] = '/tmp/test' method.application = self.mock_application method(msg) self.assertTrue(self.mock_application.logger.error.called)
def test_user_exists(self): method = Login() msg = QueueReqMessage() msg.params = {'login': '******', 'password': '******'} method.request = msg method.success = MagicMock() method.process(**msg.params) self.assertTrue(self.mock_user.select.called) self.assertTrue(method.success.called)
def test_call_docx(self): """Test get text when docx given""" method = GetText() msg = QueueReqMessage() msg.params = {} msg.params['file_path'] = '/tmp/test.docx' method.application = self.mock_application method(msg) self.assertFalse(self.mock_application.logger.error.called) self.mock_docx.assert_called_with('/tmp/test.docx') self.assertTrue(self.mock_application.enitity.save.called)
def test_routing_call_fajna(self): application = MagicMock('application') application.send_response = MagicMock('application.send_response') router = Router(application) router.add_methods([{'class': MagicMock(return_value=self.method), 'name': 'fajna', 'original_name': 'fajna'}]) message = QueueReqMessage() message.method = 'fajna' message.params = {'key': 'test'} router.find_method(message) application.send_response.assert_called_with self.assertFalse(application.send_response.called) self.assertFalse(self.method.process.called) self.method.fajna.assert_called_with(key='test')
def test_user_not_exists(self): method = Login() self.mock_user.select.return_value = [] self.patch_user = patch('redsparrow.methods.front.User', self.mock_user).start() msg = QueueReqMessage() msg.params = {'login': '******', 'password': '******'} method.request = msg method.success = MagicMock() method.error = MagicMock() method.process(**msg.params) self.assertTrue(self.mock_user.select.called) self.assertFalse(method.success.called) self.assertTrue(method.error.called)
def test_user_already_exists(self): method = Register() msg = QueueReqMessage() msg.params = { 'login': '******', 'password': '******', 'surname': 'test', 'name': 'test', 'email': 'test1' } method.request = msg method.process(**msg.params) self.assertTrue(self.mock_user.select.called) self.assertFalse(self.mock_user.insert.called) self.assertTrue(self.mock_application.send_response.called)
def test_user_register(self): method = Register() self.mock_user.select.return_value = [] self.patch_user = patch('redsparrow.methods.front.User', self.mock_user).start() msg = QueueReqMessage() msg.params = { 'login': '******', 'password': '******', 'surname': 'test', 'name': 'test', 'email': 'test1' } method.request = msg method.process(**msg.params) self.assertTrue(self.mock_user.select.called) self.assertTrue(self.mock_user.called) self.assertTrue(self.mock_application.send_response.called)
def add_thesis(send): # msg = QueueReqMessage(id='1', params={'login': '******', 'password': '******'}, method='login') msg = QueueReqMessage() msg.id = 1 msg.method = 'thesismethods-add_thesis' # msg.params = {'thesis_name': 'test', 'user_id': 1, 'supervisor_id': 2, 'fos_id': 1,'filepath': '/home/aldor/workspace/RedSparrow/lecturer_database/Mariusz_Starzak-praca_magisterska.odt'} msg.params = { 'thesis_name': 'test', 'user_id': 1, 'supervisor_id': 2, 'fos_id': 1, 'filepath': '/home/aldor/workspace/RedSparrow/lecturer_database/webrtc - praca magisterska.docx' } send.on_recv(callback) send.send_string(str(msg))
def on_data(self, data): req = QueueReqMessage(json_data=data[0].decode("UTF-8")) try: self.router.find_method(req) except Exception as err: logging.error('Internal error {}, request {}'.format(err, req)) exc_type, exc_value, exc_traceback = sys.exc_info() traceback.print_exception(exc_type, exc_value, exc_traceback, limit=3, file=sys.stderr) response = QueueRepMessage(id=req.id) response.error = { "code": -32603, "message": "Internal error %s" % err } self.send_response(response)
def test_unique_Id(self): msg1 = QueueReqMessage() msg2 = QueueReqMessage() self.assertTrue(msg1.id != msg2.id)
def test_get_params(self): msg = QueueReqMessage(json_data="""{"params": "test"}""") self.assertEqual(msg.params, 'test')
def test_get_method(self): msg = QueueReqMessage(json_data="""{"method": "test"}""") self.assertEqual(msg.method, 'test')