thunderbrewhq/common
1
0
Fork 0
mirror of https://github.com/thunderbrewhq/common.git synced 2025-12-12 03:02:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(datastore): add getter and setter for size

Browse source
This commit is contained in:
fallenoak 2023-02-25 10:18:54 -06:00
parent 08725b8455
commit f98814690c
No known key found for this signature in database
GPG key ID: 7628F8E61AEA070D
3 changed files with 38 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
common/datastore/CDataStore.cpp
View file

@ -346,6 +346,14 @@ CDataStore& CDataStore::Set(uint32_t pos, uint16_t val) {
return *this;
}
void CDataStore::SetSize(uint32_t size) {
this->m_size = size;
}
uint32_t CDataStore::Size() {
return this->m_size;
}
bool CDataStore::Sub8CBBF0(uint32_t a2) {
return this->m_read <= this->m_size && this->m_size - this->m_read >= a2;
}

2
common/datastore/CDataStore.hpp
View file

@ -46,6 +46,8 @@ class CDataStore {
CDataStore& PutData(const void* val, uint32_t bytes);
CDataStore& PutString(const char* val);
CDataStore& Set(uint32_t pos, uint16_t val);
void SetSize(uint32_t size);
uint32_t Size();
bool Sub8CBBF0(uint32_t a2);
};

28
test/DataStore.cpp
View file

@ -108,3 +108,31 @@ TEST_CASE("CDataStore::Get", "[datastore]") {
REQUIRE(msg.m_read == 0);
}
}
TEST_CASE("CDataStore::SetSize", "[datastore]") {
SECTION("sets size") {
uint8_t writeVal = 8;
CDataStore msg;
msg.Put(writeVal);
CHECK(msg.Size() == 1);
msg.SetSize(0);
CHECK(msg.Size() == 0);
}
}
TEST_CASE("CDataStore::Size", "[datastore]") {
SECTION("gets size") {
uint8_t writeVal1 = 8;
uint16_t writeVal2 = 9;
CDataStore msg;
msg.Put(writeVal1);
msg.Put(writeVal2);
REQUIRE(msg.Size() == 3);
}
}