// Copyright (c) 2021 AUTHORS All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

//go:build linux || darwin || freebsd || illumos || solaris

package peercred

import (
	"fmt"
	"net"
	"os"
	"path/filepath"
	"testing"
)

func TestUnixSock(t *testing.T) {
	d := t.TempDir()
	path := filepath.Join(d, "foo.sock")
	sock, err := net.Listen("unix", path)
	if err != nil {
		t.Fatal(err)
	}
	defer sock.Close()

	clientConnCh := make(chan net.Conn, 1)
	go func() {
		defer close(clientConnCh)
		c, err := net.Dial("unix", path)
		if err != nil {
			t.Error(err)
			return
		}
		clientConnCh <- c
	}()
	clientConn, ok := <-clientConnCh
	if !ok {
		return
	}
	defer clientConn.Close()

	c, err := sock.Accept()
	if err != nil {
		t.Fatalf("Accept: %v", err)
	}
	defer c.Close()

	creds := Get(c)
	if err != nil {
		t.Fatalf("Get: %v", err)
	}

	uid := creds.UID()
	if uid == "" {
		t.Errorf("no UID")
	}
	if got, want := uid, fmt.Sprint(os.Getuid()); got != want {
		t.Errorf("UID = %q; want %q", got, want)
	}

	pid := creds.PID()
	if pid == 0 {
		t.Errorf("no PID")
	}
	if got, want := pid, os.Getpid(); got != want {
		t.Errorf("PID = %v; want %v", got, want)
	}
}