X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/0bd1c28bed9a0756c61037947d5a9dccd5066f00..b6f73f7acffa9b2f37c78826ff6cee89987a7edc:/services/keepstore/perms_test.go

diff --git a/services/keepstore/perms_test.go b/services/keepstore/perms_test.go
index d1c6b50496..d0081cd010 100644
--- a/services/keepstore/perms_test.go
+++ b/services/keepstore/perms_test.go
@@ -5,7 +5,7 @@ import (
 	"time"
 )
 
-var (
+const (
 	known_hash    = "acbd18db4cc2f85cedef654fccc4a4d8"
 	known_locator = known_hash + "+3"
 	known_token   = "hocfupkn2pjhrpgp2vxv8rsku7tvtx49arbc9s4bvu7p7wxqvk"
@@ -18,7 +18,8 @@ var (
 		"786u5rw2a9gx743dj3fgq2irk"
 	known_signature      = "257f3f5f5f0a4e4626a18fc74bd42ec34dcb228a"
 	known_timestamp      = "7fffffff"
-	known_signed_locator = known_locator + "+A" + known_signature + "@" + known_timestamp
+	known_sig_hint       = "+A" + known_signature + "@" + known_timestamp
+	known_signed_locator = known_locator + known_sig_hint
 )
 
 func TestSignLocator(t *testing.T) {
@@ -43,14 +44,34 @@ func TestVerifySignature(t *testing.T) {
 	}
 }
 
+func TestVerifySignatureExtraHints(t *testing.T) {
+	PermissionSecret = []byte(known_key)
+	defer func() { PermissionSecret = nil }()
+
+	if !VerifySignature(known_locator + "+K@xyzzy" + known_sig_hint, known_token) {
+		t.Fatal("Verify cannot handle hint before permission signature")
+	}
+
+	if !VerifySignature(known_locator + known_sig_hint + "+Zfoo", known_token) {
+		t.Fatal("Verify cannot handle hint after permission signature")
+	}
+
+	if !VerifySignature(known_locator + "+K@xyzzy" + known_sig_hint + "+Zfoo", known_token) {
+		t.Fatal("Verify cannot handle hints around permission signature")
+	}
+}
+
 // The size hint on the locator string should not affect signature validation.
 func TestVerifySignatureWrongSize(t *testing.T) {
 	PermissionSecret = []byte(known_key)
 	defer func() { PermissionSecret = nil }()
 
-	signed_locator_wrong_size := known_hash + "+999999+A" + known_signature + "@" + known_timestamp
-	if !VerifySignature(signed_locator_wrong_size, known_token) {
-		t.Fail()
+	if !VerifySignature(known_hash + "+999999" + known_sig_hint, known_token) {
+		t.Fatal("Verify cannot handle incorrect size hint")
+	}
+
+	if !VerifySignature(known_hash + known_sig_hint, known_token) {
+		t.Fatal("Verify cannot handle missing size hint")
 	}
 }