From aefa0033550a0de4ba37a5df537c6b26b64828eb Mon Sep 17 00:00:00 2001 From: 9seconds Date: Tue, 23 Mar 2021 21:26:57 +0300 Subject: [PATCH] Add tests for events --- mtglib/events_test.go | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/mtglib/events_test.go b/mtglib/events_test.go index f0e3975..133fce3 100644 --- a/mtglib/events_test.go +++ b/mtglib/events_test.go @@ -32,6 +32,28 @@ func (suite *EventsTestSuite) TestEventFinish() { suite.Equal("CONNID", evt.StreamID()) } +func (suite *EventsTestSuite) TestEventConnectedToDC() { + evt := mtglib.EventConnectedToDC{ + CreatedAt: time.Now(), + ConnID: "CONNID", + RemoteIP: net.ParseIP("10.0.0.10"), + DC: 3, + } + + suite.Equal("CONNID", evt.StreamID()) +} + +func (suite *EventsTestSuite) TestEventTraffic() { + evt := mtglib.EventTraffic{ + CreatedAt: time.Now(), + ConnID: "CONNID", + Traffic: 3, + IsRead: true, + } + + suite.Equal("CONNID", evt.StreamID()) +} + func (suite *EventsTestSuite) TestEventConcurrencyLimited() { suite.Empty(mtglib.EventConcurrencyLimited{}.StreamID()) }