4
4
import com .theokanning .openai .messages .MessageRequest ;
5
5
import com .theokanning .openai .threads .Thread ;
6
6
import com .theokanning .openai .threads .ThreadRequest ;
7
+ import org .junit .jupiter .api .MethodOrderer ;
8
+ import org .junit .jupiter .api .Order ;
7
9
import org .junit .jupiter .api .Test ;
10
+ import org .junit .jupiter .api .TestMethodOrder ;
8
11
9
12
import java .util .*;
10
13
11
14
import static org .junit .jupiter .api .Assertions .*;
12
15
13
-
16
+ @ TestMethodOrder ( MethodOrderer . OrderAnnotation . class )
14
17
public class ThreadTest {
15
18
16
19
String token = System .getenv ("OPENAI_TOKEN" );
17
20
OpenAiService service = new OpenAiService (token );
18
21
22
+ static String threadId ;
23
+
19
24
@ Test
25
+ @ Order (1 )
20
26
void createThread () {
21
27
MessageRequest messageRequest = MessageRequest .builder ()
22
28
.content ("Hello" )
@@ -27,34 +33,34 @@ void createThread() {
27
33
.build ();
28
34
29
35
Thread thread = service .createThread (threadRequest );
30
- System . out . println ( thread .getId () );
36
+ threadId = thread .getId ();
31
37
assertEquals ("thread" , thread .getObject ());
32
38
}
33
39
34
40
@ Test
41
+ @ Order (2 )
35
42
void retrieveThread () {
36
- String threadId = "thread_K82pTg9kmhxpplGqalW6IHlc" ;
37
-
38
43
Thread thread = service .retrieveThread (threadId );
39
44
System .out .println (thread .getMetadata ());
40
45
assertEquals ("thread" , thread .getObject ());
41
46
}
42
47
43
48
@ Test
49
+ @ Order (3 )
44
50
void modifyThread () {
45
- String threadId = "thread_K82pTg9kmhxpplGqalW6IHlc" ;
46
51
Map <String , String > metadata = new HashMap <>();
47
52
metadata .put ("action" , "modify" );
48
53
ThreadRequest threadRequest = ThreadRequest .builder ()
49
54
.metadata (metadata )
50
55
.build ();
51
56
Thread thread = service .modifyThread (threadId , threadRequest );
52
57
assertEquals ("thread" , thread .getObject ());
58
+ assertEquals ("modify" , thread .getMetadata ().get ("action" ));
53
59
}
54
60
55
61
@ Test
62
+ @ Order (4 )
56
63
void deleteThread () {
57
- String threadId = "thread_K82pTg9kmhxpplGqalW6IHlc" ;
58
64
DeleteResult deleteResult = service .deleteThread (threadId );
59
65
assertEquals ("thread.deleted" , deleteResult .getObject ());
60
66
}
0 commit comments