add create alarm

[doctor.git] / tests / main.py

diff --git a/tests/main.py b/tests/main.py
index 540a924..45683d1 100644 (file)
--- a/tests/main.py
+++ b/tests/main.py
@@ -10,6 +10,7 @@ import os
 from os.path import isfile, join
 import sys
 
+from alarm import Alarm
 import config
 from image import Image
 from instance import Instance
@@ -28,6 +29,7 @@ class DoctorTest(object):
         self.user = User(self.conf, LOG)
         self.network = Network(self.conf, LOG)
         self.instance = Instance(self.conf, LOG)
+        self.alarm = Alarm(self.conf, LOG)
 
     def setup(self):
         # prepare the cloud env
@@ -44,6 +46,9 @@ class DoctorTest(object):
         self.instance.create()
         self.instance.wait_for_vm_launch()
 
+        # creating alarm...
+        self.alarm.create()
+
     def run(self):
         """run doctor test"""
         try:
@@ -62,6 +67,7 @@ class DoctorTest(object):
             self.cleanup()
 
     def cleanup(self):
+        self.alarm.delete()
         self.instance.delete()
         self.network.delete()
         self.image.delete()