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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
use axum::extract::State;
use axum_extra::extract::Query;
use futures::{future, stream::StreamExt as _};
use crate::test::fixtures::{self, future::Expect as _};
#[tokio::test]
async fn terminates_on_token_expiry() {
// Set up the environment
let app = fixtures::scratch_app().await;
let conversation = fixtures::conversation::create(&app, &fixtures::now()).await;
let sender = fixtures::user::create(&app, &fixtures::now()).await;
let resume_point = fixtures::boot::resume_point(&app).await;
// Subscribe via the endpoint
let subscriber_creds = fixtures::user::create_with_password(&app, &fixtures::now()).await;
let subscriber =
fixtures::identity::logged_in(&app, &subscriber_creds, &fixtures::ancient()).await;
let super::Response(events) = super::handler(
State(app.clone()),
subscriber,
None,
Query(super::QueryParams { resume_point }),
)
.await
.expect("subscribe never fails");
// Verify the resulting stream's behaviour
app.tokens()
.expire(&fixtures::now())
.await
.expect("expiring tokens succeeds");
// These should not be delivered.
let messages = [
fixtures::message::send(&app, &conversation, &sender, &fixtures::now()).await,
fixtures::message::send(&app, &conversation, &sender, &fixtures::now()).await,
fixtures::message::send(&app, &conversation, &sender, &fixtures::now()).await,
];
events
.filter_map(fixtures::event::stream::message)
.filter_map(fixtures::event::stream::message::sent)
.filter(|event| future::ready(messages.iter().any(|message| &event.message == message)))
.next()
.expect_none("end of stream")
.await;
}
#[tokio::test]
async fn terminates_on_logout() {
// Set up the environment
let app = fixtures::scratch_app().await;
let conversation = fixtures::conversation::create(&app, &fixtures::now()).await;
let sender = fixtures::user::create(&app, &fixtures::now()).await;
let resume_point = fixtures::boot::resume_point(&app).await;
// Subscribe via the endpoint
let subscriber = fixtures::identity::create(&app, &fixtures::now()).await;
let super::Response(events) = super::handler(
State(app.clone()),
subscriber.clone(),
None,
Query(super::QueryParams { resume_point }),
)
.await
.expect("subscribe never fails");
// Verify the resulting stream's behaviour
app.tokens()
.logout(&subscriber.token)
.await
.expect("expiring tokens succeeds");
// These should not be delivered.
let messages = [
fixtures::message::send(&app, &conversation, &sender, &fixtures::now()).await,
fixtures::message::send(&app, &conversation, &sender, &fixtures::now()).await,
fixtures::message::send(&app, &conversation, &sender, &fixtures::now()).await,
];
events
.filter_map(fixtures::event::stream::message)
.filter_map(fixtures::event::stream::message::sent)
.filter(|event| future::ready(messages.iter().any(|message| &event.message == message)))
.next()
.expect_none("end of stream")
.await;
}
#[tokio::test]
async fn terminates_on_password_change() {
// Set up the environment
let app = fixtures::scratch_app().await;
let conversation = fixtures::conversation::create(&app, &fixtures::now()).await;
let sender = fixtures::user::create(&app, &fixtures::now()).await;
let resume_point = fixtures::boot::resume_point(&app).await;
// Subscribe via the endpoint
let creds = fixtures::user::create_with_password(&app, &fixtures::now()).await;
let cookie = fixtures::cookie::logged_in(&app, &creds, &fixtures::now()).await;
let subscriber = fixtures::identity::from_cookie(&app, &cookie, &fixtures::now()).await;
let super::Response(events) = super::handler(
State(app.clone()),
subscriber.clone(),
None,
Query(super::QueryParams { resume_point }),
)
.await
.expect("subscribe never fails");
// Verify the resulting stream's behaviour
let (_, password) = creds;
let to = fixtures::user::propose_password();
app.tokens()
.change_password(&subscriber.user, &password, &to, &fixtures::now())
.await
.expect("expiring tokens succeeds");
// These should not be delivered.
let messages = [
fixtures::message::send(&app, &conversation, &sender, &fixtures::now()).await,
fixtures::message::send(&app, &conversation, &sender, &fixtures::now()).await,
fixtures::message::send(&app, &conversation, &sender, &fixtures::now()).await,
];
events
.filter_map(fixtures::event::stream::message)
.filter_map(fixtures::event::stream::message::sent)
.filter(|event| future::ready(messages.iter().any(|message| &event.message == message)))
.next()
.expect_none("end of stream")
.await;
}
|