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  }