mirror of
https://github.com/ScuroNeko/mtg.git
synced 2026-08-31 21:44:03 +03:00
Rework stats
This commit is contained in:
@@ -90,8 +90,8 @@ func (suite *EventStreamTestSuite) TestEventConnectedToDC() {
|
||||
time.Sleep(100 * time.Millisecond)
|
||||
}
|
||||
|
||||
func (suite *EventStreamTestSuite) TestEventTraffic() {
|
||||
evt := mtglib.EventTraffic{
|
||||
func (suite *EventStreamTestSuite) TestEventTelegramTraffic() {
|
||||
evt := mtglib.EventTelegramTraffic{
|
||||
CreatedAt: time.Now(),
|
||||
ConnID: "connID",
|
||||
Traffic: 1024,
|
||||
@@ -100,10 +100,10 @@ func (suite *EventStreamTestSuite) TestEventTraffic() {
|
||||
|
||||
for _, v := range []*ObserverMock{suite.observerMock1, suite.observerMock2} {
|
||||
v.
|
||||
On("EventTraffic", mock.Anything).
|
||||
On("EventTelegramTraffic", mock.Anything).
|
||||
Once().
|
||||
Run(func(args mock.Arguments) {
|
||||
caught := args.Get(0).(mtglib.EventTraffic)
|
||||
caught := args.Get(0).(mtglib.EventTelegramTraffic)
|
||||
|
||||
suite.Equal(evt.CreatedAt, caught.CreatedAt)
|
||||
suite.Equal(evt.ConnID, caught.ConnID)
|
||||
|
||||
Reference in New Issue
Block a user