add helper methods to pitaya package

app.go

@@ -232,6 +232,21 @@ func SetSerializer(seri serialize.Serializer) {
 	app.serializer = seri
 }
 
+// GetSerializer gets the app serializer
+func GetSerializer() serialize.Serializer {
+	return app.serializer
+}
+
+// GetServer returns the server with the specified id
+func GetServer(id string) (*cluster.Server, error) {
+	return app.serviceDiscovery.GetServer(id)
+}
+
+// GetServersByType get all servers of type
+func GetServersByType(t string) (map[string]*cluster.Server, error) {
+	return app.serviceDiscovery.GetServersByType(t)
+}
+
 // AddMetricsReporter to be used
 func AddMetricsReporter(mr metrics.Reporter) {
 	app.metricsReporters = append(app.metricsReporters, mr)

app_test.go

@@ -168,6 +168,30 @@ func TestGetDieChan(t *testing.T) {
 	assert.Equal(t, app.dieChan, GetDieChan())
 }
 
+func TestGetSerializer(t *testing.T) {
+	assert.Equal(t, app.serializer, GetSerializer())
+}
+
+func TestGetServer(t *testing.T) {
+	r, err := cluster.NewEtcdServiceDiscovery(app.config, app.server)
+	assert.NoError(t, err)
+	assert.NotNil(t, r)
+	SetServiceDiscoveryClient(r)
+	s, err := GetServer("id")
+	assert.Nil(t, s)
+	assert.EqualError(t, constants.ErrNoServerWithID, err.Error())
+}
+
+func TestGetServersByType(t *testing.T) {
+	r, err := cluster.NewEtcdServiceDiscovery(app.config, app.server)
+	assert.NoError(t, err)
+	assert.NotNil(t, r)
+	SetServiceDiscoveryClient(r)
+	s, err := GetServersByType("id")
+	assert.Nil(t, s)
+	assert.EqualError(t, constants.ErrNoServersAvailableOfType, err.Error())
+}
+
 func TestSetHeartbeatInterval(t *testing.T) {
 	inter := 35 * time.Millisecond
 	SetHeartbeatTime(inter)