github.com/aavshr/aws-sdk-go@v1.41.3/service/glacier/customizations_test.go (about) 1 //go:build !integration 2 // +build !integration 3 4 package glacier_test 5 6 import ( 7 "bytes" 8 "testing" 9 10 "github.com/aavshr/aws-sdk-go/aws" 11 "github.com/aavshr/aws-sdk-go/aws/awsutil" 12 "github.com/aavshr/aws-sdk-go/awstesting/unit" 13 "github.com/aavshr/aws-sdk-go/service/glacier" 14 ) 15 16 var ( 17 payloadBuf = func() *bytes.Reader { 18 buf := make([]byte, 5767168) // 5.5MB buffer 19 for i := range buf { 20 buf[i] = '0' // Fill with zero characters 21 } 22 return bytes.NewReader(buf) 23 }() 24 25 svc = glacier.New(unit.Session) 26 ) 27 28 func TestCustomizations(t *testing.T) { 29 req, _ := svc.UploadArchiveRequest(&glacier.UploadArchiveInput{ 30 VaultName: aws.String("vault"), 31 Body: payloadBuf, 32 }) 33 err := req.Build() 34 if err != nil { 35 t.Errorf("expect no err, got %v", err) 36 } 37 38 // Sets API version 39 if e, a := req.ClientInfo.APIVersion, req.HTTPRequest.Header.Get("x-amz-glacier-version"); e != a { 40 t.Errorf("expect %v, got %v", e, a) 41 } 42 43 // Sets Account ID 44 v, _ := awsutil.ValuesAtPath(req.Params, "AccountId") 45 if e, a := "-", *(v[0].(*string)); e != a { 46 t.Errorf("expect %v, got %v", e, a) 47 } 48 49 // Computes checksums 50 linear := "68aff0c5a91aa0491752bfb96e3fef33eb74953804f6a2f7b708d5bcefa8ff6b" 51 tree := "154e26c78fd74d0c2c9b3cc4644191619dc4f2cd539ae2a74d5fd07957a3ee6a" 52 if e, a := linear, req.HTTPRequest.Header.Get("x-amz-content-sha256"); e != a { 53 t.Errorf("expect %v, got %v", e, a) 54 } 55 if e, a := tree, req.HTTPRequest.Header.Get("x-amz-sha256-tree-hash"); e != a { 56 t.Errorf("expect %v, got %v", e, a) 57 } 58 } 59 60 func TestShortcircuitTreehash(t *testing.T) { 61 req, _ := svc.UploadArchiveRequest(&glacier.UploadArchiveInput{ 62 VaultName: aws.String("vault"), 63 Body: payloadBuf, 64 Checksum: aws.String("000"), 65 }) 66 err := req.Build() 67 if err != nil { 68 t.Errorf("expect no err, got %v", err) 69 } 70 71 if e, a := "000", req.HTTPRequest.Header.Get("x-amz-sha256-tree-hash"); e != a { 72 t.Errorf("expect %v, got %v", e, a) 73 } 74 } 75 76 func TestFillAccountIDWithNilStruct(t *testing.T) { 77 req, _ := svc.ListVaultsRequest(nil) 78 err := req.Build() 79 if err != nil { 80 t.Errorf("expect no err, got %v", err) 81 } 82 83 empty := "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855" 84 85 // Sets Account ID 86 v, _ := awsutil.ValuesAtPath(req.Params, "AccountId") 87 if e, a := "-", *(v[0].(*string)); e != a { 88 t.Errorf("expect %v, got %v", e, a) 89 } 90 91 // Does not set tree hash 92 if e, a := empty, req.HTTPRequest.Header.Get("x-amz-content-sha256"); e != a { 93 t.Errorf("expect %v, got %v", e, a) 94 } 95 if e, a := "", req.HTTPRequest.Header.Get("x-amz-sha256-tree-hash"); e != a { 96 t.Errorf("expect %v, got %v", e, a) 97 } 98 } 99 100 func TestHashOnce(t *testing.T) { 101 req, _ := svc.UploadArchiveRequest(&glacier.UploadArchiveInput{ 102 VaultName: aws.String("vault"), 103 Body: payloadBuf, 104 }) 105 req.HTTPRequest.Header.Set("X-Amz-Sha256-Tree-Hash", "0") 106 107 err := req.Build() 108 if err != nil { 109 t.Errorf("expect no err, got %v", err) 110 } 111 112 if e, a := "0", req.HTTPRequest.Header.Get("x-amz-sha256-tree-hash"); e != a { 113 t.Errorf("expect %v, got %v", e, a) 114 } 115 }