Mypal/intl/lwbrk/gtest/TestLineBreak.cpp

324 lines
9.2 KiB
C++

/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include <stdio.h>
#include "nsXPCOM.h"
#include "nsIComponentManager.h"
#include "nsISupports.h"
#include "nsServiceManagerUtils.h"
#include "nsILineBreaker.h"
#include "nsIWordBreaker.h"
#include "nsLWBrkCIID.h"
#include "nsString.h"
#include "gtest/gtest.h"
NS_DEFINE_CID(kLBrkCID, NS_LBRK_CID);
NS_DEFINE_CID(kWBrkCID, NS_WBRK_CID);
static char teng1[] =
// 1 2 3 4 5 6 7
//01234567890123456789012345678901234567890123456789012345678901234567890123456789
"This is a test to test(reasonable) line break. This 0.01123 = 45 x 48.";
static uint32_t lexp1[] = {
4,7,9,14,17,34,39,40,41,42,49,54,62,64,67,69,73
};
static uint32_t wexp1[] = {
4,5,7,8,9,10,14,15,17,18,22,23,33,34,35,39,43,48,49,50,54,55,56,57,62,63,
64,65,67,68,69,70,72
};
static char teng2[] =
// 1 2 3 4 5 6 7
//01234567890123456789012345678901234567890123456789012345678901234567890123456789
"()((reasonab(l)e) line break. .01123=45x48.";
static uint32_t lexp2[] = {
17,22,23,30,44
};
static uint32_t wexp2[] = {
4,12,13,14,15,16,17,18,22,24,29,30,31,32,37,38,43
};
static char teng3[] =
// 1 2 3 4 5 6 7
//01234567890123456789012345678901234567890123456789012345678901234567890123456789
"It's a test to test(ronae ) line break....";
static uint32_t lexp3[] = {
4,6,11,14,25,27,32,42
};
static uint32_t wexp3[] = {
2,3,4,5,6,7,11,12,14,15,19,20,25,26,27,28,32,33,38
};
static char ruler1[] =
" 1 2 3 4 5 6 7 ";
static char ruler2[] =
"0123456789012345678901234567890123456789012345678901234567890123456789012";
bool
Check(const char* in, const uint32_t* out, uint32_t outlen, uint32_t i,
uint32_t res[256])
{
bool ok = true;
if (i != outlen) {
ok = false;
printf("WARNING!!! return size wrong, expect %d but got %d \n",
outlen, i);
}
for (uint32_t j = 0; j < i; j++) {
if (j < outlen) {
if (res[j] != out[j]) {
ok = false;
printf("[%d] expect %d but got %d\n", j, out[j], res[j]);
}
} else {
ok = false;
printf("[%d] additional %d\n", j, res[j]);
}
}
if (!ok) {
printf("string = \n%s\n", in);
printf("%s\n", ruler1);
printf("%s\n", ruler2);
printf("Expect = \n");
for (uint32_t j = 0; j < outlen; j++) {
printf("%d,", out[j]);
}
printf("\nResult = \n");
for (uint32_t j = 0; j < i; j++) {
printf("%d,", res[j]);
}
printf("\n");
}
return ok;
}
bool
TestASCIILB(nsILineBreaker *lb,
const char* in,
const uint32_t* out, uint32_t outlen)
{
NS_ConvertASCIItoUTF16 eng1(in);
uint32_t i;
uint32_t res[256];
int32_t curr;
for (i = 0, curr = 0;
curr != NS_LINEBREAKER_NEED_MORE_TEXT && i < 256;
i++) {
curr = lb->Next(eng1.get(), eng1.Length(), curr);
res[i] = curr != NS_LINEBREAKER_NEED_MORE_TEXT ? curr : eng1.Length();
}
return Check(in, out, outlen, i, res);
}
bool
TestASCIIWB(nsIWordBreaker *lb,
const char* in,
const uint32_t* out, uint32_t outlen)
{
NS_ConvertASCIItoUTF16 eng1(in);
uint32_t i;
uint32_t res[256];
int32_t curr = 0;
for (i = 0, curr = lb->NextWord(eng1.get(), eng1.Length(), curr);
curr != NS_WORDBREAKER_NEED_MORE_TEXT && i < 256;
curr = lb->NextWord(eng1.get(), eng1.Length(), curr), i++) {
res [i] = curr != NS_WORDBREAKER_NEED_MORE_TEXT ? curr : eng1.Length();
}
return Check(in, out, outlen, i, res);
}
TEST(LineBreak, LineBreaker)
{
nsILineBreaker *t = nullptr;
nsresult res = CallGetService(kLBrkCID, &t);
ASSERT_TRUE(NS_SUCCEEDED(res) && t);
NS_IF_RELEASE(t);
res = CallGetService(kLBrkCID, &t);
ASSERT_TRUE(NS_SUCCEEDED(res) && t);
ASSERT_TRUE(TestASCIILB(t, teng1, lexp1, sizeof(lexp1) / sizeof(uint32_t)));
ASSERT_TRUE(TestASCIILB(t, teng2, lexp2, sizeof(lexp2) / sizeof(uint32_t)));
ASSERT_TRUE(TestASCIILB(t, teng3, lexp3, sizeof(lexp3) / sizeof(uint32_t)));
NS_RELEASE(t);
}
TEST(LineBreak, WordBreaker)
{
nsIWordBreaker *t = nullptr;
nsresult res = CallGetService(kWBrkCID, &t);
ASSERT_TRUE(NS_SUCCEEDED(res) && t);
NS_IF_RELEASE(t);
res = CallGetService(kWBrkCID, &t);
ASSERT_TRUE(NS_SUCCEEDED(res) && t);
ASSERT_TRUE(TestASCIIWB(t, teng1, wexp1, sizeof(wexp1) / sizeof(uint32_t)));
ASSERT_TRUE(TestASCIIWB(t, teng2, wexp2, sizeof(wexp2) / sizeof(uint32_t)));
ASSERT_TRUE(TestASCIIWB(t, teng3, wexp3, sizeof(wexp3) / sizeof(uint32_t)));
NS_RELEASE(t);
}
// 012345678901234
static const char wb0[] = "T";
static const char wb1[] = "h";
static const char wb2[] = "is is a int";
static const char wb3[] = "ernationali";
static const char wb4[] = "zation work.";
static const char* wb[] = { wb0, wb1, wb2, wb3, wb4 };
void
TestPrintWordWithBreak()
{
uint32_t numOfFragment = sizeof(wb) / sizeof(char*);
nsIWordBreaker* wbk = nullptr;
CallGetService(kWBrkCID, &wbk);
nsAutoString result;
for (uint32_t i = 0; i < numOfFragment; i++) {
NS_ConvertASCIItoUTF16 fragText(wb[i]);
int32_t cur = 0;
cur = wbk->NextWord(fragText.get(), fragText.Length(), cur);
uint32_t start = 0;
for (uint32_t j = 0; cur != NS_WORDBREAKER_NEED_MORE_TEXT; j++) {
result.Append(Substring(fragText, start, cur - start));
result.Append('^');
start = (cur >= 0 ? cur : cur - start);
cur = wbk->NextWord(fragText.get(), fragText.Length(), cur);
}
result.Append(Substring(fragText, fragText.Length() - start));
if (i != numOfFragment - 1) {
NS_ConvertASCIItoUTF16 nextFragText(wb[i+1]);
bool canBreak = true;
canBreak = wbk->BreakInBetween(fragText.get(),
fragText.Length(),
nextFragText.get(),
nextFragText.Length());
if (canBreak) {
result.Append('^');
}
fragText.Assign(nextFragText);
}
}
ASSERT_STREQ("is^ ^is^ ^a^ ^ is a intzation^ ^work^ation work.",
NS_ConvertUTF16toUTF8(result).get());
NS_IF_RELEASE(wbk);
}
void
TestFindWordBreakFromPosition(uint32_t fragN, uint32_t offset,
const char* expected)
{
uint32_t numOfFragment = sizeof(wb) / sizeof(char*);
nsIWordBreaker* wbk = nullptr;
CallGetService(kWBrkCID, &wbk);
NS_ConvertASCIItoUTF16 fragText(wb[fragN]);
nsWordRange res = wbk->FindWord(fragText.get(), fragText.Length(), offset);
bool canBreak;
nsAutoString result(Substring(fragText, res.mBegin, res.mEnd-res.mBegin));
if ((uint32_t)fragText.Length() == res.mEnd) {
// if we hit the end of the fragment
nsAutoString curFragText = fragText;
for(uint32_t p = fragN +1; p < numOfFragment ;p++)
{
NS_ConvertASCIItoUTF16 nextFragText(wb[p]);
canBreak = wbk->BreakInBetween(curFragText.get(),
curFragText.Length(),
nextFragText.get(),
nextFragText.Length());
if (canBreak) {
break;
}
nsWordRange r = wbk->FindWord(nextFragText.get(), nextFragText.Length(),
0);
result.Append(Substring(nextFragText, r.mBegin, r.mEnd - r.mBegin));
if ((uint32_t)nextFragText.Length() != r.mEnd) {
break;
}
nextFragText.Assign(curFragText);
}
}
if (0 == res.mBegin) {
// if we hit the beginning of the fragment
nsAutoString curFragText = fragText;
for (uint32_t p = fragN; p > 0; p--) {
NS_ConvertASCIItoUTF16 prevFragText(wb[p-1]);
canBreak = wbk->BreakInBetween(prevFragText.get(),
prevFragText.Length(),
curFragText.get(),
curFragText.Length());
if (canBreak) {
break;
}
nsWordRange r = wbk->FindWord(prevFragText.get(), prevFragText.Length(),
prevFragText.Length());
result.Insert(Substring(prevFragText, r.mBegin, r.mEnd - r.mBegin), 0);
if (0 != r.mBegin) {
break;
}
prevFragText.Assign(curFragText);
}
}
ASSERT_STREQ(expected, NS_ConvertUTF16toUTF8(result).get())
<< "FindWordBreakFromPosition(" << fragN << ", " << offset << ")";
NS_IF_RELEASE(wbk);
}
TEST(LineBreak, WordBreakUsage)
{
TestPrintWordWithBreak();
TestFindWordBreakFromPosition(0, 0, "This");
TestFindWordBreakFromPosition(1, 0, "his");
TestFindWordBreakFromPosition(2, 0, "is");
TestFindWordBreakFromPosition(2, 1, "is");
TestFindWordBreakFromPosition(2, 9, " ");
TestFindWordBreakFromPosition(2, 10, "internationalization");
TestFindWordBreakFromPosition(3, 4, "ernationalization");
TestFindWordBreakFromPosition(3, 8, "ernationalization");
TestFindWordBreakFromPosition(4, 6, " ");
TestFindWordBreakFromPosition(4, 7, "work");
}