From 84695d9869daac4b1da8677c5e102068dd8edcbc Mon Sep 17 00:00:00 2001
From: Anton Beloglazov <anton.beloglazov@gmail.com>
Date: Mon, 6 Aug 2012 18:46:48 +1000
Subject: [PATCH] Started testing the collector

---
 neat/collector.py       | 27 ++++++++++++++++++++++-----
 tests/test_collector.py | 15 ++++++++++-----
 tests/test_config.py    | 27 +++++++++++++++++++++++++++
 3 files changed, 59 insertions(+), 10 deletions(-)
 create mode 100644 tests/test_config.py

diff --git a/neat/collector.py b/neat/collector.py
index 2142968..47ed1b3 100755
--- a/neat/collector.py
+++ b/neat/collector.py
@@ -16,6 +16,15 @@ import sys
 import libvirt
 
 
+def start(iterations):
+    collect()
+    return iterations
+
+
+def collect():
+    pass
+
+
 def getNumberOfPhysicalCpus(connection):
     return connection.getInfo()[2]
 
@@ -37,13 +46,21 @@ def getCpuUtilization(numberOfPhysicalCpus, domain, previousTime, previousCpuTim
 def collectCpuUtilization(numberOfPhysicalCpus, timeInterval, reportingFunction):
     pass
 
+# temporarily commented
+#conn = libvirt.openReadOnly(None)
+#if conn is None:
+#    print 'Failed to open connection to the hypervisor'
+#    sys.exit(1)
+#
+#numberOfPhysicalCpus = getNumberOfPhysicalCpus(conn)
+
+
+
+
+
+
 
-conn = libvirt.openReadOnly(None)
-if conn is None:
-    print 'Failed to open connection to the hypervisor'
-    sys.exit(1)
 
-numberOfPhysicalCpus = getNumberOfPhysicalCpus(conn)
 #print "Host CPUs: " + str(numberOfPhysicalCpus)
 
 
diff --git a/tests/test_collector.py b/tests/test_collector.py
index c27cdb3..3c184fe 100644
--- a/tests/test_collector.py
+++ b/tests/test_collector.py
@@ -13,16 +13,21 @@
 # limitations under the License.
 
 from pyqcy import *
+from mocktest import *
+
+import neat.collector
 
 
-class Arithmetic(TestCase):
+class DataCollector(TestCase):
 
     @qc
-    def addition_actually_works(
-        x=int_(min=0), y=int_(min=0)
+    def start(
+        iterations=int_(0, 10)
     ):
-        the_sum = x + y
-        assert the_sum >= x and the_sum >= y
+        expect(neat.collector).collect().and_return('ok').exactly(iteration).times()
+
+        print 'ok'
+        assert neat.collector.start(iterations) == iterations
 
     @qc
     def dumb():
diff --git a/tests/test_config.py b/tests/test_config.py
new file mode 100644
index 0000000..53dae1c
--- /dev/null
+++ b/tests/test_config.py
@@ -0,0 +1,27 @@
+# Copyright 2012 Anton Beloglazov
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from pyqcy import *
+
+from neat.config import readConfig
+
+
+class Config(TestCase):
+
+    @qc
+    def addition_actually_works(
+        x=int_(min=0), y=int_(min=0)
+    ):
+        the_sum = x + y
+        assert the_sum >= x and the_sum >= y