package seededpostgres

import (
	"os"
	"testing"
)

func TestSeededPostgresNew(t *testing.T) {
	sp, err := New()
	if err != nil {
		t.Fatal(err)
	}

	if sp.Port() == 0 {
		t.Fatalf("Port not found")
	}

	db, err := sp.DB()
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()

	err = db.Ping()
	if err != nil {
		t.Fatal(err)
	}

	err = sp.Close()
	if err != nil {
		t.Fatal(err)
	}

	err = db.Ping()
	if err == nil {
		t.Fatalf("Expected error pinging database after closing SeededPostgres")
	}

	// ensure the temp directory is removed
	_, err = os.Open(sp.tempDir)
	if !os.IsNotExist(err) {
		t.Fatalf("Expected os.IsNotExist to return true for error: %v", err)
	}
}