1 files changed, 104 insertions, 0 deletions

diff --git a/src/unit/queuebuf.unit b/src/unit/queuebuf.unit
new file mode 100644
index 0000000..9ae7d36
--- /dev/null
+++ b/src/unit/queuebuf.unit
@@ -0,0 +1,104 @@
+// vim: syntax=cpp
+/*
+ * Copyright (C) 2007-2010 Xagasoft, All rights reserved.
+ *
+ * This file is part of the libbu++ library and is released under the
+ * terms of the license contained in the file LICENSE.
+ */
+
+#include <stdlib.h>
+#include <time.h>
+
+#include "bu/queuebuf.h"
+#include "bu/md5.h"
+
+#define RNDCHR ((char)(((double)random()/(double)RAND_MAX)*256.0))
+
+{=Init}
+
+{%testBasic01}
+{
+        Bu::QueueBuf qb;
+        unitTest( qb.write("ab", 2 ) == 2 );
+        unitTest( qb.write("cde", 3 ) == 3 );
+        unitTest( qb.write("FG", 2 ) == 2 );
+
+        char buf[8];
+        buf[7] = '\0';
+        unitTest( qb.read( buf, 7 ) == 7 );
+        unitTest( !strncmp( buf, "abcdeFG", 7 ) );
+        unitTest( qb.read( buf, 7 ) == 0 );
+}
+
+void QBUF_RANDSTR( Bu::FString &fill, unsigned int iSize )
+{
+        char c;
+        for( unsigned int i=0; i<iSize; ++i )
+        {
+                c = RNDCHR;
+                fill.append(&c,1);
+        }
+}
+
+{%testAmounts}
+{
+        srandom(time(NULL));
+        Bu::QueueBuf qb;
+        Bu::FString sTmp;
+        char buf[4096];
+
+        for( int i=0; i<200; ++i )
+        {
+                unsigned int iAmt = (int)RNDCHR+128;
+                sTmp.clear();
+                QBUF_RANDSTR( sTmp, iAmt );
+                unitTest( qb.write( sTmp.getStr(), sTmp.getSize() ) == 
+                        (uint32_t)sTmp.getSize() );
+                size_t iRead = qb.read( buf, 4096 );
+                unitTest( iRead == iAmt );
+        }
+}
+
+void QBUF_HEXOUT( const char *s, int iSize )
+{
+        for( int i=0; i<iSize; ++i )
+                printf("%02x",(int)(uint8_t)s[i]);
+}
+
+void QBUF_HASH( Bu::FString &fill, const char *s, int iSize )
+{
+        Bu::Md5 hash;
+        hash.reset();
+        hash.addData( s, iSize );
+        const Bu::FString &sTmp = hash.getResult();
+        fill.append( sTmp.getStr(), 16 );
+}
+
+{%testRandomData}
+{
+        srandom(time(NULL));
+        Bu::QueueBuf qb;
+        Bu::FString sTmp;
+        Bu::FString sTmp2;
+        char buf[4096];
+
+        for( int i=0; i<200; ++i )
+        {
+                uint32_t iAmt = (uint32_t)RNDCHR+128;
+                sTmp.clear();
+                sTmp.append( (const char *)&iAmt, 4 );
+                QBUF_RANDSTR( sTmp, iAmt );
+                sTmp2.clear();
+                QBUF_HASH( sTmp2, sTmp.getStr()+4, iAmt );
+                sTmp.append( sTmp2 );
+                unitTest( qb.write( sTmp.getStr(), sTmp.getSize() ) == 
+                        (uint32_t)sTmp.getSize() );
+                size_t iRead = qb.read( buf, 4096 );
+                uint32_t iGotSize = *((uint32_t *)buf);
+                unitTest( iRead == iGotSize+4+16 );
+                sTmp2.clear();
+                QBUF_HASH( sTmp2, buf+4, iGotSize );
+                unitTest( !strncmp(sTmp2.getStr(),buf+4+iGotSize,16) );
+        }
+}
+