diff options
Diffstat (limited to 'pkg/cleanup')
-rw-r--r-- | pkg/cleanup/BUILD | 17 | ||||
-rw-r--r-- | pkg/cleanup/cleanup_state_autogen.go | 3 | ||||
-rw-r--r-- | pkg/cleanup/cleanup_test.go | 66 |
3 files changed, 3 insertions, 83 deletions
diff --git a/pkg/cleanup/BUILD b/pkg/cleanup/BUILD deleted file mode 100644 index 5c34b9872..000000000 --- a/pkg/cleanup/BUILD +++ /dev/null @@ -1,17 +0,0 @@ -load("//tools:defs.bzl", "go_library", "go_test") - -package(licenses = ["notice"]) - -go_library( - name = "cleanup", - srcs = ["cleanup.go"], - visibility = ["//:sandbox"], - deps = [ - ], -) - -go_test( - name = "cleanup_test", - srcs = ["cleanup_test.go"], - library = ":cleanup", -) diff --git a/pkg/cleanup/cleanup_state_autogen.go b/pkg/cleanup/cleanup_state_autogen.go new file mode 100644 index 000000000..8373268fa --- /dev/null +++ b/pkg/cleanup/cleanup_state_autogen.go @@ -0,0 +1,3 @@ +// automatically generated by stateify. + +package cleanup diff --git a/pkg/cleanup/cleanup_test.go b/pkg/cleanup/cleanup_test.go deleted file mode 100644 index ab3d9ed95..000000000 --- a/pkg/cleanup/cleanup_test.go +++ /dev/null @@ -1,66 +0,0 @@ -// Copyright 2020 The gVisor Authors. -// -// 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 cleanup - -import "testing" - -func testCleanupHelper(clean, cleanAdd *bool, release bool) func() { - cu := Make(func() { - *clean = true - }) - cu.Add(func() { - *cleanAdd = true - }) - defer cu.Clean() - if release { - return cu.Release() - } - return nil -} - -func TestCleanup(t *testing.T) { - clean := false - cleanAdd := false - testCleanupHelper(&clean, &cleanAdd, false) - if !clean { - t.Fatalf("cleanup function was not called.") - } - if !cleanAdd { - t.Fatalf("added cleanup function was not called.") - } -} - -func TestRelease(t *testing.T) { - clean := false - cleanAdd := false - cleaner := testCleanupHelper(&clean, &cleanAdd, true) - - // Check that clean was not called after release. - if clean { - t.Fatalf("cleanup function was called.") - } - if cleanAdd { - t.Fatalf("added cleanup function was called.") - } - - // Call the cleaner function and check that both cleanup functions are called. - cleaner() - if !clean { - t.Fatalf("cleanup function was not called.") - } - if !cleanAdd { - t.Fatalf("added cleanup function was not called.") - } -} |