// Copyright (c) 2017-2022, Mudita Sp. z.o.o. All rights reserved.
// For licensing, see https://github.com/mudita/MuditaOS/LICENSE.md
#include "Helpers.hpp"
#include <Database/Database.hpp>
#include "module-db/databases/ContactsDB.hpp"
#include "module-db/databases/SmsDB.hpp"
#include <Interface/ContactRecord.hpp>
#include <Interface/SMSRecord.hpp>
#include <Interface/ThreadRecord.hpp>
#include <country.hpp>
#include <PhoneNumber.hpp>
#include <catch2/catch.hpp>
#include <algorithm>
#include <cstdint>
#include <cstdio>
#include <cstring>
#include <module-db/queries/messages/sms/QuerySMSGetForList.hpp>
struct test
{
UTF8 test;
UTF8 test2;
};
TEST_CASE("SMS Record tests")
{
db::tests::DatabaseUnderTest<ContactsDB> contactsDb{"contacts.db", db::tests::getPurePhoneScriptsPath()};
db::tests::DatabaseUnderTest<SmsDB> smsDb{"sms.db", db::tests::getPurePhoneScriptsPath()};
const uint32_t dateTest = 123456789;
const uint32_t errorCodeTest = 555;
auto numberTest = utils::PhoneNumber("+48600123456", utils::country::Id::UNKNOWN).getView();
auto numberTest2 = utils::PhoneNumber("222333444", utils::country::Id::UNKNOWN).getView();
const char *bodyTest = "Test SMS Body";
const char *bodyTest2 = "Test SMS Body2";
const SMSType typeTest = SMSType::DRAFT;
SMSRecordInterface smsRecInterface(&smsDb.get(), &contactsDb.get());
SMSRecord recordIN;
recordIN.date = dateTest;
recordIN.errorCode = errorCodeTest;
recordIN.number = numberTest;
recordIN.body = bodyTest;
recordIN.type = typeTest;
const auto smsCount = smsRecInterface.GetCount() + 1;
// clear sms table
for (std::uint32_t id = 1; id <= smsCount; id++) {
smsRecInterface.RemoveByID(id);
}
SECTION("SMS Record Test")
{
// Add 2 records
REQUIRE(smsRecInterface.Add(recordIN));
REQUIRE(smsRecInterface.Add(recordIN));
// Get all available records
auto records = smsRecInterface.GetLimitOffset(0, 100);
REQUIRE(records->size() == 2);
// Check if fetched records contain valid data
for (const auto &w : *records) {
REQUIRE(w.body == bodyTest);
REQUIRE(w.number == numberTest);
}
// Get all available records by specified thread ID and check for invalid data
records = smsRecInterface.GetLimitOffsetByField(0, 100, SMSRecordField::ThreadID, "1");
for (const auto &w : *records) {
REQUIRE(w.body == bodyTest);
REQUIRE(w.number == numberTest);
}
// Remove records one by one
REQUIRE(smsRecInterface.RemoveByID(1));
REQUIRE(smsRecInterface.RemoveByID(2));
// SMS database should not contain any records
REQUIRE(smsRecInterface.GetCount() == 0);
// Test updating record
REQUIRE(smsRecInterface.Add(recordIN));
recordIN.ID = 1;
recordIN.body = bodyTest2;
REQUIRE(smsRecInterface.Update(recordIN));
auto record = smsRecInterface.GetByID(1);
REQUIRE(record.ID != 0);
REQUIRE(record.body == bodyTest2);
// SMS database should contain 1 record
REQUIRE(smsRecInterface.GetCount() == 1);
// Remove existing record
REQUIRE(smsRecInterface.RemoveByID(1));
// SMS database should be empty
REQUIRE(smsRecInterface.GetCount() == 0);
// Test fetching record by using Thread ID
// Add records with different numbers, they should be placed in separate threads&contacts dbs
recordIN.body = bodyTest;
recordIN.number = numberTest;
REQUIRE(smsRecInterface.Add(recordIN));
REQUIRE(smsRecInterface.Add(recordIN));
recordIN.number = numberTest2;
REQUIRE(smsRecInterface.Add(recordIN));
REQUIRE(smsRecInterface.Add(recordIN));
// Get all available records by specified thread ID and check for invalid data
records = smsRecInterface.GetLimitOffsetByField(0, 100, SMSRecordField::ThreadID, "1");
for (const auto &w : *records) {
REQUIRE(w.body == bodyTest);
REQUIRE(w.number == numberTest);
}
// Get all available records by specified thread ID and check for invalid data
records = smsRecInterface.GetLimitOffsetByField(0, 100, SMSRecordField::ThreadID, "2");
for (const auto &w : *records) {
REQUIRE(w.body == bodyTest);
REQUIRE(w.number == numberTest2);
}
// Remove sms records in order to check automatic management of threads and contact databases
ThreadRecordInterface threadRecordInterface(&smsDb.get(), &contactsDb.get());
REQUIRE(smsRecInterface.RemoveByID(1));
REQUIRE(smsRecInterface.RemoveByID(2));
REQUIRE(smsRecInterface.RemoveByID(3));
REQUIRE(smsRecInterface.RemoveByID(4));
// Test removing a message which belongs to non-existent thread
REQUIRE(smsRecInterface.Add(recordIN));
REQUIRE(smsDb.get().threads.removeById(1)); // stealthy thread remove
REQUIRE(smsRecInterface.RemoveByID(1));
// Test handling of missmatch in sms vs. thread tables
auto trueCount = 10;
// prepare
for (auto added = 0; added < trueCount; added++) {
recordIN.date = added; // for proper order
recordIN.body = std::to_string(added + 1); // == ID
REQUIRE(smsRecInterface.Add(recordIN)); // threadID = 1
}
ThreadRecord threadRec = threadRecordInterface.GetByID(1);
REQUIRE(threadRec.isValid());
ThreadsTableRow threadRaw{Record(threadRec.ID),
.date = threadRec.date,
.msgCount = threadRec.msgCount,
.unreadMsgCount = threadRec.unreadMsgCount,
.contactID = DB_ID_NONE,
.snippet = threadRec.snippet,
.type = threadRec.type};
threadRaw.msgCount = trueCount + 1; // break the DB
REQUIRE(smsDb.get().threads.update(threadRaw));
REQUIRE(static_cast<int>(
smsRecInterface.GetLimitOffsetByField(0, 100, SMSRecordField::ThreadID, "1")->size()) == trueCount);
// end of preparation, now test
for (auto latest = trueCount; latest > 0; latest--) {
REQUIRE(smsRecInterface.RemoveByID(latest)); // remove the latest
switch (latest) { // was just removed
case 3: // remaining 2 or more
default:
REQUIRE(threadRecordInterface.GetByID(1).snippet.c_str() ==
std::to_string(latest - 1)); // next to newest
break;
case 2: // remaining 1
REQUIRE(threadRecordInterface.GetByID(1).snippet.c_str() == std::to_string(1)); // only one remaining
break;
case 1: // no sms remaining
// make sure there is no thread nor sms
REQUIRE(threadRecordInterface.GetCount() == 0);
REQUIRE(smsRecInterface.GetCount() == 0);
break;
}
}
// Test removing by field
recordIN.number = numberTest;
REQUIRE(smsRecInterface.Add(recordIN));
REQUIRE(smsRecInterface.Add(recordIN));
REQUIRE(smsRecInterface.Add(recordIN));
REQUIRE(smsRecInterface.Add(recordIN));
REQUIRE(smsRecInterface.RemoveByField(SMSRecordField::ThreadID, "1"));
recordIN.number = numberTest;
REQUIRE(smsRecInterface.Add(recordIN));
REQUIRE(smsRecInterface.Add(recordIN));
REQUIRE(smsRecInterface.Add(recordIN));
REQUIRE(smsRecInterface.Add(recordIN));
}
SECTION("SMS Record Draft and Input test")
{
recordIN.type = SMSType::INBOX;
// Add 3 INBOX Records
REQUIRE(smsRecInterface.Add(recordIN));
REQUIRE(smsRecInterface.Add(recordIN));
REQUIRE(smsRecInterface.Add(recordIN));
// Add 1 Draft Records
recordIN.type = SMSType::DRAFT;
recordIN.body = "Draft Message";
REQUIRE(smsRecInterface.Add(recordIN));
auto query = std::make_shared<db::query::SMSGetForList>(1, 0, 100);
auto ret = smsRecInterface.runQuery(query);
auto result = dynamic_cast<db::query::SMSGetForListResult *>(ret.get());
REQUIRE(result != nullptr);
}
}