Skip to content

Commit

Permalink
testutil/pkgdata: Add TarEntry shorthand constructors
Browse files Browse the repository at this point in the history
Introduce shorthand constructors for testutil.TarEntry structures. These
are REG(), DIR() and LNK() functions.

The rationale for this addition is to make the test case definition less
verbose. There are other changes in the queue that construct custom
packages to test various test cases. In all these new tests (and old
ones as well) we only care about file's type, path, mode and content.
With these shorthand constructors and function aliases we can define tar
entries like:

	DIR(0755, "./data/"),
	REG(0600, "./data/document.txt", "words words words"),
	LNK(0777, "./data/document", "document.txt"),

Instead of:

	testutil.TarEntry{
		Header: tar.Header{
			Name: "./data/",
			Mode: 0755,
		},
	},
	testutil.TarEntry{
		Header: tar.Header{
			Name: "./document.txt",
			Mode: 0600,
		},
		Content: []byte("words words words"),
	},
	testutil.TarEntry{
		Header: tar.Header{
			Name:     "./document.txt",
			Mode:     0777,
			Linkname: "document.txt",
		},
	},

The reason for the 3 letter names and the order of arguments is to make
the list of paths aligned on the same column in tarball definitions.

These function only create barebone TarEntry. It'll still get adjusted
when passed through fixupTarEntry().
  • Loading branch information
woky committed Oct 9, 2023
1 parent 379ddae commit 2bc9b0d
Show file tree
Hide file tree
Showing 2 changed files with 71 additions and 0 deletions.
32 changes: 32 additions & 0 deletions internal/testutil/pkgdata.go
Original file line number Diff line number Diff line change
Expand Up @@ -277,3 +277,35 @@ func MustMakeDeb(entries []TarEntry) []byte {
}
return data
}

func REG(mode int64, path, content string) TarEntry {
return TarEntry{
Header: tar.Header{
Typeflag: tar.TypeReg,
Name: path,
Mode: mode,
},
Content: []byte(content),
}
}

func DIR(mode int64, path string) TarEntry {
return TarEntry{
Header: tar.Header{
Typeflag: tar.TypeDir,
Name: path,
Mode: mode,
},
}
}

func LNK(mode int64, path, target string) TarEntry {
return TarEntry{
Header: tar.Header{
Typeflag: tar.TypeSymlink,
Name: path,
Mode: mode,
Linkname: target,
},
}
}
39 changes: 39 additions & 0 deletions internal/testutil/pkgdata_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -398,3 +398,42 @@ func (s *S) TestMustMakeDeb(c *C) {
},
}})
}

func (s *S) TestTarEntryShortHands(c *C) {
var testCases = []struct {
shorthand testutil.TarEntry
result testutil.TarEntry
}{{
testutil.REG(0600, "./document.txt", "cats are best"),
testutil.TarEntry{
Header: tar.Header{
Typeflag: tar.TypeReg,
Name: "./document.txt",
Mode: 0600,
},
Content: []byte("cats are best"),
},
}, {
testutil.DIR(0755, "./home/user"),
testutil.TarEntry{
Header: tar.Header{
Typeflag: tar.TypeDir,
Name: "./home/user",
Mode: 0755,
},
},
}, {
testutil.LNK(0755, "./lib", "./usr/lib/"),
testutil.TarEntry{
Header: tar.Header{
Typeflag: tar.TypeSymlink,
Name: "./lib",
Mode: 0755,
Linkname: "./usr/lib/",
},
},
}}
for _, test := range testCases {
c.Assert(test.shorthand, DeepEquals, test.result)
}
}

0 comments on commit 2bc9b0d

Please sign in to comment.