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:
parent
6512ef175a
commit
e08f826ce6
|
@ -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
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue