1 /*
2 * Copyright 2023 Google LLC.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * https://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include "act/fake_act.h"
17
18 #include <gmock/gmock.h>
19 #include <gtest/gtest.h>
20
21 #include <memory>
22 #include <string>
23 #include <tuple>
24 #include <utility>
25 #include <vector>
26
27 #include "private_join_and_compute/util/status.inc"
28 #include "private_join_and_compute/util/status_testing.inc"
29
30 namespace private_join_and_compute {
31 namespace anonymous_counting_tokens {
32 namespace {
33
TEST(FakeActTest,FingerprintsAreEqualExactlyWhenMessagesAreEqual)34 TEST(FakeActTest, FingerprintsAreEqualExactlyWhenMessagesAreEqual) {
35 auto fake_act = FakeAnonymousCountingTokens::Create();
36
37 std::vector<std::string> messages_1 = {"m1", "m2"};
38 std::vector<std::string> fingerprints_1;
39
40 std::vector<std::string> messages_2 = {"m3", "m1"};
41 std::vector<std::string> fingerprints_2;
42
43 // Empty parameters for testing the fake.
44 SchemeParameters scheme_parameters;
45 ClientPublicParameters client_public_parameters;
46 ClientPrivateParameters client_private_parameters;
47 ServerPublicParameters server_public_parameters;
48
49 ASSERT_OK_AND_ASSIGN(
50 std::tie(fingerprints_1, std::ignore, std::ignore),
51 fake_act->GenerateTokensRequest(
52 messages_1, scheme_parameters, client_public_parameters,
53 client_private_parameters, server_public_parameters));
54 ASSERT_OK_AND_ASSIGN(
55 std::tie(fingerprints_2, std::ignore, std::ignore),
56 fake_act->GenerateTokensRequest(
57 messages_2, scheme_parameters, client_public_parameters,
58 client_private_parameters, server_public_parameters));
59
60 // Only the fingerprints for "m1" should be the same.
61 EXPECT_EQ(fingerprints_1[0], fingerprints_2[1]);
62 EXPECT_NE(fingerprints_1[0], fingerprints_2[0]);
63 EXPECT_NE(fingerprints_1[1], fingerprints_2[1]);
64 EXPECT_NE(fingerprints_1[1], fingerprints_2[0]);
65 }
66
TEST(FakeActTest,DifferentTokensAreNotEqual)67 TEST(FakeActTest, DifferentTokensAreNotEqual) {
68 auto fake_act = FakeAnonymousCountingTokens::Create();
69
70 std::vector<std::string> messages = {"m1", "m2"};
71 // Empty parameters and messages for testing the fake.
72 TokensRequest tokens_request;
73 TokensRequestPrivateState tokens_request_private_state;
74 TokensResponse tokens_response;
75 SchemeParameters scheme_parameters;
76 ClientPublicParameters client_public_parameters;
77 ClientPrivateParameters client_private_parameters;
78 ServerPublicParameters server_public_parameters;
79
80 ASSERT_OK_AND_ASSIGN(
81 std::vector<Token> tokens,
82 fake_act->RecoverTokens(
83 messages, tokens_request, tokens_request_private_state,
84 tokens_response, scheme_parameters, client_public_parameters,
85 client_private_parameters, server_public_parameters));
86
87 EXPECT_NE(tokens[0].SerializeAsString(), tokens[1].SerializeAsString());
88 }
89
90 } // namespace
91 } // namespace anonymous_counting_tokens
92
93 } // namespace private_join_and_compute
94