mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-28 11:24:30 +01:00
1436092a35
This commit introduces the ChatViewModel, which manages the business logic for chat interactions associated with tasks. The ViewModel communicates with a mock data source, offering functionalities like fetching chats for a specific task and sending chat messages. In addition to the implementation, comprehensive tests for ChatViewModel have been provided to ensure its behavior is consistent with our design goals and expectations. Key Features: Chat management in ChatViewModel. Tests covering all major functionalities of ChatViewModel. Mock data source updates to emulate chat data interactions.
49 lines
1.9 KiB
Dart
49 lines
1.9 KiB
Dart
import 'package:auto_gpt_flutter_client/models/chat.dart';
|
|
import 'package:auto_gpt_flutter_client/models/message_type.dart';
|
|
import 'package:auto_gpt_flutter_client/viewmodels/chat_viewmodel.dart';
|
|
import 'package:flutter_test/flutter_test.dart';
|
|
|
|
void main() {
|
|
// Initialize the ChatViewModel
|
|
// TODO: Dependency injection in view models for testing purposes when we implement services
|
|
final viewModel = ChatViewModel();
|
|
|
|
group('ChatViewModel', () {
|
|
test('fetch chats for a specific task', () {
|
|
viewModel
|
|
.fetchChatsForTask(1); // Assuming task with ID 1 exists in mock data
|
|
expect(viewModel.chats.isNotEmpty, true);
|
|
expect(viewModel.chats.every((chat) => chat.taskId == 1), true);
|
|
});
|
|
|
|
test('send chat message for a specific task', () {
|
|
final initialChatsLength = viewModel.chats.length;
|
|
viewModel.sendChatMessage(1, 'Test message');
|
|
expect(viewModel.chats.length,
|
|
initialChatsLength + 2); // One user message and one agent reply
|
|
expect(viewModel.chats.last.messageType,
|
|
MessageType.agent); // Last message should be agent's reply
|
|
});
|
|
|
|
// TODO: Refactor to return errors when we implement service
|
|
test('fetch chats for invalid task id', () {
|
|
viewModel.fetchChatsForTask(
|
|
9999); // Assuming task with ID 9999 does not exist in mock data
|
|
expect(
|
|
viewModel.chats.where((chat) => chat.taskId == 9999).isEmpty, true);
|
|
});
|
|
|
|
// TODO: Refactor to return errors when we implement service
|
|
test('send chat message for invalid task id', () {
|
|
final initialChatsLength = viewModel.chats.length;
|
|
viewModel.sendChatMessage(9999, 'Invalid test message');
|
|
expect(
|
|
viewModel.chats.length,
|
|
initialChatsLength +
|
|
2); // Even for invalid tasks, we're currently adding mock replies
|
|
expect(viewModel.chats.last.messageType,
|
|
MessageType.agent); // Last message should be agent's reply
|
|
});
|
|
});
|
|
}
|