Function TestServer_TelemetryDisabled_FinalReport

cli/server_test.go:2531–2637

(t *testing.T)

}

func TestServer_TelemetryDisabled_FinalReport(t *testing.T) {
	t.Parallel()

	telemetryServerURL, deployment, snapshot := mockTelemetryServer(t)
	dbConnURL, err := dbtestutil.Open(t)
	require.NoError(t, err)

	cacheDir := t.TempDir()
	runServer := func(t *testing.T, opts runServerOpts) (chan error, context.CancelFunc) {
		ctx, cancelFunc := context.WithCancel(context.Background())
		inv, _ := clitest.New(t,
			"server",
			"--postgres-url", dbConnURL,
			"--http-address", ":0",
			"--access-url", "http://example.com",
			"--telemetry="+strconv.FormatBool(!opts.telemetryDisabled),
			"--telemetry-url", telemetryServerURL.String(),
			"--cache-dir", cacheDir,
			"--log-filter", ".*",
		)
		inv.Logger = inv.Logger.Named(opts.name)

		errChan := make(chan error, 1)
		pty := ptytest.New(t).Named(opts.name).Attach(inv)
		go func() {
			errChan <- inv.WithContext(ctx).Run()
			// close the pty here so that we can start tearing down resources. This test creates multiple servers with
			// associated ptys. There is a `t.Cleanup()` that does this, but it waits until the whole test is complete.
			_ = pty.Close()
		}()

		if opts.waitForSnapshot {
			pty.ExpectMatchContext(testutil.Context(t, testutil.WaitLong), "submitted snapshot")
		}
		if opts.waitForTelemetryDisabledCheck {
			pty.ExpectMatchContext(testutil.Context(t, testutil.WaitLong), "finished telemetry status check")
		}
		return errChan, cancelFunc
	}
	waitForShutdown := func(t *testing.T, errChan chan error) error {
		t.Helper()
		select {
		case err := <-errChan:
			return err
		case <-time.After(testutil.WaitMedium):
			t.Fatalf("timed out waiting for server to shutdown")
		}
		return nil
	}

	errChan, cancelFunc := runServer(t, runServerOpts{
		telemetryDisabled: true, waitForTelemetryDisabledCheck: true, name: "0disabled",
	})
	cancelFunc()
	require.NoError(t, waitForShutdown(t, errChan))

	// Since telemetry was disabled, we expect no deployments or snapshots.
	require.Empty(t, deployment)