Nil packer and unittests pack/unpack for nil

This commit is contained in:
Anton Dutov
2015-06-29 01:33:48 +05:00
parent c8b826c6a0
commit deac492817
4 changed files with 29 additions and 1 deletions

View File

@ -15,7 +15,10 @@ bool MsgPackPrivate::compatibilityMode = false;
quint8 *MsgPackPrivate::pack(const QVariant &v, quint8 *p, bool wr) quint8 *MsgPackPrivate::pack(const QVariant &v, quint8 *p, bool wr)
{ {
QMetaType::Type t = (QMetaType::Type)v.type(); QMetaType::Type t = (QMetaType::Type)v.type();
if (t == QMetaType::Int)
if (v.isNull())
p = pack_nil(p, wr);
else if (t == QMetaType::Int)
p = pack_int(v.toInt(), p, wr); p = pack_int(v.toInt(), p, wr);
else if (t == QMetaType::UInt) else if (t == QMetaType::UInt)
p = pack_uint(v.toUInt(), p, wr); p = pack_uint(v.toUInt(), p, wr);
@ -47,6 +50,13 @@ quint8 *MsgPackPrivate::pack(const QVariant &v, quint8 *p, bool wr)
return p; return p;
} }
quint8 *MsgPackPrivate::pack_nil(quint8 *p, bool wr)
{
if (wr)
*p = 0xc0;
return p + 1;
}
quint8 *MsgPackPrivate::pack_int(qint32 i, quint8 *p, bool wr) quint8 *MsgPackPrivate::pack_int(qint32 i, quint8 *p, bool wr)
{ {
if (i >= -32 && i <= 127) { if (i >= -32 && i <= 127) {

View File

@ -23,6 +23,7 @@ extern bool compatibilityMode;
quint8 * pack(const QVariant &v, quint8 *p, bool wr); quint8 * pack(const QVariant &v, quint8 *p, bool wr);
quint8 * pack_nil(quint8 *p, bool wr);
quint8 * pack_int(qint32 i, quint8 *p, bool wr); quint8 * pack_int(qint32 i, quint8 *p, bool wr);
quint8 * pack_uint(quint32 i, quint8 *p, bool wr); quint8 * pack_uint(quint32 i, quint8 *p, bool wr);
quint8 * pack_longlong(qint64 i, quint8 *p, bool wr); quint8 * pack_longlong(qint64 i, quint8 *p, bool wr);

View File

@ -9,6 +9,7 @@ class PackTest : public QObject
Q_OBJECT Q_OBJECT
private Q_SLOTS: private Q_SLOTS:
void test_nil();
void test_bool(); void test_bool();
void test_fixint(); void test_fixint();
void test_integer8(); void test_integer8();
@ -21,6 +22,14 @@ private Q_SLOTS:
void test_array(); void test_array();
}; };
void PackTest::test_nil()
{
QByteArray arr = MsgPack::pack(QVariant());
quint8 *p = (quint8 *)arr.data();
QVERIFY(arr.size() == 1);
QVERIFY(p[0] == 0xc0);
}
void PackTest::test_bool() void PackTest::test_bool()
{ {
QByteArray arr = MsgPack::pack(false); QByteArray arr = MsgPack::pack(false);

View File

@ -9,6 +9,7 @@ class UnpackTest : public QObject
Q_OBJECT Q_OBJECT
private Q_SLOTS: private Q_SLOTS:
void test_nil();
void test_booleans(); void test_booleans();
void test_integers(); void test_integers();
void test_floats(); void test_floats();
@ -17,6 +18,13 @@ private Q_SLOTS:
void test_array(); void test_array();
}; };
void UnpackTest::test_nil()
{
QByteArray pack = MsgPack::pack(QVariantList() << true << QVariant());
QVariantList u = MsgPack::unpack(pack).toList();
QVERIFY(u[0] == true);
QVERIFY(u[1].isNull());
}
void UnpackTest::test_booleans() void UnpackTest::test_booleans()
{ {