Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -21,4 +21,4 @@ def embedDocuments(self, documentsIds: List[str]) -> List[DocumentOperationRespo
Returns:
List[DocumentOperationResponse]: the response of the operation.
"""
return self.useCase.embedDocuemnts([DocumentId(documentId) for documentId in documentsIds])
return self.useCase.embedDocuments([DocumentId(documentId) for documentId in documentsIds])
Original file line number Diff line number Diff line change
Expand Up @@ -3,5 +3,5 @@
from domain.document.document_operation_response import DocumentOperationResponse

class EmbedDocumentsUseCase:
def embedDocuemnts(self, documentsIds: List[DocumentId]) -> List[DocumentOperationResponse]:
def embedDocuments(self, documentsIds: List[DocumentId]) -> List[DocumentOperationResponse]:
pass
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ def __init__(self, getDocumentsContent: GetDocumentsContent, embeddingsUploader:
self.embeddingsUploader = embeddingsUploader
self.getDocumentStatus = getDocumentStatus

def embedDocuemnts(self, documentsIds: List[DocumentId]) -> List[DocumentOperationResponse]:
def embedDocuments(self, documentsIds: List[DocumentId]) -> List[DocumentOperationResponse]:
verifiedDocumentsIds =[]
verifiedDocumentsStatus = []
for documentId, documentStatus in zip(documentsIds, self.getDocumentStatus.getDocumentsStatus(documentsIds)):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,7 @@ def test_askChatbot_with_existent_chat(mocker):
askChatbotController = AskChatbotController(useCaseMock)

with unittest.mock.patch('adapter._in.web.ask_chatbot_controller.ChatId') as MockChatId, \
unittest.mock.patch('adapter._in.web.ask_chatbot_controller.Message') as MockMessage:

unittest.mock.patch('adapter._in.web.ask_chatbot_controller.Message') as MockMessage:
MockChatId.return_value = ChatId(1)
MockMessage.return_value = Message("response", unittest.mock.ANY, None, MessageSender.CHATBOT)

Expand All @@ -28,8 +27,8 @@ def test_askChatbot_with_existent_chat(mocker):
None,
MessageSender.USER
)
assert isinstance(response, MessageResponse)

assert isinstance(response, MessageResponse)

def test_askChatbot_without_chat(mocker):
useCaseMock = mocker.Mock()
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
from adapter._in.web.change_configuration_controller import ChangeConfigurationController

from domain.configuration.configuration_operation_response import ConfigurationOperationResponse

def test_changeLLMModel_with_existent_model(mocker):
useCaseMock = mocker.Mock()
useCaseMock.changeLLMModel.return_value = ConfigurationOperationResponse(True, "Model changed successfully")

changeConfigurationController = ChangeConfigurationController(useCaseMock)

response = changeConfigurationController.changeLLMModel("OPENAI")

assert isinstance(response, ConfigurationOperationResponse)

def test_changeLLMModel_with_absent_model(mocker):
useCaseMock = mocker.Mock()
useCaseMock.changeLLMModel.return_value = ConfigurationOperationResponse(True, "Model changed successfully")

changeConfigurationController = ChangeConfigurationController(useCaseMock)

response = changeConfigurationController.changeLLMModel("NONEXISTENT_MODEL")

assert response is None
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
import unittest
from adapter._in.web.conceal_documents_controller import ConcealDocumentsController
from domain.document.document_operation_response import DocumentOperationResponse
from domain.document.document_id import DocumentId

def test_concealDocuments(mocker):
useCaseMock = mocker.Mock()
useCaseMock.concealDocuments.return_value = [DocumentOperationResponse(DocumentId("Prova.pdf"), True, "Document concealed successfully")]

with unittest.mock.patch("adapter._in.web.conceal_documents_controller.DocumentId") as MockDocumentId:
MockDocumentId.return_value = DocumentId("Prova.pdf")

concealDocumentsController = ConcealDocumentsController(useCaseMock)

response = concealDocumentsController.concealDocuments(["Prova.pdf"])

MockDocumentId.assert_called_once_with("Prova.pdf")

assert isinstance(response[0], DocumentOperationResponse)
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
import unittest
from adapter._in.web.delete_chats_controller import DeleteChatsController
from domain.chat.chat_operation_response import ChatOperationResponse
from domain.chat.chat_id import ChatId

def test_deleteChats(mocker):
useCaseMock = mocker.Mock()
useCaseMock.deleteChats.return_value = [ChatOperationResponse(True, "Chat deleted successfully", ChatId(1))]

with unittest.mock.patch("adapter._in.web.delete_chats_controller.ChatId") as MockChatId:
MockChatId.return_value = ChatId(1)

deleteChatsController = DeleteChatsController(useCaseMock)

response = deleteChatsController.deleteChats([1])

MockChatId.assert_called_once_with(1)

assert isinstance(response[0], ChatOperationResponse)
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
import unittest
from adapter._in.web.delete_documents_controller import DeleteDocumentsController
from domain.document.document_operation_response import DocumentOperationResponse
from domain.document.document_id import DocumentId

def test_deleteDocuments(mocker):
useCaseMock = mocker.Mock()
useCaseMock.deleteDocuments.return_value = [DocumentOperationResponse(DocumentId("Prova.pdf"), True, "Document deleted successfully")]

with unittest.mock.patch("adapter._in.web.delete_documents_controller.DocumentId") as MockDocumentId:
MockDocumentId.return_value = DocumentId("Prova.pdf")

deleteDocumentsController = DeleteDocumentsController(useCaseMock)

response = deleteDocumentsController.deleteDocuments(["Prova.pdf"])

MockDocumentId.assert_called_once_with("Prova.pdf")

assert isinstance(response[0], DocumentOperationResponse)
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
import unittest
from adapter._in.web.embed_documents_controller import EmbedDocumentsController
from domain.document.document_operation_response import DocumentOperationResponse
from domain.document.document_id import DocumentId

def test_embedDocuments(mocker):
useCaseMock = mocker.Mock()
useCaseMock.embedDocuments.return_value = [DocumentOperationResponse(DocumentId("Prova.pdf"), True, "Document embedded successfully")]

with unittest.mock.patch("adapter._in.web.embed_documents_controller.DocumentId") as MockDocumentId:
MockDocumentId.return_value = DocumentId("Prova.pdf")

embedDocumentsController = EmbedDocumentsController(useCaseMock)

response = embedDocumentsController.embedDocuments(["Prova.pdf"])

MockDocumentId.assert_called_once_with("Prova.pdf")

assert isinstance(response[0], DocumentOperationResponse)
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
import unittest
from adapter._in.web.enable_documents_controller import EnableDocumentsController
from domain.document.document_operation_response import DocumentOperationResponse
from domain.document.document_id import DocumentId

def test_enableDocuments(mocker):
useCaseMock = mocker.Mock()
useCaseMock.enableDocuments.return_value = [DocumentOperationResponse(DocumentId("Prova.pdf"), True, "Document enabled successfully")]

with unittest.mock.patch("adapter._in.web.enable_documents_controller.DocumentId") as MockDocumentId:
MockDocumentId.return_value = DocumentId("Prova.pdf")

enableDocumentsController = EnableDocumentsController(useCaseMock)

response = enableDocumentsController.enableDocuments(["Prova.pdf"])

MockDocumentId.assert_called_once_with("Prova.pdf")

assert isinstance(response[0], DocumentOperationResponse)
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
import unittest
from adapter._in.web.get_chat_messages_controller import GetChatMessagesController
from domain.chat.chat import Chat
from domain.chat.chat_id import ChatId
from domain.chat.message import Message, MessageSender

def test_getChatMessages(mocker):
useCaseMock = mocker.Mock()
useCaseMock.getChatMessages.return_value = Chat("Prova", ChatId(1), [Message("message", unittest.mock.ANY, None, MessageSender.USER)])

with unittest.mock.patch("adapter._in.web.get_chat_messages_controller.ChatId") as MockChatId:
MockChatId.return_value = 1

getChatMessagesController = GetChatMessagesController(useCaseMock)

response = getChatMessagesController.getChatMessages(1)

MockChatId.assert_called_once_with(1)

assert isinstance(response, Chat)