github.com/taubyte/tau-cli@v0.1.13-0.20240326000942-487f0d57edfc/singletons/session/discovery_test.go (about)

     1  package session
     2  
     3  import (
     4  	"os"
     5  	"testing"
     6  
     7  	"github.com/mitchellh/go-ps"
     8  )
     9  
    10  func TestDiscovery(t *testing.T) {
    11  	// Overriding the prefix so that the test does not conflict with actual values
    12  	oldPrefix := sessionDirPrefix
    13  	sessionDirPrefix = "tau-test"
    14  	defer func() {
    15  		sessionDirPrefix = oldPrefix
    16  	}()
    17  
    18  	parentProcess, err := ps.FindProcess(os.Getppid())
    19  	if err != nil {
    20  		t.Error(err)
    21  		return
    22  	}
    23  
    24  	expectedDir := directoryFromPid(parentProcess.PPid())
    25  	dir, err := discoverOrCreateConfigFileLoc()
    26  	if err != nil {
    27  		t.Error(err)
    28  		return
    29  	}
    30  
    31  	if dir != expectedDir {
    32  		t.Errorf("Expected %s, got %s", expectedDir, dir)
    33  		return
    34  	}
    35  
    36  	err = os.Remove(expectedDir)
    37  	if err != nil {
    38  		t.Error(err)
    39  		return
    40  	}
    41  }