Function TestRemoveSSHHostEntryFirstEntry

cmd/delete_test.go:91–119

TestRemoveSSHHostEntryFirstEntry verifies that the removeSSHHostEntry function correctly handles removal of the first entry in the SSH config file.

(t *testing.T)

// TestRemoveSSHHostEntryFirstEntry verifies that the removeSSHHostEntry function
// correctly handles removal of the first entry in the SSH config file.
func TestRemoveSSHHostEntryFirstEntry(t *testing.T) {
	env := testutils.SetupTestEnvironment(t)
	defer env.Cleanup()

	sshDir := filepath.Join(env.TempDir, ".ssh")
	require.NoError(t, os.MkdirAll(sshDir, 0700))

	sshConfigPath := filepath.Join(sshDir, "config")
	sshConfig := `Host tnr-first
				  HostName 192.168.1.100
				  User ubuntu

			      Host tnr-second
			   	  HostName 192.168.1.101
				  User ubuntu
	`

	require.NoError(t, os.WriteFile(sshConfigPath, []byte(sshConfig), 0600))

	err := removeSSHHostEntry(sshConfigPath, "first")
	require.NoError(t, err)

	configData, err := os.ReadFile(sshConfigPath)
	require.NoError(t, err)

	configContent := string(configData)
	assert.NotContains(t, configContent, "tnr-first")
	assert.Contains(t, configContent, "tnr-second")
}

// TestRemoveSSHHostEntryLastEntry verifies that the removeSSHHostEntry function
// correctly handles removal of the last entry in the SSH config file.