copilot-sdk/rust/tests/e2e/rpc_schedule.rs at copilot/issue-1884-java-docs · github/copilot-sdk · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
use github_copilot_sdk::rpc::ScheduleStopRequest;

use super::support::with_e2e_context;

#[tokio::test]
async fn should_list_no_schedules_for_fresh_session() {
    with_e2e_context(
        "rpc_schedule",
        "should_list_no_schedules_for_fresh_session",
        |ctx| {
            Box::pin(async move {
                ctx.set_default_copilot_user();
                let client = ctx.start_client().await;
                let session = client
                    .create_session(ctx.approve_all_session_config())
                    .await
                    .expect("create session");

                let schedules = session
                    .rpc()
                    .schedule()
                    .list()
                    .await
                    .expect("list schedules");
                assert!(schedules.entries.is_empty());

                session.disconnect().await.expect("disconnect session");
                client.stop().await.expect("stop client");
            })
        },
    )
    .await;
}

#[tokio::test]
async fn should_return_null_entry_when_stopping_unknown_schedule() {
    with_e2e_context(
        "rpc_schedule",
        "should_return_null_entry_when_stopping_unknown_schedule",
        |ctx| {
            Box::pin(async move {
                ctx.set_default_copilot_user();
                let client = ctx.start_client().await;
                let session = client
                    .create_session(ctx.approve_all_session_config())
                    .await
                    .expect("create session");

                let stopped = session
                    .rpc()
                    .schedule()
                    .stop(ScheduleStopRequest { id: i64::MAX })
                    .await
                    .expect("stop missing schedule");
                assert!(stopped.entry.is_none());
                assert!(
                    session
                        .rpc()
                        .schedule()
                        .list()
                        .await
                        .expect("list schedules")
                        .entries
                        .is_empty()
                );

                session.disconnect().await.expect("disconnect session");
                client.stop().await.expect("stop client");
            })
        },
    )
    .await;
}