feat(toxpk): Add 'ToxPk::getPkSize()' method

qTox · Feb 7, 2017 · f5c0d61 · f5c0d61
1 parent d351f9e
commit f5c0d61
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/src/core/toxpk.cpp b/src/core/toxpk.cpp
@@ -83,8 +83,8 @@ QString ToxPk::toString() const
 
 /**
  * @brief Returns a pointer to the raw key data.
- * @return Pointer to the raw key data, which is exactly TOX_PUBLIC_KEY_SIZE bytes
- *         long. Returns a nullptr if the ToxPk is empty.
+ * @return Pointer to the raw key data, which is exactly `ToxPk::getPkSize()`
+ *         bytes long. Returns a nullptr if the ToxPk is empty.
  */
 const uint8_t* ToxPk::getBytes() const
 {
@@ -113,3 +113,12 @@ bool ToxPk::isEmpty() const
 {
     return key.isEmpty();
 }
+
+/**
+ * @brief Get size of public key in bytes.
+ * @return Size of public key in bytes.
+ */
+int ToxPk::getPkSize()
+{
+    return TOX_PUBLIC_KEY_SIZE;
+}
diff --git a/src/core/toxpk.h b/src/core/toxpk.h
@@ -20,6 +20,7 @@ class ToxPk
     const uint8_t* getBytes() const;
     bool isEmpty() const;
 
+    static int getPkSize();
 private:
     QByteArray key;
 };

diff --git a/test/core/toxpk_test.cpp b/test/core/toxpk_test.cpp
@@ -58,7 +58,7 @@ START_TEST(publicKeyTest)
 {
     ToxPk pk(testPk);
     ck_assert(testPk == pk.getKey());
-    for (int i = 0; i < 32; i++) {
+    for (int i = 0; i < ToxPk::getPkSize(); i++) {
         ck_assert(testPkArray[i] == pk.getBytes()[i]);
     }
 }