cadvisor/integration/tests/api/event_test.go
Victor Marmol 651ee5648f Use watch instead of sh.
Use of sh was making the test flaky and harder to read.
2015-07-01 14:54:35 -07:00

80 lines
2.2 KiB
Go

// Copyright 2015 Google Inc. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package api
import (
"strings"
"testing"
"time"
info "github.com/google/cadvisor/info/v1"
"github.com/google/cadvisor/integration/framework"
"github.com/stretchr/testify/require"
)
func TestStreamingEventInformationIsReturned(t *testing.T) {
// TODO(vmarmol): De-flake and re-enable.
t.Skip()
fm := framework.New(t)
defer fm.Cleanup()
// Watch for container deletions
einfo := make(chan *info.Event)
go func() {
err := fm.Cadvisor().Client().EventStreamingInfo("?deletion_events=true&stream=true&subcontainers=true", einfo)
require.NoError(t, err)
}()
// Create a short-lived container.
containerId := fm.Docker().RunBusybox("sleep", "2")
// Wait for the deletion event.
timeout := time.After(30 * time.Second)
done := false
for !done {
select {
case ev := <-einfo:
if ev.EventType == info.EventContainerDeletion {
if strings.Contains(ev.ContainerName, containerId) {
done = true
}
}
case <-timeout:
t.Errorf(
"timeout happened before destruction event was detected for container %q", containerId)
done = true
}
}
// We should have already received a creation event.
waitForStaticEvent(containerId, "?creation_events=true&subcontainers=true", t, fm, info.EventContainerCreation)
}
func waitForStaticEvent(containerId string, urlRequest string, t *testing.T, fm framework.Framework, typeEvent info.EventType) {
einfo, err := fm.Cadvisor().Client().EventStaticInfo(urlRequest)
require.NoError(t, err)
found := false
for _, ev := range einfo {
if ev.EventType == typeEvent {
if strings.Contains(ev.ContainerName, containerId) {
found = true
break
}
}
}
require.True(t, found)
}