mirror of
https://github.com/coder/coder.git
synced 2026-06-02 20:48:20 +00:00
Spike Curtis
bddb808b25
Fixes all our Go file imports to match the preferred spec that we've _mostly_ been using. For example: ``` import ( "context" "time" "github.com/prometheus/client_golang/prometheus" "golang.org/x/xerrors" "gopkg.in/natefinch/lumberjack.v2" "cdr.dev/slog/v3" "github.com/coder/coder/v2/codersdk/agentsdk" "github.com/coder/serpent" ) ``` 3 groups: standard library, 3rd partly libs, Coder libs. This PR makes the change across the codebase. The PR in the stack above modifies our formatting to maintain this state of affairs, and is a separate PR so it's possible to review that one in detail.
121 lines
3.5 KiB
Go
121 lines
3.5 KiB
Go
package provisionersdk_test
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/spf13/afero"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"cdr.dev/slog/v3"
|
|
"github.com/coder/coder/v2/provisionersdk/tfpath"
|
|
"github.com/coder/coder/v2/testutil"
|
|
)
|
|
|
|
const workDirectory = "/tmp/coder/provisioner-34/work"
|
|
|
|
var now = time.Date(2023, time.June, 3, 4, 5, 6, 0, time.UTC)
|
|
|
|
func TestStaleSessions(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
prepare := func() (afero.Fs, slog.Logger) {
|
|
tempDir := t.TempDir()
|
|
fs := afero.NewBasePathFs(afero.NewOsFs(), tempDir)
|
|
logger := testutil.Logger(t).
|
|
Leveled(slog.LevelDebug).
|
|
Named("cleanup-test")
|
|
return fs, logger
|
|
}
|
|
|
|
t.Run("all sessions are stale", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort)
|
|
defer cancel()
|
|
|
|
fs, logger := prepare()
|
|
|
|
// given
|
|
first := tfpath.Session(workDirectory, uuid.NewString())
|
|
addSessionFolder(t, fs, first, now.Add(-7*24*time.Hour))
|
|
second := tfpath.Session(workDirectory, uuid.NewString())
|
|
addSessionFolder(t, fs, second, now.Add(-8*24*time.Hour))
|
|
third := tfpath.Session(workDirectory, uuid.NewString())
|
|
addSessionFolder(t, fs, third, now.Add(-9*24*time.Hour))
|
|
// tfDir is a fake session that will clean up the others
|
|
tfDir := tfpath.Session(workDirectory, uuid.NewString())
|
|
|
|
// when
|
|
err := tfDir.CleanStaleSessions(ctx, logger, fs, now)
|
|
require.NoError(t, err)
|
|
|
|
// then
|
|
entries, err := afero.ReadDir(fs, workDirectory)
|
|
require.NoError(t, err)
|
|
require.Empty(t, entries, "all session leftovers should be removed")
|
|
})
|
|
|
|
t.Run("one session is stale", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort)
|
|
defer cancel()
|
|
|
|
fs, logger := prepare()
|
|
|
|
// given
|
|
first := tfpath.Session(workDirectory, uuid.NewString())
|
|
addSessionFolder(t, fs, first, now.Add(-7*24*time.Hour))
|
|
second := tfpath.Session(workDirectory, uuid.NewString())
|
|
addSessionFolder(t, fs, second, now.Add(-6*24*time.Hour))
|
|
tfDir := tfpath.Session(workDirectory, uuid.NewString())
|
|
|
|
// when
|
|
err := tfDir.CleanStaleSessions(ctx, logger, fs, now)
|
|
require.NoError(t, err)
|
|
|
|
// then
|
|
entries, err := afero.ReadDir(fs, workDirectory)
|
|
require.NoError(t, err)
|
|
require.Len(t, entries, 1, "one session should be present")
|
|
require.Equal(t, second.WorkDirectory(), filepath.Join(workDirectory, entries[0].Name()), 1)
|
|
})
|
|
|
|
t.Run("no stale sessions", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitShort)
|
|
defer cancel()
|
|
|
|
fs, logger := prepare()
|
|
|
|
// given
|
|
first := tfpath.Session(workDirectory, uuid.NewString())
|
|
addSessionFolder(t, fs, first, now.Add(-6*24*time.Hour))
|
|
second := tfpath.Session(workDirectory, uuid.NewString())
|
|
addSessionFolder(t, fs, second, now.Add(-5*24*time.Hour))
|
|
tfDir := tfpath.Session(workDirectory, uuid.NewString())
|
|
|
|
// when
|
|
err := tfDir.CleanStaleSessions(ctx, logger, fs, now)
|
|
require.NoError(t, err)
|
|
|
|
// then
|
|
entries, err := afero.ReadDir(fs, workDirectory)
|
|
require.NoError(t, err)
|
|
require.Len(t, entries, 2, "both sessions should be present")
|
|
})
|
|
}
|
|
|
|
func addSessionFolder(t *testing.T, fs afero.Fs, files tfpath.Layout, modTime time.Time) {
|
|
workdir := files.WorkDirectory()
|
|
err := fs.MkdirAll(workdir, 0o755)
|
|
require.NoError(t, err, "can't create session folder")
|
|
require.NoError(t, fs.Chtimes(workdir, now, modTime), "can't chtime of session dir")
|
|
require.NoError(t, err, "can't set times")
|
|
}
|