60 lines
2.0 KiB
Dart
60 lines
2.0 KiB
Dart
import 'package:flutter_test/flutter_test.dart';
|
|
|
|
import 'package:flutter_app/core/network/api_repository.dart';
|
|
import 'package:flutter_app/features/chat/presentation/cubit/chat_cubit.dart';
|
|
import 'package:flutter_app/features/chat/presentation/cubit/chat_state.dart';
|
|
|
|
class _VersionTestApi extends ApiRepository {
|
|
_VersionTestApi() : super(baseUrl: 'http://127.0.0.1:8000');
|
|
|
|
final responses = <List<String>>[
|
|
['الجواب الأول'],
|
|
['الجواب الثاني'],
|
|
];
|
|
SavedConversation? lastSaved;
|
|
|
|
@override
|
|
Future<String> getModelName() async => 'gemma4:e2b';
|
|
|
|
@override
|
|
Future<List<String>> getAvailableModels() async => ['gemma4:e2b'];
|
|
|
|
@override
|
|
Future<List<SavedConversation>> listConversations() async => [];
|
|
|
|
@override
|
|
Future<void> saveConversation(SavedConversation conversation) async {
|
|
lastSaved = conversation;
|
|
}
|
|
|
|
@override
|
|
Stream<String> completeStream(
|
|
List<ChatMessage> messages, {
|
|
String? model,
|
|
}) => Stream<String>.fromIterable(responses.removeAt(0));
|
|
}
|
|
|
|
void main() {
|
|
test('regeneration keeps prior answers and selection is persisted', () async {
|
|
final api = _VersionTestApi();
|
|
final cubit = ChatCubit(api);
|
|
await Future<void>.delayed(Duration.zero);
|
|
|
|
await cubit.send('ما معنى API؟');
|
|
expect(cubit.state.messages.last.answerVersions, ['الجواب الأول']);
|
|
|
|
await cubit.regenerateLastAnswer();
|
|
expect(cubit.state.messages.last.text, 'الجواب الثاني');
|
|
expect(cubit.state.messages.last.answerVersions, ['الجواب الأول', 'الجواب الثاني']);
|
|
expect(cubit.state.messages.last.selectedVersion, 1);
|
|
|
|
await cubit.selectAnswerVersion(1, 0);
|
|
expect(cubit.state.messages.last.text, 'الجواب الأول');
|
|
expect(cubit.state.messages.last.selectedVersion, 0);
|
|
expect(api.lastSaved?.messages.last.answerVersions, ['الجواب الأول', 'الجواب الثاني']);
|
|
expect(api.lastSaved?.messages.last.selectedVersion, 0);
|
|
|
|
await cubit.close();
|
|
});
|
|
}
|