[test] package: add unit test for linkRequiredBy

2 files changed, 46 insertions, 0 deletions

diff --git a/package.go b/package.go
index ea5030a..475f7f4 100644
--- a/package.go
+++ b/package.go
@@ -274,6 +274,10 @@ func (pkg *Package) linkDep(dep *Package) bool {
 	return true
 }
 
+//
+// linkRequiredBy add the parent package as requirement by current package,
+// only if it's exist yet.
+//
 func (pkg *Package) linkRequiredBy(parentPkg *Package) bool {
 	for x := 0; x < len(pkg.RequiredBy); x++ {
 		if parentPkg.ImportPath == pkg.RequiredBy[x] {
diff --git a/package_test.go b/package_test.go
index f8dafea..0cf5022 100644
--- a/package_test.go
+++ b/package_test.go
@@ -183,3 +183,45 @@ func TestAddDep(t *testing.T) {
 	gitCurPkg.Deps = nil
 	gitCurPkg.DepsMissing = nil
 }
+
+func TestLinkRequiredBy(t *testing.T) {
+	cases := []struct {
+		desc          string
+		parentPkg     *Package
+		exp           bool
+		expRequiredBy []string
+	}{{
+		desc: "Not exist yet",
+		parentPkg: &Package{
+			ImportPath: "github.com/shuLhan/share",
+		},
+		exp: true,
+		expRequiredBy: []string{
+			"github.com/shuLhan/share",
+		},
+	}, {
+		desc: "Already exist",
+		parentPkg: &Package{
+			ImportPath: "github.com/shuLhan/share",
+		},
+		expRequiredBy: []string{
+			"github.com/shuLhan/share",
+		},
+	}}
+
+	gitCurPkg.RequiredBy = nil
+
+	var got bool
+
+	for _, c := range cases {
+		t.Log(c.desc)
+
+		got = gitCurPkg.linkRequiredBy(c.parentPkg)
+
+		test.Assert(t, "return value", c.exp, got, true)
+		test.Assert(t, "RequiredBy", c.expRequiredBy,
+			gitCurPkg.RequiredBy, true)
+	}
+
+	gitCurPkg.RequiredBy = nil
+}