Go version changed

[ric-plt/sdlgo.git] / internal / sdlgoredis / dbstate_test.go

diff --git a/internal/sdlgoredis/dbstate_test.go b/internal/sdlgoredis/dbstate_test.go
index 1727bbc..9a3a5e9 100644 (file)
--- a/internal/sdlgoredis/dbstate_test.go
+++ b/internal/sdlgoredis/dbstate_test.go
@@ -37,6 +37,10 @@ func setupDbState() *dbStateMock {
        return new(dbStateMock)
 }
 
+func (ds *dbStateMock) setError(err error) {
+       ds.state.Err = err
+}
+
 func (ds *dbStateMock) setPrimaryError(err error) {
        ds.state.PrimaryDbState.Err = err
 }
@@ -86,6 +90,14 @@ func (ds *dbStateMock) addSentinelFields(ip, port, flags string) {
        ds.state.SentinelsDbState.States = append(ds.state.SentinelsDbState.States, newState)
 }
 
+func TestIsOnlineFailureIfErrorHasSet(t *testing.T) {
+       testErr := errors.New("Some error")
+       st := setupDbState()
+       st.setError(testErr)
+       err := st.state.IsOnline()
+       assert.Equal(t, testErr, err)
+}
+
 func TestIsOnlineWhenSinglePrimarySuccessfully(t *testing.T) {
        st := setupDbState()
        st.setPrimaryFields("master", "1.2.3.4", "60000", "0", "master")