Skip to content

Commit

Permalink
Test functions working with ciphertext.
Browse files Browse the repository at this point in the history
  • Loading branch information
armfazh committed Apr 24, 2024
1 parent 64431bb commit a4252c7
Showing 1 changed file with 20 additions and 2 deletions.
22 changes: 20 additions & 2 deletions abe/cpabe/tkn20/format_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -46,15 +46,33 @@ func TestCiphertext(t *testing.T) {
if err != nil {
t.Fatalf("Unable to read ciphertext data")
}
policyKey, err := os.ReadFile("testdata/attributeKey")
attributeKey, err := os.ReadFile("testdata/attributeKey")
if err != nil {
t.Fatalf("Unable to read secret key")
}
sk := AttributeKey{}
err = sk.UnmarshalBinary(policyKey)
err = sk.UnmarshalBinary(attributeKey)
if err != nil {
t.Fatalf("unable to parse secret key")
}
attrs := Attributes{}
attrs.FromMap(map[string]string{"country": "NL", "EU": "true"})
if !attrs.CouldDecrypt(ciphertext) {
t.Fatal("these attributes will be unable to decrypt message")
}
policy := Policy{}
err = policy.FromString("EU: true")
if err != nil {
t.Fatal("error creating policy from string")
}
gotPolicy := new(Policy)
err = gotPolicy.ExtractFromCiphertext(ciphertext)
if err != nil {
t.Fatal("error extracting policy from ciphertext")
}
if !policy.Equal(gotPolicy) {
t.Fatal("ciphertext's policy mismatches the original policy")
}
msg, err := sk.Decrypt(ciphertext)
if err != nil {
t.Fatal("unable to decrypt message")
Expand Down

0 comments on commit a4252c7

Please sign in to comment.