AutoGPT

1
import 'package:auto_gpt_flutter_client/models/chat.dart';
2
import 'package:auto_gpt_flutter_client/models/message_type.dart';
3
import 'package:flutter_test/flutter_test.dart';
4

5
void main() {
6
  group('Chat', () {
7
    // Test the properties of the Chat class
8
    test('Chat properties', () {
9
      final chat = Chat(
10
          id: 1,
11
          taskId: 1,
12
          message: 'Test Message',
13
          timestamp: DateTime.now(),
14
          messageType: MessageType.user);
15

16
      expect(chat.id, 1);
17
      expect(chat.taskId, 1);
18
      expect(chat.message, 'Test Message');
19
      expect(chat.messageType, MessageType.user);
20
    });
21

22
    // Test Chat.fromMap method
23
    test('Chat.fromMap', () {
24
      final chat = Chat.fromMap({
25
        'id': 1,
26
        'taskId': 1,
27
        'message': 'Test Message',
28
        'timestamp': DateTime.now().toString(),
29
        'messageType': 'user'
30
      });
31

32
      expect(chat.id, 1);
33
      expect(chat.taskId, 1);
34
      expect(chat.message, 'Test Message');
35
      expect(chat.messageType, MessageType.user);
36
    });
37

38
    // Test that two Chat objects with the same properties are equal
39
    test('Two chats with same properties are equal', () {
40
      final chat1 = Chat(
41
          id: 3,
42
          taskId: 3,
43
          message: 'Same Message',
44
          timestamp: DateTime.now(),
45
          messageType: MessageType.agent);
46
      final chat2 = Chat(
47
          id: 3,
48
          taskId: 3,
49
          message: 'Same Message',
50
          timestamp: chat1.timestamp,
51
          messageType: MessageType.agent);
52

53
      expect(chat1, chat2);
54
    });
55

56
    // Test that toString() returns a string representation of the Chat
57
    test('toString returns string representation', () {
58
      final chat = Chat(
59
          id: 4,
60
          taskId: 4,
61
          message: 'Test toString',
62
          timestamp: DateTime.now(),
63
          messageType: MessageType.user);
64

65
      expect(chat.toString(),
66
          'Chat(id: 4, taskId: 4, message: Test toString, timestamp: ${chat.timestamp}, messageType: MessageType.user)');
67
    });
68
  });
69
}
70