diff --git a/go.mod b/go.mod index c728a465c..f2014b43f 100644 --- a/go.mod +++ b/go.mod @@ -13,7 +13,7 @@ require ( github.com/Azure/go-autorest/autorest/azure/auth v0.5.11 github.com/TheZeroSlave/zapsentry v1.11.0 github.com/aws/aws-sdk-go v1.44.83 - github.com/bill-rich/disk-buffer-reader v0.1.3 + github.com/bill-rich/disk-buffer-reader v0.1.4 github.com/bill-rich/go-syslog v0.0.0-20220413021637-49edb52a574c github.com/bitfinexcom/bitfinex-api-go v0.0.0-20210608095005-9e0b26f200fb github.com/bradleyfalzon/ghinstallation/v2 v2.1.0 diff --git a/go.sum b/go.sum index e41286dab..c69e71ecd 100644 --- a/go.sum +++ b/go.sum @@ -104,8 +104,8 @@ github.com/aws/aws-sdk-go v1.44.83 h1:7+Rtc2Eio6EKUNoZeMV/IVxzVrY5oBQcNPtCcgIHYJ github.com/aws/aws-sdk-go v1.44.83/go.mod h1:y4AeaBuwd2Lk+GepC1E9v0qOiTws0MIWAX4oIKwKHZo= github.com/benbjohnson/clock v1.1.0 h1:Q92kusRqC1XV2MjkWETPvjJVqKetz1OzxZB7mHJLju8= github.com/benbjohnson/clock v1.1.0/go.mod h1:J11/hYXuz8f4ySSvYwY0FKfm+ezbsZBKZxNJlLklBHA= -github.com/bill-rich/disk-buffer-reader v0.1.3 h1:8RUWjBrLB52kwJUjVDdrU0S/7z07DrMshgudfiy6QFs= -github.com/bill-rich/disk-buffer-reader v0.1.3/go.mod h1:VVzzsK1Ac2AnpOfp/5r9JlIFaFkZ9uSf7zisZayCt0Y= +github.com/bill-rich/disk-buffer-reader v0.1.4 h1:boaUh/NBiTGyH/ZmJvBE48CqK3UAsnezkONE4GT7H1w= +github.com/bill-rich/disk-buffer-reader v0.1.4/go.mod h1:VVzzsK1Ac2AnpOfp/5r9JlIFaFkZ9uSf7zisZayCt0Y= github.com/bill-rich/go-syslog v0.0.0-20220413021637-49edb52a574c h1:tSME5FDS02qQll3JYodI6RZR/g4EKOHApGv1wMZT+Z0= github.com/bill-rich/go-syslog v0.0.0-20220413021637-49edb52a574c/go.mod h1:+sCc6hztur+oZCLOsNk6wCCy+GLrnSNHSRmTnnL+8iQ= github.com/bitfinexcom/bitfinex-api-go v0.0.0-20210608095005-9e0b26f200fb h1:9v7Bzlg+1EBYi2IYcUmOwHReBEfqBbYIj3ZCi9cIe1Q= diff --git a/pkg/handlers/archive_test.go b/pkg/handlers/archive_test.go index 96a68472c..99fa1289d 100644 --- a/pkg/handlers/archive_test.go +++ b/pkg/handlers/archive_test.go @@ -4,6 +4,8 @@ import ( "net/http" "regexp" "testing" + + diskbufferreader "github.com/bill-rich/disk-buffer-reader" ) func TestArchiveHandler(t *testing.T) { @@ -52,6 +54,11 @@ func TestArchiveHandler(t *testing.T) { 1543, "AKIAYVP4CIPPH5TNP3SW", }, + "zip-single": { + "https://raw.githubusercontent.com/bill-rich/bad-secrets/master/aws-canary-creds.zip", + 1, + "AKIAYVP4CIPPH5TNP3SW", + }, } for name, testCase := range tests { @@ -64,7 +71,11 @@ func TestArchiveHandler(t *testing.T) { archive := Archive{} archive.New() - archiveChan := archive.FromFile(resp.Body) + newReader, err := diskbufferreader.New(resp.Body) + if err != nil { + t.Errorf("error creating reusable reader: %s", err) + } + archiveChan := archive.FromFile(newReader) count := 0 re := regexp.MustCompile(testCase.matchString)