From: Matthias Wachs <wachs@net.in.tum.de>
Date: Fri, 22 Oct 2010 15:35:29 +0000 (+0000)
Subject: test now works
X-Git-Tag: initial-import-from-subversion-38251~19970
X-Git-Url: https://git.librecmc.org/?a=commitdiff_plain;h=16c5255cb5610343f67de5318e3c5c7df18eede7;p=oweals%2Fgnunet.git

test now works
---

diff --git a/src/transport/test_quota_compliance.c b/src/transport/test_quota_compliance.c
index c6da0e024..65c5e55f4 100644
--- a/src/transport/test_quota_compliance.c
+++ b/src/transport/test_quota_compliance.c
@@ -34,7 +34,7 @@
 #include "gnunet_transport_service.h"
 #include "transport.h"
 
-#define VERBOSE GNUNET_YES
+#define VERBOSE GNUNET_NO
 
 #define VERBOSE_ARM GNUNET_NO
 
@@ -118,7 +118,7 @@ end ()
 	    GNUNET_SCHEDULER_cancel (sched, measurement_counter_task);
 	    measurement_counter_task = GNUNET_SCHEDULER_NO_TASK;
   }
-
+  GNUNET_SCHEDULER_shutdown (sched);
 #if VERBOSE
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Disconnecting from transports!\n");
 #endif
@@ -128,8 +128,7 @@ end ()
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 	      "Transports disconnected, returning success!\n");
 #endif
-  ok = 0;
-
+  GNUNET_SCHEDULER_shutdown (sched);
 }
 
 
@@ -218,6 +217,9 @@ notify_ready_new (void *cls, size_t size, void *buf)
 
   transmit_handle = NULL;
 
+  if (measurement_task == GNUNET_SCHEDULER_NO_TASK)
+	  return 0;
+
   if (buf == NULL)
     {
       GNUNET_break (0);
@@ -270,6 +272,7 @@ static void measurement_counter
 
   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
 	return;
+
 #if VERBOSE
   fprintf(stderr,".");
 #endif
@@ -294,13 +297,29 @@ measurement_end (void *cls,
     GNUNET_SCHEDULER_cancel (sched, measurement_counter_task);
     measurement_counter_task = GNUNET_SCHEDULER_NO_TASK;
   }
-/*
+
   if (transmit_handle != NULL)
+  {
 	  GNUNET_TRANSPORT_notify_transmit_ready_cancel(transmit_handle);
-*/
-  fprintf (stderr, "\n");
-  GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
-		  "Measurement finished: \n Quota allowed: %llu kb/s\n Throughput: %llu kb/s\n", (current_quota_p1 / (1024)) , (total_bytes/(duration.value / 1000)/1024));
+	  transmit_handle = NULL;
+  }
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+		  "\n Measurement finished: \n Quota allowed: %llu kb/s\n Throughput: %llu kb/s\n", (current_quota_p1 / (1024)) , (total_bytes/(duration.value / 1000)/1024));
+
+  if (current_quota_p1 < total_bytes/(duration.value / 1000))
+  {
+	  GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+			  "\nQuota compliance failed: \n"\
+			  "Quota allowed: %llu kb/s\n"\
+			  "Throughput   : %llu kb/s\n", (current_quota_p1 / (1024)) , (total_bytes/(duration.value / 1000)/1024));
+	  ok = 1;
+	  end();
+	  return;
+  }
+  else
+  {
+	  ok = 0;
+  }
   if (current_quota_p1 < (MEASUREMENT_MIN_QUOTA))
 	  end();
   else
@@ -327,7 +346,11 @@ static void measure (unsigned long long quota_p1, unsigned long long quota_p2 )
 			  GNUNET_BANDWIDTH_value_init (current_quota_p2),
 			  GNUNET_TIME_UNIT_FOREVER_REL,
 			  NULL, NULL);
-		transmit_handle =GNUNET_TRANSPORT_notify_transmit_ready (p2.th,
+
+		if (transmit_handle != NULL)
+			  GNUNET_TRANSPORT_notify_transmit_ready_cancel(transmit_handle);
+
+		transmit_handle = GNUNET_TRANSPORT_notify_transmit_ready (p2.th,
 											  &p1.id,
 											  get_size_new (0), 0, TIMEOUT,
 											  &notify_ready_new,
@@ -509,7 +532,6 @@ main (int argc, char *argv[])
                       argv1, "test-quota-compliance", "nohelp",
                       options, &run, &ok);
   ret = ok;
-
   stop_arm (&p1);
   stop_arm (&p2);
   GNUNET_DISK_directory_remove ("/tmp/test_quota_compliance_peer1");