mirror of
https://github.com/gohugoio/hugo.git
synced 2025-04-27 22:21:07 +03:00
parent
35e9b3ed1e
commit
21d17566a3
4 changed files with 30 additions and 10 deletions
|
@ -299,10 +299,20 @@ func (s *IntegrationTestBuilder) initBuilder() error {
|
|||
|
||||
isBinaryRe := regexp.MustCompile(`^(.*)(\.png|\.jpg)$`)
|
||||
|
||||
const dataSourceFilenamePrefix = "sourcefilename:"
|
||||
|
||||
for _, f := range s.data.Files {
|
||||
filename := filepath.Join(s.Cfg.WorkingDir, f.Name)
|
||||
data := bytes.TrimSuffix(f.Data, []byte("\n"))
|
||||
if isBinaryRe.MatchString(filename) {
|
||||
datastr := strings.TrimSpace(string(data))
|
||||
if strings.HasPrefix(datastr, dataSourceFilenamePrefix) {
|
||||
// Read from file relative to tue current dir.
|
||||
var err error
|
||||
wd, _ := os.Getwd()
|
||||
filename := filepath.Join(wd, strings.TrimSpace(strings.TrimPrefix(datastr, dataSourceFilenamePrefix)))
|
||||
data, err = os.ReadFile(filename)
|
||||
s.Assert(err, qt.IsNil)
|
||||
} else if isBinaryRe.MatchString(filename) {
|
||||
var err error
|
||||
data, err = base64.StdEncoding.DecodeString(string(data))
|
||||
s.Assert(err, qt.IsNil)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue