Update the BaseURLs of CSAPI clients when restarting a deployment

internal/docker/deployer.go

@@ -198,6 +198,12 @@ func (d *Deployer) Restart(hsDep *HomeserverDeployment, cfg *config.Complement)
 	}
 
 	// Wait for the container to be ready.
+	baseURL, fedBaseURL, err := waitForPorts(ctx, d.Docker, hsDep.ContainerID)
+	if err != nil {
+		return fmt.Errorf("Restart: Failed to restart container %s: %s", hsDep.ContainerID, err)
+	}
+	hsDep.SetEndpoints(baseURL, fedBaseURL)
+
 	stopTime := time.Now().Add(cfg.SpawnHSTimeout)
 	_, err = waitForContainer(ctx, d.Docker, hsDep, stopTime)
 	if err != nil {

internal/docker/deployment.go

@@ -31,6 +31,16 @@ type HomeserverDeployment struct {
 	CSAPIClients        []*client.CSAPI
 }
 
+// Updates the client and federation base URLs of the homeserver deployment.
+func (hsDep *HomeserverDeployment) SetEndpoints(baseURL string, fedBaseURL string) {
+	hsDep.BaseURL = baseURL
+	hsDep.FedBaseURL = fedBaseURL
+
+	for _, client := range hsDep.CSAPIClients {
+		client.BaseURL = baseURL
+	}
+}
+
 // Destroy the entire deployment. Destroys all running containers. If `printServerLogs` is true,
 // will print container logs before killing the container.
 func (d *Deployment) Destroy(t *testing.T) {