From e3c864e08db2875049ea46c487f4ad7ac4f2322b Mon Sep 17 00:00:00 2001 From: NejcZdovc Date: Tue, 15 Jan 2019 06:45:55 +0100 Subject: [PATCH] Fixes tests --- .../browser/publisher_info_database.cc | 40 +- .../publisher_info_database_unittest.cc | 372 ++++++++++++++++-- components/brave_rewards/browser/test_util.h | 209 ---------- 3 files changed, 355 insertions(+), 266 deletions(-) diff --git a/components/brave_rewards/browser/publisher_info_database.cc b/components/brave_rewards/browser/publisher_info_database.cc index 7de1927fc7d8..7530d511bc36 100644 --- a/components/brave_rewards/browser/publisher_info_database.cc +++ b/components/brave_rewards/browser/publisher_info_database.cc @@ -17,6 +17,7 @@ #include "sql/statement.h" #include "sql/transaction.h" #include "content_site.h" +#include "recurring_donation.h" namespace brave_rewards { @@ -236,8 +237,9 @@ bool PublisherInfoDatabase::InsertOrUpdatePublisherInfo( bool initialized = Init(); DCHECK(initialized); - if (!initialized) + if (!initialized || info.id.empty()) { return false; + } sql::Statement publisher_info_statement( GetDB().GetCachedStatement(SQL_FROM_HERE, @@ -409,35 +411,26 @@ bool PublisherInfoDatabase::InsertOrUpdateActivityInfo( bool initialized = Init(); DCHECK(initialized); - if (!initialized) + if (!initialized) { return false; + } - // Insert publisher if it doesn't exist - sql::Statement publisher_info_statement( - GetDB().GetCachedStatement(SQL_FROM_HERE, - "INSERT OR IGNORE INTO publisher_info " - "(publisher_id, verified, excluded, " - "name, url, provider, favIcon) " - "VALUES (?, ?, ?, ?, ?, ?, ?)")); - - publisher_info_statement.BindString(0, info.id); - publisher_info_statement.BindBool(1, info.verified); - publisher_info_statement.BindInt(2, static_cast(info.excluded)); - publisher_info_statement.BindString(3, info.name); - publisher_info_statement.BindString(4, info.url); - publisher_info_statement.BindString(5, info.provider); - publisher_info_statement.BindString(6, info.favicon_url); + LOG(ERROR) << "NEJC 1"; - if (!publisher_info_statement.Run()) { + if (!InsertOrUpdatePublisherInfo(info)) { return false; } + LOG(ERROR) << "NEJC 2"; + + LOG(ERROR) << info.year; + sql::Statement activity_info_insert( GetDB().GetCachedStatement(SQL_FROM_HERE, "INSERT OR REPLACE INTO activity_info " "(publisher_id, duration, score, percent, " - "weight, month, year, reconcile_stamp) " - "VALUES (?, ?, ?, ?, ?, ?, ?, ?)")); + "weight, month, year, reconcile_stamp, visits) " + "VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)")); activity_info_insert.BindString(0, info.id); activity_info_insert.BindInt64(1, static_cast(info.duration)); @@ -447,6 +440,9 @@ bool PublisherInfoDatabase::InsertOrUpdateActivityInfo( activity_info_insert.BindInt(5, info.month); activity_info_insert.BindInt(6, info.year); activity_info_insert.BindInt64(7, info.reconcile_stamp); + activity_info_insert.BindInt64(8, info.visits); + + LOG(ERROR) << "NEJC 3"; return activity_info_insert.Run(); } @@ -629,7 +625,7 @@ bool PublisherInfoDatabase::InsertOrUpdateMediaPublisherInfo( bool initialized = Init(); DCHECK(initialized); - if (!initialized) { + if (!initialized || media_key.empty() || publisher_id.empty()) { return false; } @@ -726,7 +722,7 @@ bool PublisherInfoDatabase::InsertOrUpdateRecurringDonation( bool initialized = Init(); DCHECK(initialized); - if (!initialized) { + if (!initialized || info.publisher_key.empty()) { return false; } diff --git a/components/brave_rewards/browser/publisher_info_database_unittest.cc b/components/brave_rewards/browser/publisher_info_database_unittest.cc index db5a3cbd28cb..14966c230bdb 100644 --- a/components/brave_rewards/browser/publisher_info_database_unittest.cc +++ b/components/brave_rewards/browser/publisher_info_database_unittest.cc @@ -14,7 +14,6 @@ #include "base/strings/utf_string_conversions.h" #include "sql/database.h" #include "sql/statement.h" -#include "sql/transaction.h" #include "third_party/sqlite/sqlite3.h" #include "testing/gmock/include/gmock/gmock.h" #include "testing/gtest/include/gtest/gtest.h" @@ -23,61 +22,51 @@ namespace brave_rewards { -// Test data directory, relative to source root -const base::FilePath::CharType kTestDataRelativePath[] = - FILE_PATH_LITERAL("brave/vendor/bat-native-ledger/test/data"); - class PublisherInfoDatabaseTest : public ::testing::Test { protected: PublisherInfoDatabaseTest() { - // You can do set-up work for each test here } ~PublisherInfoDatabaseTest() override { - // You can do clean-up work that doesn't throw exceptions here } - // If the constructor and destructor are not enough for setting up and - // cleaning up each test, you can use the following methods - - void SetUp() override { - // Code here will be called immediately after the constructor (right before - // each test) - - base::ScopedTempDir temp_dir; - ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); - ASSERT_TRUE(base::DirectoryExists(temp_dir.GetPath())); - base::FilePath db_file = - temp_dir.GetPath().AppendASCII("PublisherInfoDatabaseTest.db"); + sql::Database& GetDB() { + return publisher_info_database_->GetDB(); + } - base::FilePath temp_file; - ASSERT_TRUE( - base::CreateTemporaryFileInDir(temp_dir.GetPath(), &temp_file)); + void CreateTempDatabase(base::ScopedTempDir* temp_dir, + base::FilePath* db_file) { + ASSERT_TRUE(temp_dir->CreateUniqueTempDir()); + *db_file = temp_dir->GetPath().AppendASCII("PublisherInfoDatabaseTest.db"); + sql::Database::Delete(*db_file); - publisher_info_database_ = std::make_unique(temp_file); + publisher_info_database_ = + std::make_unique(*db_file); ASSERT_NE(publisher_info_database_, nullptr); } - void TearDown() override { - // Code here will be called immediately after each test (right before the - // destructor) - } + int CountTableRows(const std::string& table) { + std::string sql = "SELECT COUNT(*) FROM " + table; + sql::Statement s(GetDB().GetUniqueStatement(sql.c_str())); - // Objects declared here can be used by all tests in the test case - std::string GetMockDataPath(const std::string& filename) { - base::FilePath path(kTestDataRelativePath); - path = path.AppendASCII(filename); - return path.value(); - } + if (!s.Step()) { + return -1; + } - sql::Database& GetDB() { - return publisher_info_database_->GetDB(); + return static_cast(s.ColumnInt64(0)); } std::unique_ptr publisher_info_database_; }; TEST_F(PublisherInfoDatabaseTest, InsertContributionInfo) { + /** + * Good path + */ + base::ScopedTempDir temp_dir; + base::FilePath db_file; + CreateTempDatabase(&temp_dir, &db_file); + ContributionInfo info; info.probi = "12345678901234567890123456789012345678901234"; info.month = ledger::ACTIVITY_MONTH::JANUARY; @@ -95,6 +84,7 @@ TEST_F(PublisherInfoDatabaseTest, InsertContributionInfo) { info_sql.BindString(0, info.publisher_key); EXPECT_TRUE(info_sql.Step()); + EXPECT_EQ(CountTableRows("contribution_info"), 1); EXPECT_EQ(info_sql.ColumnString(0), info.publisher_key); EXPECT_EQ(info_sql.ColumnString(1), info.probi); EXPECT_EQ(info_sql.ColumnInt64(2), info.date); @@ -103,4 +93,316 @@ TEST_F(PublisherInfoDatabaseTest, InsertContributionInfo) { EXPECT_EQ(info_sql.ColumnInt(5), info.year); } +TEST_F(PublisherInfoDatabaseTest, InsertOrUpdatePublisherInfo) { + /** + * Good path + */ + base::ScopedTempDir temp_dir; + base::FilePath db_file; + CreateTempDatabase(&temp_dir, &db_file); + + ledger::PublisherInfo info; + info.id = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"; + info.verified = false; + info.excluded = ledger::PUBLISHER_EXCLUDE::DEFAULT; + info.name = "name"; + info.url = "https://brave.com"; + info.provider = ""; + info.favicon_url = ""; + + bool success = publisher_info_database_->InsertOrUpdatePublisherInfo(info); + EXPECT_TRUE(success); + + std::string query = "SELECT * FROM publisher_info WHERE publisher_id=?"; + sql::Statement info_sql(GetDB().GetUniqueStatement(query.c_str())); + + info_sql.BindString(0, info.id); + + EXPECT_TRUE(info_sql.Step()); + EXPECT_EQ(CountTableRows("publisher_info"), 1); + EXPECT_EQ(info_sql.ColumnString(0), info.id); + EXPECT_EQ(info_sql.ColumnBool(1), info.verified); + EXPECT_EQ(static_cast(info_sql.ColumnInt(2)), + info.excluded); + EXPECT_EQ(info_sql.ColumnString(3), info.name); + EXPECT_EQ(info_sql.ColumnString(4), info.favicon_url); + EXPECT_EQ(info_sql.ColumnString(5), info.url); + EXPECT_EQ(info_sql.ColumnString(6), info.provider); + + /** + * Make sure that second insert is update and not insert + */ + info.verified = true; + info.excluded = ledger::PUBLISHER_EXCLUDE::ALL; + info.name = "updated"; + info.url = "https://clifton.com"; + info.provider = ""; + info.favicon_url = "1"; + + success = publisher_info_database_->InsertOrUpdatePublisherInfo(info); + EXPECT_TRUE(success); + + query = "SELECT * FROM publisher_info WHERE publisher_id=?"; + sql::Statement info_sql_1(GetDB().GetUniqueStatement(query.c_str())); + + info_sql_1.BindString(0, info.id); + + EXPECT_TRUE(info_sql_1.Step()); + EXPECT_EQ(CountTableRows("publisher_info"), 1); + EXPECT_EQ(info_sql_1.ColumnString(0), info.id); + EXPECT_EQ(info_sql_1.ColumnBool(1), info.verified); + EXPECT_EQ(static_cast(info_sql_1.ColumnInt(2)), + info.excluded); + EXPECT_EQ(info_sql_1.ColumnString(3), info.name); + EXPECT_EQ(info_sql_1.ColumnString(4), info.favicon_url); + EXPECT_EQ(info_sql_1.ColumnString(5), info.url); + EXPECT_EQ(info_sql_1.ColumnString(6), info.provider); + + /** + * Publisher key is missing + */ + info.id = ""; + + success = publisher_info_database_->InsertOrUpdatePublisherInfo(info); + EXPECT_FALSE(success); + + query = "SELECT * FROM publisher_info WHERE publisher_id=?"; + sql::Statement info_sql_2(GetDB().GetUniqueStatement(query.c_str())); + + info_sql_2.BindString(0, info.id); + + EXPECT_FALSE(info_sql_2.Step()); +} + +TEST_F(PublisherInfoDatabaseTest, InsertOrUpdateActivityInfo) { + /** + * Good path + */ + base::ScopedTempDir temp_dir; + base::FilePath db_file; + CreateTempDatabase(&temp_dir, &db_file); + + ledger::PublisherInfo info; + info.id = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"; + info.verified = false; + info.excluded = ledger::PUBLISHER_EXCLUDE::DEFAULT; + info.name = "name"; + info.url = "https://brave.com"; + info.provider = ""; + info.favicon_url = ""; + info.duration = 10; + info.score = 1.1; + info.percent = 100; + info.weight = 1.5; + info.month = ledger::ACTIVITY_MONTH::JANUARY; + info.year = 1970; + info.reconcile_stamp = 0; + info.visits = 1; + + bool success = publisher_info_database_->InsertOrUpdateActivityInfo(info); + EXPECT_TRUE(success); + + std::string query = "SELECT * FROM activity_info WHERE publisher_id=?"; + sql::Statement info_sql(GetDB().GetUniqueStatement(query.c_str())); + + info_sql.BindString(0, info.id); + + EXPECT_TRUE(info_sql.Step()); + EXPECT_EQ(CountTableRows("activity_info"), 1); + EXPECT_EQ(info_sql.ColumnString(0), info.id); + EXPECT_EQ(static_cast(info_sql.ColumnInt64(1)), info.duration); + EXPECT_EQ(info_sql.ColumnInt64(2), info.visits); + EXPECT_EQ(info_sql.ColumnDouble(3), info.score); + EXPECT_EQ(info_sql.ColumnInt64(4), info.percent); + EXPECT_EQ(info_sql.ColumnDouble(5), info.weight); + EXPECT_EQ(info_sql.ColumnInt(6), info.month); + EXPECT_EQ(info_sql.ColumnInt(7), info.year); + EXPECT_EQ(static_cast(info_sql.ColumnInt64(8)), + info.reconcile_stamp); + + /** + * Make sure that second insert is update and not insert, + * month, year and stamp is unique key + */ + info.verified = true; + info.excluded = ledger::PUBLISHER_EXCLUDE::ALL; + info.name = "update"; + info.url = "https://slo-tech.com"; + info.provider = "1"; + info.favicon_url = "1"; + info.duration = 11; + info.score = 2.1; + info.percent = 200; + info.weight = 2.5; + info.visits = 2; + + success = publisher_info_database_->InsertOrUpdateActivityInfo(info); + EXPECT_TRUE(success); + + query = "SELECT * FROM activity_info WHERE publisher_id=?"; + sql::Statement info_sql_1(GetDB().GetUniqueStatement(query.c_str())); + + info_sql_1.BindString(0, info.id); + + EXPECT_TRUE(info_sql_1.Step()); + EXPECT_EQ(CountTableRows("activity_info"), 1); + EXPECT_EQ(info_sql_1.ColumnString(0), info.id); + EXPECT_EQ(static_cast(info_sql_1.ColumnInt64(1)), info.duration); + EXPECT_EQ(info_sql_1.ColumnInt64(2), info.visits); + EXPECT_EQ(info_sql_1.ColumnDouble(3), info.score); + EXPECT_EQ(info_sql_1.ColumnInt64(4), info.percent); + EXPECT_EQ(info_sql_1.ColumnDouble(5), info.weight); + EXPECT_EQ(info_sql_1.ColumnInt(6), info.month); + EXPECT_EQ(info_sql_1.ColumnInt(7), info.year); + EXPECT_EQ(static_cast(info_sql_1.ColumnInt64(8)), + info.reconcile_stamp); + +} + +TEST_F(PublisherInfoDatabaseTest, InsertOrUpdateMediaPublisherInfo) { + /** + * Good path + */ + base::ScopedTempDir temp_dir; + base::FilePath db_file; + CreateTempDatabase(&temp_dir, &db_file); + + std::string publisher_id = "id"; + std::string media_key = "key"; + + bool success = publisher_info_database_->InsertOrUpdateMediaPublisherInfo( + media_key, + publisher_id); + EXPECT_TRUE(success); + + std::string query = + "SELECT * FROM media_publisher_info WHERE media_key=?"; + sql::Statement info_sql(GetDB().GetUniqueStatement(query.c_str())); + + info_sql.BindString(0, media_key); + + EXPECT_TRUE(info_sql.Step()); + EXPECT_EQ(CountTableRows("media_publisher_info"), 1); + EXPECT_EQ(info_sql.ColumnString(0), media_key); + EXPECT_EQ(info_sql.ColumnString(1), publisher_id); + + /** + * Make sure that second insert is update and not insert + */ + publisher_id = "id_new"; + + success = publisher_info_database_->InsertOrUpdateMediaPublisherInfo( + media_key, + publisher_id); + EXPECT_TRUE(success); + + query = "SELECT * FROM media_publisher_info WHERE media_key=?"; + sql::Statement info_sql_1(GetDB().GetUniqueStatement(query.c_str())); + + info_sql_1.BindString(0, media_key); + + EXPECT_TRUE(info_sql_1.Step()); + EXPECT_EQ(CountTableRows("media_publisher_info"), 1); + EXPECT_EQ(info_sql_1.ColumnString(0), media_key); + EXPECT_EQ(info_sql_1.ColumnString(1), publisher_id); + + /** + * Publisher key is missing + */ + media_key = "missing"; + success = publisher_info_database_->InsertOrUpdateMediaPublisherInfo( + media_key, + ""); + EXPECT_FALSE(success); + + query = "SELECT * FROM media_publisher_info WHERE media_key=?"; + sql::Statement info_sql_2(GetDB().GetUniqueStatement(query.c_str())); + + info_sql_2.BindString(0, media_key); + + EXPECT_FALSE(info_sql_2.Step()); + + /** + * Media key is missing + */ + publisher_id = "new_stuff"; + success = publisher_info_database_->InsertOrUpdateMediaPublisherInfo( + "", + publisher_id); + EXPECT_FALSE(success); + + query = "SELECT * FROM media_publisher_info WHERE publisher_id=?"; + sql::Statement info_sql_3(GetDB().GetUniqueStatement(query.c_str())); + + info_sql_3.BindString(0, publisher_id); + + EXPECT_FALSE(info_sql_3.Step()); +} + +TEST_F(PublisherInfoDatabaseTest, InsertOrUpdateRecurringDonation) { + /** + * Good path + */ + base::ScopedTempDir temp_dir; + base::FilePath db_file; + CreateTempDatabase(&temp_dir, &db_file); + + brave_rewards::RecurringDonation info; + info.publisher_key = "key"; + info.amount = 20; + info.added_date = base::Time::Now().ToJsTime(); + + bool success = publisher_info_database_->InsertOrUpdateRecurringDonation( + info); + EXPECT_TRUE(success); + + std::string query = "SELECT * FROM recurring_donation WHERE publisher_id=?"; + sql::Statement info_sql(GetDB().GetUniqueStatement(query.c_str())); + + info_sql.BindString(0, info.publisher_key); + + EXPECT_TRUE(info_sql.Step()); + EXPECT_EQ(CountTableRows("recurring_donation"), 1); + EXPECT_EQ(info_sql.ColumnString(0), info.publisher_key); + EXPECT_EQ(info_sql.ColumnDouble(1), info.amount); + EXPECT_EQ(info_sql.ColumnInt64(2), info.added_date); + + /** + * Make sure that second insert is update and not insert + */ + info.amount = 30; + + success = publisher_info_database_->InsertOrUpdateRecurringDonation(info); + EXPECT_TRUE(success); + + query ="SELECT * FROM recurring_donation WHERE publisher_id=?"; + sql::Statement info_sql_1(GetDB().GetUniqueStatement(query.c_str())); + + info_sql_1.BindString(0, info.publisher_key); + + EXPECT_TRUE(info_sql_1.Step()); + EXPECT_EQ(CountTableRows("recurring_donation"), 1); + EXPECT_EQ(info_sql_1.ColumnString(0), info.publisher_key); + EXPECT_EQ(info_sql_1.ColumnDouble(1), info.amount); + EXPECT_EQ(info_sql_1.ColumnInt64(2), info.added_date); + + /** + * Publisher key is missing + */ + info.publisher_key = ""; + success = publisher_info_database_->InsertOrUpdateRecurringDonation(info); + EXPECT_FALSE(success); + + query = "SELECT * FROM recurring_donation WHERE publisher_id=?"; + sql::Statement info_sql_2(GetDB().GetUniqueStatement(query.c_str())); + + info_sql_2.BindString(0, info.publisher_key); + + EXPECT_FALSE(info_sql_2.Step()); +} + +TEST_F(PublisherInfoDatabaseTest, InsertPendingContribution) { + +} + } // namespace brave_rewards diff --git a/components/brave_rewards/browser/test_util.h b/components/brave_rewards/browser/test_util.h index 8f42676074de..ecbd65934593 100644 --- a/components/brave_rewards/browser/test_util.h +++ b/components/brave_rewards/browser/test_util.h @@ -14,215 +14,6 @@ class Profile; namespace brave_rewards { -class MockLedgerClient : public ledger::Ledger { - public: - MockLedgerClient(); - ~MockLedgerClient() override; - - MOCK_METHOD2(OnLoad, void(const ledger::VisitData& visit_data, - const uint64_t& current_time)); - - MOCK_METHOD0(Initialize, void()); - - MOCK_METHOD0(CreateWallet, bool()); - - MOCK_METHOD1(MakePayment, void(const ledger::PaymentData& payment_data)); - - MOCK_METHOD2(AddRecurringPayment, void(const std::string& publisher_id, - const double& value)); - - MOCK_METHOD3(DoDirectDonation, void(const ledger::PublisherInfo& publisher, - const int amount, - const std::string& currency)); - - MOCK_METHOD2(OnUnload, void(uint32_t tab_id, const uint64_t& current_time)); - - MOCK_METHOD2(OnShow, void(uint32_t tab_id, const uint64_t& current_time)); - - MOCK_METHOD2(OnHide, void(uint32_t tab_id, const uint64_t& current_time)); - - MOCK_METHOD2(OnForeground, void(uint32_t tab_id, - const uint64_t& current_time)); - - MOCK_METHOD2(OnBackground, void(uint32_t tab_id, - const uint64_t& current_time)); - - MOCK_METHOD2(OnMediaStart, void(uint32_t tab_id, - const uint64_t& current_time)); - - MOCK_METHOD2(OnMediaStop, void(uint32_t tab_id, - const uint64_t& current_time)); - MOCK_METHOD6(OnXHRLoad, void( - uint32_t tab_id, - const std::string& url, - const std::map& parts, - const std::string& first_party_url, - const std::string& referrer, - const ledger::VisitData& visit_data)); - - MOCK_METHOD5(OnPostData, void( - const std::string& url, - const std::string& first_party_url, - const std::string& referrer, - const std::string& post_data, - const ledger::VisitData& visit_data)); - - MOCK_METHOD1(OnTimer, void(uint32_t timer_id)); - - MOCK_METHOD1(URIEncode, std::string(const std::string& value)); - - MOCK_METHOD2(SetPublisherInfo, - void(std::unique_ptr publisher_info, - ledger::PublisherInfoCallback callback)); - - MOCK_METHOD2(SetActivityInfo, - void(std::unique_ptr publisher_info, - ledger::PublisherInfoCallback callback)); - - MOCK_METHOD2(GetPublisherInfo, void(const std::string& publisher_key, - ledger::PublisherInfoCallback callback)); - - MOCK_METHOD2(GetActivityInfo, void(const ledger::ActivityInfoFilter& filter, - ledger::PublisherInfoCallback callback)); - - MOCK_METHOD2(SetMediaPublisherInfo, void(const std::string& media_key, - const std::string& publisher_id)); - - MOCK_METHOD2(GetMediaPublisherInfo, void( - const std::string& media_key, - ledger::PublisherInfoCallback callback)); - - MOCK_METHOD4(GetActivityInfoList, void( - uint32_t start, - uint32_t limit, - const ledger::ActivityInfoFilter &filter, - ledger::PublisherInfoListCallback callback)); - - MOCK_METHOD0(GetRecurringDonationPublisherInfo, - std::vector()); - - MOCK_METHOD4(GetActivityInfoList, void( - uint32_t start, - uint32_t limit, - const ledger::ActivityInfoFilter& filter, - ledger::PublisherInfoListCallback callback)); - - MOCK_METHOD1(SetRewardsMainEnabled, void(bool enabled)); - - MOCK_METHOD1(SetPublisherMinVisitTime, void(uint64_t duration_in_seconds)); - - MOCK_METHOD1(SetPublisherMinVisits, void(unsigned int visits)); - - MOCK_METHOD1(SetPublisherAllowNonVerified, void(bool allow)); - - MOCK_METHOD1(SetPublisherAllowVideos, void(bool allow)); - - MOCK_METHOD1(SetContributionAmount, void(double amount)); - - MOCK_METHOD0(SetUserChangedContribution, void()); - - MOCK_METHOD1(SetAutoContribute, void(bool enabled)); - - MOCK_METHOD3(SetBalanceReport, void( - ledger::ACTIVITY_MONTH month, - int year, - const ledger::BalanceReportInfo& report_info)); - - MOCK_CONST_METHOD0(GetBATAddress, const std::string&()); - - MOCK_CONST_METHOD0(GetBTCAddress, const std::string&()); - - MOCK_CONST_METHOD0(GetETHAddress, const std::string&()); - - MOCK_CONST_METHOD0(GetLTCAddress, const std::string&()); - - MOCK_CONST_METHOD0(GetReconcileStamp, uint64_t()); - - MOCK_CONST_METHOD0(GetRewardsMainEnabled, bool()); - - MOCK_CONST_METHOD0(GetPublisherMinVisitTime, uint64_t()); - - MOCK_CONST_METHOD0(GetPublisherMinVisits, unsigned int()); - - MOCK_CONST_METHOD0(GetNumExcludedSites, unsigned int()); - - MOCK_CONST_METHOD0(GetPublisherAllowNonVerified, bool()); - - MOCK_CONST_METHOD0(GetPublisherAllowVideos, bool()); - - MOCK_CONST_METHOD0(GetContributionAmount, double()); - - MOCK_CONST_METHOD0(GetAutoContribute, bool()); - - MOCK_CONST_METHOD0(FetchWalletProperties, void()); - - MOCK_CONST_METHOD2(FetchGrant, void(const std::string& lang, - const std::string& paymentId)); - - MOCK_CONST_METHOD1(SolveGrantCaptcha, void(const std::string& solution)); - - MOCK_CONST_METHOD0(GetGrantCaptcha, void()); - - MOCK_CONST_METHOD0(GetWalletPassphrase, std::string()); - - MOCK_CONST_METHOD3(GetBalanceReport, bool( - ledger::ACTIVITY_MONTH month, - int year, - ledger::BalanceReportInfo* report_info)); - - MOCK_CONST_METHOD0(GetAllBalanceReports, - std::map()); - - MOCK_CONST_METHOD1(RecoverWallet, void(const std::string& passPhrase)); - - MOCK_METHOD4(SaveMediaVisit, void(const std::string& publisher_id, - const ledger::VisitData& visit_data, - const uint64_t& duration, - const uint64_t window_id)); - - MOCK_METHOD2(SetPublisherExclude, void( - const std::string& publisher_id, - const ledger::PUBLISHER_EXCLUDE& exclude)); - - MOCK_METHOD3(SetPublisherPanelExclude, void( - const std::string& publisher_id, - const ledger::PUBLISHER_EXCLUDE& exclude, - uint64_t windowId)); - - MOCK_METHOD0(RestorePublishers, void()); - - MOCK_CONST_METHOD0(IsWalletCreated, bool()); - - MOCK_METHOD2(GetPublisherActivityFromUrl, void( - uint64_t windowId, - const ledger::VisitData& visit_data)); - - MOCK_METHOD4(SetBalanceReportItem, void(ledger::ACTIVITY_MONTH month, - int year, - ledger::ReportType type, - const std::string& probi)); - - MOCK_METHOD2(GetPublisherBanner, void( - const std::string& publisher_id, - ledger::PublisherBannerCallback callback)); - - MOCK_METHOD0(GetBalance, double()); - - MOCK_METHOD6(OnReconcileCompleteSuccess, void( - const std::string& viewing_id, - const ledger::REWARDS_CATEGORY category, - const std::string& probi, - const ledger::ACTIVITY_MONTH month, - const int year, - const uint32_t date)); - - MOCK_METHOD1(RemoveRecurring, void(const std::string& publisher_key)); - - MOCK_METHOD0(GetDefaultContributionAmount, double()); - - MOCK_CONST_METHOD0(GetBootStamp, uint64_t()); -}; - std::unique_ptr CreateBraveRewardsProfile(const base::FilePath& path); } // namespace brave_rewards