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 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 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')