Fix test

Author: anisaoshafi

test/integration/emulator_select_test.go

@@ -26,7 +26,9 @@ func freshConfigEnv(t *testing.T) (env.Environ, string) {
 	require.NoError(t, os.MkdirAll(filepath.Join(tmpHome, ".config"), 0755))
 	return env.Without(env.Home).
 		With(env.Home, tmpHome).
-		With("USERPROFILE", tmpHome), tmpHome
+		With("USERPROFILE", tmpHome).
+		With(env.Keyring, "file").
+		With(env.DisableEvents, "1"), tmpHome
 }
 
 func TestEmulatorFlagSwitchesConfigToSnowflake(t *testing.T) {
@@ -79,23 +81,25 @@ func TestFirstRunShowsEmulatorSelectionPrompt(t *testing.T) {
 		close(outputCh)
 	}()
 
-	require.Eventually(t, func() bool {
+	if !assert.Eventually(t, func() bool {
 		return bytes.Contains(out.Bytes(), []byte("Which emulator would you like to use?"))
-	}, 10*time.Second, 100*time.Millisecond, "emulator selection prompt should appear on first run")
+	}, 100*time.Second, 100*time.Millisecond, "emulator selection prompt should appear on first run") {
+		t.Logf("process output so far:\n%s", out.String())
+		t.FailNow()
+	}
 
-	// Choose Snowflake.
-	_, err = ptmx.Write([]byte("s"))
+	// Accept the default (AWS) by pressing Enter.
+	_, err = ptmx.Write([]byte("\r"))
 	require.NoError(t, err)
 
-	require.Eventually(t, func() bool {
-		return bytes.Contains(out.Bytes(), []byte("Snowflake emulator selected"))
-	}, 5*time.Second, 100*time.Millisecond, "confirmation should appear after selection")
-
-	// Config is written before the confirmation is emitted, so it is safe to read now.
+	// The config is written inside onSelected, before container.Start is attempted.
+	// Polling the file directly is more reliable than waiting for the TUI confirmation
+	// message, which may not render if container.Start fails fast (e.g. license check).
 	configPath := filepath.Join(tmpHome, ".config", "lstk", "config.toml")
-	got, err := os.ReadFile(configPath)
-	require.NoError(t, err, "config file should have been created on first run")
-	assert.Contains(t, string(got), `type = "snowflake"`)
+	require.Eventually(t, func() bool {
+		got, readErr := os.ReadFile(configPath)
+		return readErr == nil && bytes.Contains(got, []byte(`type = "aws"`))
+	}, 10*time.Second, 100*time.Millisecond, "config file should be written with aws emulator on first run")
 
 	// Kill the process — we do not need the emulator to actually start.
 	cancel()