summaryrefslogtreecommitdiff
path: root/internal/stupidgcm/common_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/stupidgcm/common_test.go')
-rw-r--r--internal/stupidgcm/common_test.go17
1 files changed, 9 insertions, 8 deletions
diff --git a/internal/stupidgcm/common_test.go b/internal/stupidgcm/common_test.go
index 8123ce2..a8080ca 100644
--- a/internal/stupidgcm/common_test.go
+++ b/internal/stupidgcm/common_test.go
@@ -162,25 +162,26 @@ func testCorruption(t *testing.T, c cipher.AEAD) {
}
}
-type Wiper interface {
- Wipe()
-}
-
func testWipe(t *testing.T, c cipher.AEAD) {
switch c2 := c.(type) {
case *StupidGCM:
c2.Wipe()
- if c2.key != nil {
- t.Fatal("key is not nil")
+ if !c2.Wiped() {
+ t.Error("c2.wiped is not set")
+ }
+ for _, v := range c2.key {
+ if v != 0 {
+ t.Fatal("c2._key is not zeroed")
+ }
}
case *stupidChacha20poly1305:
c2.Wipe()
- if !c2.wiped {
+ if !c2.Wiped() {
t.Error("c2.wiped is not set")
}
for _, v := range c2.key {
if v != 0 {
- t.Fatal("c2.key is not zeroed")
+ t.Fatal("c2._key is not zeroed")
}
}
case *stupidXchacha20poly1305: