refix failing unittest

14 years ago · 16884a0dea
parent 552e72e35e
commit 16884a0dea
1 changed files with 8 additions and 6 deletions
--- a/src/flow.c
+++ b/src/flow.c
@ -599,12 +599,14 @@ static int FlowPrune(FlowQueue *q, struct timeval *ts, int try_cnt)
            continue;
        }

-#ifndef UNITTESTS
-        if (FlowForceReassemblyForFlowV2(f) == 1) {
-            Flow *prev_f = f;
-            f = f->lnext;
-            SCMutexUnlock(&prev_f->m);
-            continue;
+#ifdef UNITTESTS
+        if (run_mode != RUNMODE_UNITTEST) {
+            if (FlowForceReassemblyForFlowV2(f) == 1) {
+                Flow *prev_f = f;
+                f = f->lnext;
+                SCMutexUnlock(&prev_f->m);
+                continue;
+            }
        }
 #endif