Backport of NET-5187: Upgrade test timeout due to log producer errors into release/1.16.x (#18472)

backport of commit 59bfc81d49a905480329a38c0d8d0109ea100d3e

Co-authored-by: NiniOak <anita.akaeze@hashicorp.com>
This commit is contained in:
hc-github-team-consul-core 2023-08-15 12:20:44 -05:00 committed by GitHub
parent 6512ef175a
commit e08f826ce6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 19 additions and 5 deletions

View File

@ -68,14 +68,17 @@ func (g ConnectContainer) GetPort(port int) (int, error) {
} }
func (g ConnectContainer) Restart() error { func (g ConnectContainer) Restart() error {
_, err := g.GetStatus() var deferClean utils.ResettableDefer
if err != nil { defer deferClean.Execute()
return fmt.Errorf("error fetching sidecar container state %s", err)
if utils.FollowLog {
if err := g.container.StopLogProducer(); err != nil {
return fmt.Errorf("stopping log producer: %w", err)
}
} }
fmt.Printf("Stopping container: %s\n", g.GetName()) fmt.Printf("Stopping container: %s\n", g.GetName())
err = g.container.Stop(g.ctx, nil) err := g.container.Stop(g.ctx, nil)
if err != nil { if err != nil {
return fmt.Errorf("error stopping sidecar container %s", err) return fmt.Errorf("error stopping sidecar container %s", err)
} }
@ -85,6 +88,17 @@ func (g ConnectContainer) Restart() error {
if err != nil { if err != nil {
return fmt.Errorf("error starting sidecar container %s", err) return fmt.Errorf("error starting sidecar container %s", err)
} }
if utils.FollowLog {
if err := g.container.StartLogProducer(g.ctx); err != nil {
return fmt.Errorf("starting log producer: %w", err)
}
g.container.FollowOutput(&LogConsumer{})
deferClean.Add(func() {
_ = g.container.StopLogProducer()
})
}
return nil return nil
} }