chromium/chrome/browser/ash/policy/enrollment/psm/rlwe_test_support_unittest.cc

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ash/policy/enrollment/psm/rlwe_test_support.h"

#include "device_management_backend.pb.h"
#include "private_membership_rlwe.pb.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/private_membership/src/private_membership_rlwe_client.h"

namespace policy {

class RlweTestSupportTest
    : public testing::Test,
      public testing::WithParamInterface</*is_member*/ bool> {};

// Test that the RlweClient created via CreateClientFactory matches the
// RlweTestCase loaded vid LoadTestCase.
TEST_P(RlweTestSupportTest, CaseMatchesFactory) {
  const bool is_member = GetParam();
  const auto test_case = psm::testing::LoadTestCase(is_member);
  auto rlwe_client = psm::testing::CreateClientFactory(is_member).Run(
      private_membership::rlwe::CROS_DEVICE_STATE,
      private_membership::rlwe::RlwePlaintextId());

  // Verify that the expected membership matches the test parameter.
  EXPECT_EQ(test_case.is_positive_membership_expected(), is_member);

  // Create and compare OPRF request.
  const auto status_or_orpf = rlwe_client->CreateOprfRequest();
  ASSERT_TRUE(status_or_orpf.ok());
  EXPECT_EQ(status_or_orpf.value().SerializeAsString(),
            test_case.expected_oprf_request().SerializeAsString());

  // Create and compare Query request.
  // This also test that the OPRF response meets the RlweClient's expectations.
  const auto status_or_query =
      rlwe_client->CreateQueryRequest(test_case.oprf_response());
  ASSERT_TRUE(status_or_query.ok());
  EXPECT_EQ(status_or_query.value().SerializeAsString(),
            test_case.expected_query_request().SerializeAsString());
}

INSTANTIATE_TEST_SUITE_P(RlweTestSupportTestSuite,
                         RlweTestSupportTest,
                         /* is_member*/ testing::Bool());

}  // namespace policy