Move distbuild blocks back to distbuild

artifacttest/README.md

@@ -1,3 +0,0 @@
-# artifacttest
-
-Этот пакет содержит тесты на [artifact](../distbuild/pkg/artifact).

distbuild/pkg/artifact/cache_test.go

@@ -1,4 +1,4 @@
-package artifacttest
+package artifact_test
 
 import (
 	"errors"
@@ -32,12 +32,13 @@ func newTestCache(t *testing.T) *testCache {
 	}
 	require.NoError(t, err)
 
-	return &testCache{Cache: cache, tmpDir: tmpDir}
+	c := &testCache{Cache: cache, tmpDir: tmpDir}
+	t.Cleanup(c.cleanup)
+	return c
 }
 
 func TestCache(t *testing.T) {
 	c := newTestCache(t)
-	defer c.cleanup()
 
 	idA := build.ID{'a'}
 
@@ -73,7 +74,6 @@ func TestCache(t *testing.T) {
 
 func TestAbortWrite(t *testing.T) {
 	c := newTestCache(t)
-	defer c.cleanup()
 
 	idA := build.ID{'a'}
 
@@ -87,7 +87,6 @@ func TestAbortWrite(t *testing.T) {
 
 func TestArtifactExists(t *testing.T) {
 	c := newTestCache(t)
-	defer c.cleanup()
 
 	idA := build.ID{'a'}
 

distbuild/pkg/artifact/client_test.go

@@ -1,4 +1,4 @@
-package artifacttest
+package artifact_test
 
 import (
 	"context"
@@ -17,9 +17,7 @@ import (
 
 func TestArtifactTransfer(t *testing.T) {
 	remoteCache := newTestCache(t)
-	defer remoteCache.cleanup()
 	localCache := newTestCache(t)
-	defer localCache.cleanup()
 
 	id := build.ID{0x01}
 

distbuild/pkg/filecache/client_test.go

@@ -29,16 +29,12 @@ func newEnv(t *testing.T) *env {
 	mux := http.NewServeMux()
 
 	cache := newCache(t)
-	defer func() {
-		if cache != nil {
-			cache.cleanup()
-		}
-	}()
 
 	handler := filecache.NewHandler(l, cache.Cache)
 	handler.Register(mux)
 
 	server := httptest.NewServer(mux)
+	t.Cleanup(server.Close)
 
 	client := filecache.NewClient(l, server.URL)
 
@@ -48,19 +44,11 @@ func newEnv(t *testing.T) *env {
 		client: client,
 	}
 
-	cache = nil
 	return env
 }
 
-func (e *env) stop() {
-	e.server.Close()
-	e.cache.cleanup()
-}
-
 func TestFileUpload(t *testing.T) {
 	env := newEnv(t)
-	defer env.stop()
-
 	content := bytes.Repeat([]byte("foobar"), 1024*1024)
 
 	tmpFilePath := filepath.Join(env.cache.tmpDir, "foo.txt")
@@ -115,10 +103,8 @@ func TestFileUpload(t *testing.T) {
 
 func TestFileDownload(t *testing.T) {
 	env := newEnv(t)
-	defer env.stop()
 
 	localCache := newCache(t)
-	defer localCache.cleanup()
 
 	id := build.ID{0x01}
 

distbuild/pkg/filecache/filecache_test.go

@@ -24,7 +24,9 @@ func newCache(t *testing.T) *testCache {
 	c, err := filecache.New(tmpDir)
 	require.NoError(t, err)
 
-	return &testCache{Cache: c, tmpDir: tmpDir}
+	cc := &testCache{Cache: c, tmpDir: tmpDir}
+	t.Cleanup(cc.cleanup)
+	return cc
 }
 
 func (c *testCache) cleanup() {

distbuild/pkg/tarstream/stream_test.go

@@ -1,4 +1,4 @@
-package tarstreamtest
+package tarstream_test
 
 import (
 	"bytes"

filecachetest/README.md

@@ -1,3 +0,0 @@
-# filecachetest
-
-Этот пакет содержит тесты на [filecache](../distbuild/pkg/filecache).

tarstreamtest/README.md

@@ -1,3 +0,0 @@
-# tarstreamtest
-
-Этот пакет содержит тесты на [tarstream](../distbuild/pkg/tarstream).