ScuroNeko/mtg
1
0
Fork 0
mirror of https://github.com/ScuroNeko/mtg.git synced 2026-09-01 12:54:02 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add EventReplayAttack

Browse Source
This commit is contained in:
9seconds
2021-03-29 12:07:18 +03:00
parent bef14bd009
commit daa8b9c798
14 changed files with 114 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
events/event_stream_test.go
+22
View File
@@ -206,6 +206,28 @@ func (suite *EventStreamTestSuite) TestEventIPBlocklisted() {
time.Sleep(100 * time.Millisecond)
}
func (suite *EventStreamTestSuite) TestEventReplayAttack() {
evt := mtglib.EventReplayAttack{
CreatedAt: time.Now(),
ConnID: "CONNID",
}
for _, v := range []*ObserverMock{suite.observerMock1, suite.observerMock2} {
v.
On("EventReplayAttack", mock.Anything).
Once().
Run(func(args mock.Arguments) {
caught := args.Get(0).(mtglib.EventReplayAttack)
suite.Equal(evt.CreatedAt, caught.CreatedAt)
suite.Equal(evt.StreamID(), caught.StreamID())
})
}
suite.stream.Send(suite.ctx, evt)
time.Sleep(100 * time.Millisecond)
}
func (suite *EventStreamTestSuite) TearDownTest() {
suite.stream.Shutdown()
suite.ctxCancel()