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())
 }