segment_session_test.go

package main

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

type sessionArgs struct {
	userInfoSeparator string
	username          string
	hostname          string
	goos              string
	connection        string
	client            string
	sshIcon           string
}

func setupSession(args *sessionArgs) session {
	env := new(MockedEnvironment)
	env.On("getCurrentUser", nil).Return(args.username)
	env.On("getHostName", nil).Return(args.hostname, nil)
	env.On("getRuntimeGOOS", nil).Return(args.goos)
	env.On("getenv", "SSH_CONNECTION").Return(args.connection)
	env.On("getenv", "SSH_CLIENT").Return(args.client)
	props := &properties{
		values: map[Property]interface{}{
			UserInfoSeparator: args.userInfoSeparator,
			SSHIcon:           args.sshIcon,
		},
		foreground: "#fff",
		background: "#000",
	}
	s := session{
		env:   env,
		props: props,
	}
	return s
}

func testUserInfoWriter(args *sessionArgs) string {
	s := setupSession(args)
	return s.getFormattedText()
}

func TestWriteUserInfo(t *testing.T) {
	want := "<#fff>bill</>@<#fff>surface</>"
	args := &sessionArgs{
		userInfoSeparator: "@",
		username:          "bill",
		hostname:          "surface",
		goos:              "windows",
	}
	got := testUserInfoWriter(args)
	assert.EqualValues(t, want, got)
}

func TestWriteUserInfoWindowsIncludingHostname(t *testing.T) {
	want := "<#fff>bill</>@<#fff>surface</>"
	args := &sessionArgs{
		userInfoSeparator: "@",
		username:          "surface\\bill",
		hostname:          "surface",
		goos:              "windows",
	}
	got := testUserInfoWriter(args)
	assert.EqualValues(t, want, got)
}

func TestWriteOnlyUsername(t *testing.T) {
	args := &sessionArgs{
		userInfoSeparator: "@",
		username:          "surface\\bill",
		hostname:          "surface",
		goos:              "windows",
	}
	s := setupSession(args)
	s.props.values[DisplayHost] = false
	want := "<#fff>bill</><#fff></>"
	got := s.getFormattedText()
	assert.EqualValues(t, want, got)
}

func TestWriteOnlyHostname(t *testing.T) {
	args := &sessionArgs{
		userInfoSeparator: "@",
		username:          "surface\\bill",
		hostname:          "surface",
		goos:              "windows",
	}
	s := setupSession(args)
	s.props.values[DisplayUser] = false
	want := "<#fff></><#fff>surface</>"
	got := s.getFormattedText()
	assert.EqualValues(t, want, got)
}

func TestWriteActiveSSHSession(t *testing.T) {
	want := "ssh <#fff>bill</>@<#fff>surface</>"
	args := &sessionArgs{
		userInfoSeparator: "@",
		username:          "bill",
		hostname:          "surface",
		goos:              "windows",
		sshIcon:           "ssh ",
		connection:        "1.1.1.1",
	}
	got := testUserInfoWriter(args)
	assert.EqualValues(t, want, got)
}

func TestActiveSSHSessionInactive(t *testing.T) {
	env := new(MockedEnvironment)
	env.On("getenv", "SSH_CONNECTION").Return("")
	env.On("getenv", "SSH_CLIENT").Return("")
	s := &session{
		env: env,
	}
	assert.False(t, s.activeSSHSession())
}

func TestActiveSSHSessionActiveConnection(t *testing.T) {
	env := new(MockedEnvironment)
	env.On("getenv", "SSH_CONNECTION").Return("1.1.1.1")
	env.On("getenv", "SSH_CLIENT").Return("")
	s := &session{
		env: env,
	}
	assert.True(t, s.activeSSHSession())
}

func TestActiveSSHSessionActiveClient(t *testing.T) {
	env := new(MockedEnvironment)
	env.On("getenv", "SSH_CONNECTION").Return("")
	env.On("getenv", "SSH_CLIENT").Return("1.1.1.1")
	s := &session{
		env: env,
	}
	assert.True(t, s.activeSSHSession())
}

func TestActiveSSHSessionActiveBoth(t *testing.T) {
	env := new(MockedEnvironment)
	env.On("getenv", "SSH_CONNECTION").Return("2.2.2.2")
	env.On("getenv", "SSH_CLIENT").Return("1.1.1.1")
	s := &session{
		env: env,
	}
	assert.True(t, s.activeSSHSession())
}