b8c7596fc5
Некоторые проверки не удались
CI / test (push) Successful in 7m31s
Docker / Build and publish worker image (push) Successful in 13m54s
SSH Source-Install E2E / Alpine/Ubuntu/Arch source-install E2E (push) Failing after 30s
34 строки
1.0 KiB
Go
34 строки
1.0 KiB
Go
package distworker
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestWorkerStateRoundTripUsesPrivatePermissions(t *testing.T) {
|
|
dir := filepath.Join(t.TempDir(), "state")
|
|
require.NoError(t, os.Mkdir(dir, 0o700))
|
|
path := filepath.Join(dir, "state.json")
|
|
want := workerState{Token: "secret", WorkerID: "worker-1", VerificationKey: "key", SigningKeyID: "key-2026", AccountID: 7, ConfigVersion: 2}
|
|
require.NoError(t, saveWorkerState(path, want))
|
|
info, err := os.Stat(path)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, os.FileMode(0o600), info.Mode().Perm())
|
|
got, err := loadWorkerState(path)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, want, got)
|
|
}
|
|
|
|
func TestWorkerStateRejectsInsecurePermissions(t *testing.T) {
|
|
dir := filepath.Join(t.TempDir(), "state")
|
|
require.NoError(t, os.Mkdir(dir, 0o700))
|
|
path := filepath.Join(dir, "state.json")
|
|
require.NoError(t, os.WriteFile(path, []byte(`{"token":"secret"}`), 0o644))
|
|
_, err := loadWorkerState(path)
|
|
require.Error(t, err)
|
|
}
|