mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-21 06:31:11 +02:00
2bf8540855
This is initial PR to push the initial code, it has several known problems which are going to be addressed in follow-up PRs: 1. there's no "cluster destroy", so the only way to stop the VMs is to `pkill firecracker` 2. provisioner creates state in `/tmp` and never deletes it, that is required to keep cluster running when `osctl cluster create` finishes 3. doesn't run any controller process around firecracker to support reboots/CNI cleanup (vethxyz interfaces are lingering on the host as they're never cleaned up) The plan is to create some structure in `~/.talos` to manage cluster state, e.g. `~/.talos/clusters/<name>` which will contain all the required files (disk images, file sockets, VM logs, etc.). This directory structure will also work as a way to detect running clusters and clean them up. For point number 3, `osctl cluster create` is going to exec lightweight process to control the firecracker VM process and to simulate VM reboots if firecracker finishes cleanly (when VM reboots). 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/internal/pkg/provision/providers/firecracker/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()))
|
|
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()))
|
|
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()))
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, http.StatusNotFound, resp.StatusCode)
|
|
|
|
assert.NoError(t, resp.Body.Close())
|
|
|
|
assert.NoError(t, srv.Shutdown(context.Background()))
|
|
}
|