Export 7 errors so users can act differently on different type of errors

1 files changed, 3 insertions, 3 deletions

diff --git a/crypt_test.go b/crypt_test.go
index cb0b160..0796482 100644
--- a/crypt_test.go
+++ b/crypt_test.go
@@ -21,16 +21,16 @@ import (
 func TestEncrypt(t *testing.T) {
 	f, err := OpenFile(filepath.Join("test", "encryptSHA1.xlsx"), Options{Password: "password"})
 	assert.NoError(t, err)
-	assert.EqualError(t, f.SaveAs(filepath.Join("test", "BadEncrypt.xlsx"), Options{Password: "password"}), "not support encryption currently")
+	assert.EqualError(t, f.SaveAs(filepath.Join("test", "BadEncrypt.xlsx"), Options{Password: "password"}), ErrEncrypt.Error())
 	assert.NoError(t, f.Close())
 }
 
 func TestEncryptionMechanism(t *testing.T) {
 	mechanism, err := encryptionMechanism([]byte{3, 0, 3, 0})
 	assert.Equal(t, mechanism, "extensible")
-	assert.EqualError(t, err, "unsupport encryption mechanism")
+	assert.EqualError(t, err, ErrUnsupportEncryptMechanism.Error())
 	_, err = encryptionMechanism([]byte{})
-	assert.EqualError(t, err, "unknown encryption mechanism")
+	assert.EqualError(t, err, ErrUnknownEncryptMechanism.Error())
 }
 
 func TestHashing(t *testing.T) {