added types function

packet/type.go

@@ -27,6 +27,12 @@ const (
 	DISCONNECT
 )
 
+// Types returns a list of all known packet types.
+func Types() []Type {
+	return []Type{CONNECT, CONNACK, PUBLISH, PUBACK, PUBREC, PUBREL, PUBCOMP,
+		SUBSCRIBE, SUBACK, UNSUBSCRIBE, UNSUBACK, PINGREQ, PINGRESP, DISCONNECT}
+}
+
 // String returns the type as a string.
 func (t Type) String() string {
 	switch t {

packet/type_test.go

@@ -1,10 +1,15 @@
 package packet
 
 import (
-	"github.com/stretchr/testify/assert"
 	"testing"
+
+	"github.com/stretchr/testify/assert"
 )
 
+func TestTypes(t *testing.T) {
+	assert.Len(t, Types(), 14)
+}
+
 func TestTypeString(t *testing.T) {
 	assert.Equal(t, "Unknown", Type(99).String())
 }
@@ -14,24 +19,7 @@ func TestTypeValid(t *testing.T) {
 }
 
 func TestTypeNew(t *testing.T) {
-	list := []Type{
-		CONNECT,
-		CONNACK,
-		PUBLISH,
-		PUBACK,
-		PUBREC,
-		PUBREL,
-		PUBCOMP,
-		SUBSCRIBE,
-		SUBACK,
-		UNSUBSCRIBE,
-		UNSUBACK,
-		PINGREQ,
-		PINGRESP,
-		DISCONNECT,
-	}
-
-	for _, tt := range list {
+	for _, tt := range Types() {
 		m, err := tt.New()
 		assert.NotNil(t, m)
 		assert.NoError(t, err)