2018-01-26 18:22:03 +01:00
package integration
import (
"net/http"
"os"
"time"
2019-08-03 03:58:23 +02:00
"github.com/containous/traefik/v2/integration/try"
2018-01-26 18:22:03 +01:00
"github.com/go-check/check"
2018-08-29 11:58:03 +02:00
"github.com/gorilla/websocket"
2018-01-26 18:22:03 +01:00
checker "github.com/vdemeester/shakers"
)
type RetrySuite struct { BaseSuite }
func ( s * RetrySuite ) SetUpSuite ( c * check . C ) {
s . createComposeProject ( c , "retry" )
s . composeProject . Start ( c )
}
func ( s * RetrySuite ) TestRetry ( c * check . C ) {
whoamiEndpoint := s . composeProject . Container ( c , "whoami" ) . NetworkSettings . IPAddress
file := s . adaptFile ( c , "fixtures/retry/simple.toml" , struct {
WhoamiEndpoint string
} { whoamiEndpoint } )
defer os . Remove ( file )
cmd , display := s . traefikCmd ( withConfigFile ( file ) )
defer display ( c )
err := cmd . Start ( )
c . Assert ( err , checker . IsNil )
defer cmd . Process . Kill ( )
2019-05-16 10:58:06 +02:00
err = try . GetRequest ( "http://127.0.0.1:8080/api/rawdata" , 60 * time . Second , try . BodyContains ( "PathPrefix(`/`)" ) )
2018-01-26 18:22:03 +01:00
c . Assert ( err , checker . IsNil )
// This simulates a DialTimeout when connecting to the backend server.
response , err := http . Get ( "http://127.0.0.1:8000/" )
c . Assert ( err , checker . IsNil )
c . Assert ( response . StatusCode , checker . Equals , http . StatusOK )
}
2018-08-29 11:58:03 +02:00
func ( s * RetrySuite ) TestRetryWebsocket ( c * check . C ) {
whoamiEndpoint := s . composeProject . Container ( c , "whoami" ) . NetworkSettings . IPAddress
file := s . adaptFile ( c , "fixtures/retry/simple.toml" , struct {
WhoamiEndpoint string
} { whoamiEndpoint } )
defer os . Remove ( file )
cmd , display := s . traefikCmd ( withConfigFile ( file ) )
defer display ( c )
err := cmd . Start ( )
c . Assert ( err , checker . IsNil )
defer cmd . Process . Kill ( )
2019-05-16 10:58:06 +02:00
err = try . GetRequest ( "http://127.0.0.1:8080/api/rawdata" , 60 * time . Second , try . BodyContains ( "PathPrefix(`/`)" ) )
2018-08-29 11:58:03 +02:00
c . Assert ( err , checker . IsNil )
// This simulates a DialTimeout when connecting to the backend server.
_ , response , err := websocket . DefaultDialer . Dial ( "ws://127.0.0.1:8000/echo" , nil )
c . Assert ( err , checker . IsNil )
c . Assert ( response . StatusCode , checker . Equals , http . StatusSwitchingProtocols )
_ , response , err = websocket . DefaultDialer . Dial ( "ws://127.0.0.1:8000/echo" , nil )
c . Assert ( err , checker . IsNil )
c . Assert ( response . StatusCode , checker . Equals , http . StatusSwitchingProtocols )
}