tesseract/unittest/progress_test.cc

156 lines
4.8 KiB
C++
Raw Normal View History

///////////////////////////////////////////////////////////////////////
// File: progress_test.cc
// Description: Progress reporting API Test for Tesseract.
// Author: Jaroslaw Kubik
//
// 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.
///////////////////////////////////////////////////////////////////////
// expects clone of tessdata_fast repo in ../../tessdata_fast
2020-12-29 01:36:04 +08:00
#include "include_gunit.h"
#include <tesseract/baseapi.h>
#include <tesseract/ocrclass.h>
#include <allheaders.h>
2020-12-29 01:36:04 +08:00
#include "gmock/gmock.h"
#include <fstream>
#include <iostream>
#include <locale>
2021-03-13 05:06:34 +08:00
#include <memory> // std::unique_ptr
#include <string>
2020-12-29 01:36:04 +08:00
#include <time.h>
namespace tesseract {
class QuickTest : public testing::Test {
2021-03-13 05:06:34 +08:00
protected:
virtual void SetUp() {
start_time_ = time(nullptr);
}
virtual void TearDown() {
const time_t end_time = time(nullptr);
EXPECT_TRUE(end_time - start_time_ <= 25)
2021-03-13 05:06:34 +08:00
<< "The test took too long - " << ::testing::PrintToString(end_time - start_time_);
}
time_t start_time_;
};
class ClassicMockProgressSink {
2021-03-13 05:06:34 +08:00
public:
MOCK_METHOD1(classicProgress, bool(int));
MOCK_METHOD1(cancel, bool(int));
ETEXT_DESC monitor;
ClassicMockProgressSink() {
monitor.progress_callback = [](int progress, int, int, int, int) -> bool {
return instance->classicProgress(progress);
};
2021-03-13 05:06:34 +08:00
monitor.cancel = [](void *ths, int words) -> bool {
return ((ClassicMockProgressSink *)ths)->cancel(words);
};
monitor.cancel_this = this;
instance = this;
}
2021-03-13 05:06:34 +08:00
static ClassicMockProgressSink *instance;
};
2021-03-13 05:06:34 +08:00
ClassicMockProgressSink *ClassicMockProgressSink::instance = nullptr;
class NewMockProgressSink : public ClassicMockProgressSink {
2021-03-13 05:06:34 +08:00
public:
MOCK_METHOD1(progress, bool(int));
NewMockProgressSink() {
2021-03-13 05:06:34 +08:00
monitor.progress_callback2 = [](ETEXT_DESC *ths, int, int, int, int) -> bool {
return ((NewMockProgressSink *)ths->cancel_this)->progress(ths->progress);
};
}
};
2021-03-13 05:06:34 +08:00
void ClassicProgressTester(const char *imgname, const char *tessdatadir, const char *lang) {
using ::testing::_;
using ::testing::AllOf;
using ::testing::AtLeast;
using ::testing::DoAll;
using ::testing::Gt;
using ::testing::Le;
using ::testing::Return;
using ::testing::SaveArg;
auto api = std::make_unique<tesseract::TessBaseAPI>();
2021-03-13 05:06:34 +08:00
ASSERT_FALSE(api->Init(tessdatadir, lang)) << "Could not initialize tesseract.";
Pix *image = pixRead(imgname);
ASSERT_TRUE(image != nullptr) << "Failed to read test image.";
api->SetImage(image);
ClassicMockProgressSink progressSink;
int currentProgress = -1;
2021-03-13 05:06:34 +08:00
EXPECT_CALL(progressSink, classicProgress(AllOf(Gt<int &>(currentProgress), Le(100))))
.Times(AtLeast(5))
.WillRepeatedly(DoAll(SaveArg<0>(&currentProgress), Return(false)));
2021-03-13 05:06:34 +08:00
EXPECT_CALL(progressSink, cancel(_)).Times(AtLeast(5)).WillRepeatedly(Return(false));
EXPECT_EQ(api->Recognize(&progressSink.monitor), false);
EXPECT_GE(currentProgress, 50) << "The reported progress did not reach 50%";
api->End();
pixDestroy(&image);
}
2021-03-13 05:06:34 +08:00
void NewProgressTester(const char *imgname, const char *tessdatadir, const char *lang) {
using ::testing::_;
using ::testing::AllOf;
using ::testing::AtLeast;
using ::testing::DoAll;
using ::testing::Gt;
using ::testing::Le;
using ::testing::Return;
using ::testing::SaveArg;
auto api = std::make_unique<tesseract::TessBaseAPI>();
2021-03-13 05:06:34 +08:00
ASSERT_FALSE(api->Init(tessdatadir, lang)) << "Could not initialize tesseract.";
Pix *image = pixRead(imgname);
ASSERT_TRUE(image != nullptr) << "Failed to read test image.";
api->SetImage(image);
NewMockProgressSink progressSink;
int currentProgress = -1;
EXPECT_CALL(progressSink, classicProgress(_)).Times(0);
2021-03-13 05:06:34 +08:00
EXPECT_CALL(progressSink, progress(AllOf(Gt<int &>(currentProgress), Le(100))))
.Times(AtLeast(5))
.WillRepeatedly(DoAll(SaveArg<0>(&currentProgress), Return(false)));
2021-03-13 05:06:34 +08:00
EXPECT_CALL(progressSink, cancel(_)).Times(AtLeast(5)).WillRepeatedly(Return(false));
EXPECT_EQ(api->Recognize(&progressSink.monitor), false);
EXPECT_GE(currentProgress, 50) << "The reported progress did not reach 50%";
api->End();
pixDestroy(&image);
}
TEST(QuickTest, ClassicProgressReporting) {
2021-03-13 05:06:34 +08:00
ClassicProgressTester(TESTING_DIR "/phototest.tif", TESSDATA_DIR "_fast", "eng");
}
TEST(QuickTest, NewProgressReporting) {
NewProgressTester(TESTING_DIR "/phototest.tif", TESSDATA_DIR "_fast", "eng");
}
2021-03-13 05:06:34 +08:00
} // namespace tesseract