Individuate archive tests #2293

This commit is contained in:
Cody Rose 2024-01-12 09:39:48 -05:00 committed by GitHub
parent 651beff492
commit b03cc30263
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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)
}
})
} }
} }