41 lines
946 B
Go
41 lines
946 B
Go
|
package s3crypto_test
|
||
|
|
||
|
import (
|
||
|
"io/ioutil"
|
||
|
"strings"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/aws/aws-sdk-go/service/s3/s3crypto"
|
||
|
)
|
||
|
|
||
|
func TestCryptoReadCloserRead(t *testing.T) {
|
||
|
expectedStr := "HELLO WORLD "
|
||
|
str := strings.NewReader(expectedStr)
|
||
|
rc := &s3crypto.CryptoReadCloser{Body: ioutil.NopCloser(str), Decrypter: str}
|
||
|
|
||
|
b, err := ioutil.ReadAll(rc)
|
||
|
if err != nil {
|
||
|
t.Errorf("expected no error, but received %v", err)
|
||
|
}
|
||
|
if expectedStr != string(b) {
|
||
|
t.Errorf("expected %s, but received %s", expectedStr, string(b))
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestCryptoReadCloserClose(t *testing.T) {
|
||
|
data := "HELLO WORLD "
|
||
|
expectedStr := ""
|
||
|
|
||
|
str := strings.NewReader(data)
|
||
|
rc := &s3crypto.CryptoReadCloser{Body: ioutil.NopCloser(str), Decrypter: str}
|
||
|
rc.Close()
|
||
|
|
||
|
b, err := ioutil.ReadAll(rc)
|
||
|
if err != nil {
|
||
|
t.Errorf("expected no error, but received %v", err)
|
||
|
}
|
||
|
if expectedStr != string(b) {
|
||
|
t.Errorf("expected %s, but received %s", expectedStr, string(b))
|
||
|
}
|
||
|
}
|