mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-19 13:41:13 +02:00
9379cf9ee1
This change is only moving packages and updating import paths. Goal: expose `internal/pkg/provision` as `pkg/provision` to enable other projects to import Talos provisioning library. As cluster checks are almost always required as part of provisioning process, package `internal/pkg/cluster` was also made public as `pkg/cluster`. Other changes were direct dependencies discovered by `importvet` which were updated. Public packages (useful, general purpose packages with stable API): * `internal/pkg/conditions` -> `pkg/conditions` * `internal/pkg/tail` -> `pkg/tail` Private packages (used only on provisioning library internally): * `internal/pkg/inmemhttp` -> `pkg/provision/internal/inmemhttp` * `internal/pkg/kernel/vmlinuz` -> `pkg/provision/internal/vmlinuz` * `internal/pkg/cniutils` -> `pkg/provision/internal/cniutils` Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
62 lines
1.5 KiB
Go
62 lines
1.5 KiB
Go
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
package inmemhttp_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/talos-systems/talos/pkg/provision/internal/inmemhttp"
|
|
)
|
|
|
|
func TestServer(t *testing.T) {
|
|
srv, err := inmemhttp.NewServer("localhost:0")
|
|
assert.NoError(t, err)
|
|
|
|
contents := []byte("DEADBEEF")
|
|
|
|
assert.NoError(t, srv.AddFile("test.txt", contents))
|
|
|
|
srv.Serve()
|
|
defer srv.Shutdown(context.Background()) //nolint: errcheck
|
|
|
|
resp, err := http.Get(fmt.Sprintf("http://%s/test.txt", srv.GetAddr())) //nolint: noctx
|
|
assert.NoError(t, err)
|
|
|
|
defer resp.Body.Close() //nolint: errcheck
|
|
|
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
|
|
|
got, err := ioutil.ReadAll(resp.Body)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, contents, got)
|
|
|
|
assert.NoError(t, resp.Body.Close())
|
|
|
|
resp, err = http.Head(fmt.Sprintf("http://%s/test.txt", srv.GetAddr())) //nolint: noctx
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, http.StatusOK, resp.StatusCode)
|
|
|
|
assert.EqualValues(t, 8, resp.ContentLength)
|
|
|
|
assert.NoError(t, resp.Body.Close())
|
|
|
|
resp, err = http.Get(fmt.Sprintf("http://%s/test.txt2", srv.GetAddr())) //nolint: noctx
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, http.StatusNotFound, resp.StatusCode)
|
|
|
|
assert.NoError(t, resp.Body.Close())
|
|
|
|
assert.NoError(t, srv.Shutdown(context.Background()))
|
|
}
|