Added an extra nil checking

server/router.go

@@ -111,6 +111,10 @@ func (router *router) Stop() error {
 }
 
 func (router *router) Check() error {
+	if router.messageStore == nil || router.kvStore == nil {
+		return ErrServiceNotProvided
+	}
+
 	err := router.messageStore.Check()
 	if err != nil {
 		return err

server/router_test.go

@@ -397,6 +397,10 @@ func TestRouter_Check(t *testing.T) {
 	// Given a Multiplexer with route
 	router, _ := aRouterRoute(1)
 
+	//first the messageStore and kvStore will be nil
+	err := router.Check()
+	a.NotNil(err)
+
 	// mock messageStore
 	msMock := NewMockMessageStore(ctrl)
 	router.messageStore = msMock
@@ -408,7 +412,7 @@ func TestRouter_Check(t *testing.T) {
 	msMock.EXPECT().Check().Return(nil)
 	mockKvStore.EXPECT().Check().Return(nil)
 	//both router health checks will work
-	err := router.Check()
+	err = router.Check()
 	a.Nil(err)
 
 	//router messageStore will return error

store/sqlite.go

@@ -7,6 +7,7 @@ import (
 	_ "github.com/mattn/go-sqlite3"
 
 	"fmt"
+	"github.com/vektra/errors"
 	"io/ioutil"
 	"os"
 	"path"
@@ -164,6 +165,10 @@ func (kvStore *SqliteKVStore) Open() error {
 }
 
 func (kvStore *SqliteKVStore) Check() error {
+	if kvStore.db == nil {
+		protocol.Err("Db pointer is not initialized")
+		return errors.New("Db service is null.")
+	}
 
 	if err := kvStore.db.DB().Ping(); err != nil {
 		protocol.Err("error pinging database %q: %q", kvStore.filename, err.Error())