feat(assert): added AssertDirEmpty and AssertDirNotEmpty. fixes #37

assert.go

@@ -680,3 +680,32 @@ func AssertNotRegexp(t testRunner, regex interface{}, txt interface{}, msg ...in
 
 	internal.AssertRegexpHelper(t, regex, txt, false, msg...)
 }
+
+// AssertDirEmpty asserts that a directory is empty.
+// The test will pass when the directory is empty or does not exist.
+//
+// Example:
+//  testza.AssertDirEmpty(t, "FolderName")
+func AssertDirEmpty(t testRunner, dir string, msg ...interface{}) {
+	if test, ok := t.(helper); ok {
+		test.Helper()
+	}
+
+	if !internal.AssertDirEmptyHelper(t, dir) {
+		internal.Fail(t, "The directory !!is not!! empty.", internal.NewObjectsSingleNamed("Directory", dir))
+	}
+}
+
+// AssertDirNotEmpty asserts that a directory is not empty
+//
+// Example:
+//  testza.AssertDirNotEmpty(t, "FolderName")
+func AssertDirNotEmpty(t testRunner, dir string, msg ...interface{}) {
+	if test, ok := t.(helper); ok {
+		test.Helper()
+	}
+
+	if internal.AssertDirEmptyHelper(t, dir) {
+		internal.Fail(t, "The directory !!is!! empty.", internal.NewObjectsSingleNamed("Directory", dir))
+	}
+}

assert_test.go

@@ -4,7 +4,9 @@ import (
 	"errors"
 	"fmt"
 	"go/types"
+	"io/ioutil"
 	"math/rand"
+	"os"
 	"reflect"
 	"regexp"
 	"testing"
@@ -956,3 +958,28 @@ func TestAssertNotRegexp_fail(t *testing.T) {
 		AssertNotRegexp(t, rxp, "Hello, World!")
 	})
 }
+
+func TestAssertDirEmpty(t *testing.T) {
+	tempdir, _ := ioutil.TempDir("testdata", "temp")
+	AssertDirEmpty(t, tempdir)
+	defer os.RemoveAll(tempdir)
+}
+
+func TestAssertDirEmpty_fail(t *testing.T) {
+	AssertTestFails(t, func(t TestingPackageWithFailFunctions) {
+		AssertDirEmpty(t, "non-existent")
+		AssertDirEmpty(t, "internal")
+	})
+}
+
+func TestAssertDirNotEmpty(t *testing.T) {
+	AssertDirNotEmpty(t, "testdata")
+}
+
+func TestAssertDirNotEmpty_fail(t *testing.T) {
+	tempdir, _ := ioutil.TempDir("testdata", "temp")
+	AssertTestFails(t, func(t TestingPackageWithFailFunctions) {
+		AssertDirNotEmpty(t, tempdir)
+	})
+	defer os.RemoveAll(tempdir)
+}

internal/assertion_helper.go

@@ -2,7 +2,10 @@ package internal
 
 import (
 	"bytes"
+	"errors"
 	"fmt"
+	"io"
+	"os"
 	"reflect"
 	"regexp"
 	"strings"
@@ -262,3 +265,15 @@ func AssertRegexpHelper(t testRunner, regex interface{}, txt interface{}, should
 		}, msg...)
 	}
 }
+
+// AssertDirEmptyHelper checks for io.EOF to determine if directory empty or not
+func AssertDirEmptyHelper(t testRunner, dir string) bool {
+	f, err := os.Open(dir)
+	if err != nil {
+		Fail(t, "Error opening directory specified", NewObjectsSingleNamed("dir", dir))
+	}
+	defer f.Close()
+
+	_, err = f.Readdirnames(1)
+	return errors.Is(err, io.EOF)
+}