106 lines
2.9 KiB
Go
106 lines
2.9 KiB
Go
package integration
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gambol99/go-marathon"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
"github.com/traefik/traefik/v2/integration/try"
|
|
)
|
|
|
|
const containerNameMarathon = "marathon"
|
|
|
|
// Marathon test suites.
|
|
type MarathonSuite struct {
|
|
BaseSuite
|
|
marathonURL string
|
|
}
|
|
|
|
func TestMarathonSuite(t *testing.T) {
|
|
suite.Run(t, new(MarathonSuite))
|
|
}
|
|
|
|
func (s *MarathonSuite) SetUpSuite() {
|
|
s.BaseSuite.SetupSuite()
|
|
s.createComposeProject("marathon")
|
|
s.composeUp()
|
|
|
|
s.marathonURL = "http://" + s.getComposeServiceIP(containerNameMarathon) + ":8080"
|
|
|
|
// Wait for Marathon readiness prior to creating the client so that we
|
|
// don't run into the "all cluster members down" state right from the
|
|
// start.
|
|
err := try.GetRequest(s.marathonURL+"/v2/leader", 1*time.Minute, try.StatusCodeIs(http.StatusOK))
|
|
require.NoError(s.T(), err)
|
|
}
|
|
|
|
func (s *MarathonSuite) TearDownSuite() {
|
|
s.BaseSuite.TearDownSuite()
|
|
}
|
|
|
|
func (s *BaseSuite) deployApplication(client marathon.Marathon, application *marathon.Application) {
|
|
deploy, err := client.UpdateApplication(application, false)
|
|
require.NoError(s.T(), err)
|
|
// Wait for deployment to complete.
|
|
err = client.WaitOnDeployment(deploy.DeploymentID, 1*time.Minute)
|
|
require.NoError(s.T(), err)
|
|
}
|
|
|
|
func (s *MarathonSuite) TestConfigurationUpdate() {
|
|
s.T().Skip("doesn't work")
|
|
|
|
// Start Traefik.
|
|
file := s.adaptFile("fixtures/marathon/simple.toml", struct {
|
|
MarathonURL string
|
|
}{s.marathonURL})
|
|
|
|
s.traefikCmd(withConfigFile(file))
|
|
|
|
// Wait for Traefik to turn ready.
|
|
err := try.GetRequest("http://127.0.0.1:8000/", 2*time.Second, try.StatusCodeIs(http.StatusNotFound))
|
|
require.NoError(s.T(), err)
|
|
|
|
// Prepare Marathon client.
|
|
config := marathon.NewDefaultConfig()
|
|
config.URL = s.marathonURL
|
|
client, err := marathon.NewClient(config)
|
|
require.NoError(s.T(), err)
|
|
|
|
// Create test application to be deployed.
|
|
app := marathon.NewDockerApplication().
|
|
Name("/whoami").
|
|
CPU(0.1).
|
|
Memory(32).
|
|
AddLabel("traefik.http.Routers.rt.Rule", "PathPrefix(`/service`)")
|
|
app.Container.Docker.Bridged().
|
|
Expose(80).
|
|
Container("traefik/whoami")
|
|
|
|
// Deploy the test application.
|
|
s.deployApplication(client, app)
|
|
|
|
// Query application via Traefik.
|
|
err = try.GetRequest("http://127.0.0.1:8000/service", 30*time.Second, try.StatusCodeIs(http.StatusOK))
|
|
require.NoError(s.T(), err)
|
|
|
|
// Create test application with services to be deployed.
|
|
app = marathon.NewDockerApplication().
|
|
Name("/whoami").
|
|
CPU(0.1).
|
|
Memory(32).
|
|
AddLabel("traefik.http.Routers.app.Rule", "PathPrefix(`/app`)")
|
|
app.Container.Docker.Bridged().
|
|
Expose(80).
|
|
Container("traefik/whoami")
|
|
|
|
// Deploy the test application.
|
|
s.deployApplication(client, app)
|
|
|
|
// Query application via Traefik.
|
|
err = try.GetRequest("http://127.0.0.1:8000/app", 30*time.Second, try.StatusCodeIs(http.StatusOK))
|
|
require.NoError(s.T(), err)
|
|
}
|