mirror of
https://github.com/trufflesecurity/trufflehog.git
synced 2024-11-10 07:04:24 +00:00
Individuate archive tests #2293
This commit is contained in:
parent
651beff492
commit
b03cc30263
1 changed files with 31 additions and 29 deletions
|
@ -76,36 +76,38 @@ func TestArchiveHandler(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
for name, testCase := range tests {
|
for name, testCase := range tests {
|
||||||
resp, err := http.Get(testCase.archiveURL)
|
t.Run(name, func(t *testing.T) {
|
||||||
if err != nil || resp.StatusCode != http.StatusOK {
|
resp, err := http.Get(testCase.archiveURL)
|
||||||
t.Error(err)
|
if err != nil || resp.StatusCode != http.StatusOK {
|
||||||
}
|
t.Error(err)
|
||||||
defer resp.Body.Close()
|
|
||||||
|
|
||||||
archive := Archive{}
|
|
||||||
archive.New()
|
|
||||||
|
|
||||||
newReader, err := diskbufferreader.New(resp.Body)
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("error creating reusable reader: %s", err)
|
|
||||||
}
|
|
||||||
archiveChan := archive.FromFile(logContext.Background(), newReader)
|
|
||||||
|
|
||||||
count := 0
|
|
||||||
re := regexp.MustCompile(testCase.matchString)
|
|
||||||
matched := false
|
|
||||||
for chunk := range archiveChan {
|
|
||||||
count++
|
|
||||||
if re.Match(chunk) {
|
|
||||||
matched = true
|
|
||||||
}
|
}
|
||||||
}
|
defer resp.Body.Close()
|
||||||
if !matched && len(testCase.matchString) > 0 {
|
|
||||||
t.Errorf("%s: Expected string not found in archive.", name)
|
archive := Archive{}
|
||||||
}
|
archive.New()
|
||||||
if count != testCase.expectedChunks {
|
|
||||||
t.Errorf("%s: Unexpected number of chunks. Got %d, expected: %d", name, count, testCase.expectedChunks)
|
newReader, err := diskbufferreader.New(resp.Body)
|
||||||
}
|
if err != nil {
|
||||||
|
t.Errorf("error creating reusable reader: %s", err)
|
||||||
|
}
|
||||||
|
archiveChan := archive.FromFile(logContext.Background(), newReader)
|
||||||
|
|
||||||
|
count := 0
|
||||||
|
re := regexp.MustCompile(testCase.matchString)
|
||||||
|
matched := false
|
||||||
|
for chunk := range archiveChan {
|
||||||
|
count++
|
||||||
|
if re.Match(chunk) {
|
||||||
|
matched = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !matched && len(testCase.matchString) > 0 {
|
||||||
|
t.Errorf("%s: Expected string not found in archive.", name)
|
||||||
|
}
|
||||||
|
if count != testCase.expectedChunks {
|
||||||
|
t.Errorf("%s: Unexpected number of chunks. Got %d, expected: %d", name, count, testCase.expectedChunks)
|
||||||
|
}
|
||||||
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue