You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
76 lines
2.2 KiB
76 lines
2.2 KiB
/* |
|
* 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
|
|
|