utils/regex-match_test.cc

/*
 * Copyright (C) 2018 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "utils/regex-match.h"

#include <memory>

#include "utils/utf8/unilib.h"
#include "gmock/gmock.h"
#include "gtest/gtest.h"

namespace libtextclassifier3 {
namespace {

class LuaVerifierTest : public testing::Test {
 protected:
  LuaVerifierTest() : INIT_UNILIB_FOR_TESTING(unilib_) {}
  UniLib unilib_;
};

#ifdef TC3_UNILIB_ICU
TEST_F(LuaVerifierTest, HandlesSimpleVerification) {
  EXPECT_TRUE(VerifyMatch(/*context=*/"", /*matcher=*/nullptr, "return true;"));
}

TEST_F(LuaVerifierTest, HandlesCustomVerification) {
  UnicodeText pattern = UTF8ToUnicodeText("(\\d{16})",
                                          /*do_copy=*/true);
  UnicodeText message = UTF8ToUnicodeText("cc: 4012888888881881",
                                          /*do_copy=*/true);
  const std::string verifier = R"(
function luhn(candidate)
    local sum = 0
    local num_digits = string.len(candidate)
    local parity = num_digits % 2
    for pos = 1,num_digits do
      d = tonumber(string.sub(candidate, pos, pos))
      if pos % 2 ~= parity then
        d = d * 2
      end
      if d > 9 then
        d = d - 9
      end
      sum = sum + d
    end
    return (sum % 10) == 0
end
return luhn(match[1].text);
  )";
  auto regex_pattern = unilib_.CreateRegexPattern(pattern);
  ASSERT_TRUE(regex_pattern != nullptr);
  auto matcher = regex_pattern->Matcher(message);
  ASSERT_TRUE(matcher != nullptr);
  int status = UniLib::RegexMatcher::kNoError;
  ASSERT_TRUE(matcher->Find(&status) &&
              status == UniLib::RegexMatcher::kNoError);

  EXPECT_TRUE(VerifyMatch(message.ToUTF8String(), matcher.get(), verifier));
}
#endif

}  // namespace
}  // namespace libtextclassifier3