From cd0b46d5ef8c81bdb746a7ea987cf4ccd46202e3 Mon Sep 17 00:00:00 2001 From: Philip O'Toole Date: Sat, 6 Jan 2024 00:17:25 -0500 Subject: [PATCH] Test needs compressed data --- cluster/service_db_clstr_test.go | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/cluster/service_db_clstr_test.go b/cluster/service_db_clstr_test.go index 4daaefd9..38f80d33 100644 --- a/cluster/service_db_clstr_test.go +++ b/cluster/service_db_clstr_test.go @@ -2,6 +2,7 @@ package cluster import ( "bytes" + "compress/gzip" "encoding/binary" "errors" "fmt" @@ -287,7 +288,7 @@ func Test_ServiceBackup(t *testing.T) { if br.Format != command.BackupRequest_BACKUP_REQUEST_FORMAT_BINARY { t.Fatalf("wrong backup format requested") } - dst.Write(testData) + dst.Write(mustGZIPCompress(testData)) return nil } @@ -618,6 +619,18 @@ func removeNodeRequest(id string) *command.RemoveNodeRequest { } } +func mustGZIPCompress(b []byte) []byte { + var buf bytes.Buffer + zw := gzip.NewWriter(&buf) + if _, err := zw.Write(b); err != nil { + panic(fmt.Sprintf("failed to compress data: %s", err.Error())) + } + if err := zw.Close(); err != nil { + panic(fmt.Sprintf("failed to close gzip writer: %s", err.Error())) + } + return buf.Bytes() +} + func asJSON(v interface{}) string { enc := encoding.Encoder{} b, err := enc.JSONMarshal(v)