m

Author: ajewellamz

DynamoDbEncryption/dafny/DynamoDbEncryption/test/Beacon.dfy

@@ -32,6 +32,10 @@ module TestBaseBeacon {
     expect bytes == [0x27, 0x93, 0x93, 0x8b, 0x26, 0xe9, 0x52, 0x7e];
     var str :- expect b.hash([1,2,3], key := [1,2], bucket := []);
     expect str == "7e";
+    str :- expect b.hash([1,2,3], key := [1,2], bucket := [1]);
+    expect str == "1d";
+    str :- expect b.hash([1,2,3], key := [1,2], bucket := [2]);
+    expect str == "ef";
     bytes :- expect bb.getHmac([], key := [1,2]);
     expect bytes[7] == 0x80;
     str :- expect b.hash([], key := [1,2], bucket := []);
@@ -365,7 +369,7 @@ module TestBaseBeacon {
                     ];
   }
 
-  method GetBeaconValue(name : string, key : Bytes, value : string, length : BeaconLength) returns (output : string)
+  method GetBeaconValue(name : string, key : Bytes, value : string, length : BeaconLength, bucket : Bytes := []) returns (output : string)
   {
     var info :- expect UTF8.Encode("AWS_DBE_SCAN_BEACON" + name);
     var client :- expect Primitives.AtomicPrimitives();
@@ -379,6 +383,7 @@ module TestBaseBeacon {
                                   ));
 
     var data :- expect UTF8.Encode(value);
+    data := data + bucket;
     var input := Prim.HMacInput (
       digestAlgorithm := Prim.SHA_384,
       key := key,