shad-go/distbuild/pkg/artifact/client_test.go
Sergey Paramoshkin dc939b6ed1
update repo
2023-10-02 22:32:41 +03:00

51 lines
1.2 KiB
Go

package artifact_test
import (
"context"
"io/ioutil"
"net/http"
"net/http/httptest"
"path/filepath"
"testing"
"github.com/stretchr/testify/require"
"go.uber.org/zap/zaptest"
"gitlab.com/manytask/itmo-go/private/distbuild/pkg/artifact"
"gitlab.com/manytask/itmo-go/private/distbuild/pkg/build"
)
func TestArtifactTransfer(t *testing.T) {
remoteCache := newTestCache(t)
localCache := newTestCache(t)
id := build.ID{0x01}
dir, commit, _, err := remoteCache.Create(id)
require.NoError(t, err)
require.NoError(t, ioutil.WriteFile(filepath.Join(dir, "a.txt"), []byte("foobar"), 0o777))
require.NoError(t, commit())
l := zaptest.NewLogger(t)
h := artifact.NewHandler(l, remoteCache.Cache)
mux := http.NewServeMux()
h.Register(mux)
server := httptest.NewServer(mux)
defer server.Close()
ctx := context.Background()
require.NoError(t, artifact.Download(ctx, server.URL, localCache.Cache, id))
dir, unlock, err := localCache.Get(id)
require.NoError(t, err)
defer unlock()
content, err := ioutil.ReadFile(filepath.Join(dir, "a.txt"))
require.NoError(t, err)
require.Equal(t, []byte("foobar"), content)
err = artifact.Download(ctx, server.URL, localCache.Cache, build.ID{0x02})
require.Error(t, err)
}