class Test_Inventory:
#filename = "$HOME/.futuregrid/cloudmesh-new.yaml"
def setup(self):
self.inventory = Inventory("nosetest")
def tearDown(self):
pass
#self.inventory.disconnect()
def test00_disconnect(self):
HEADING("00 DISCONNECT")
print "NOT YET IMPLEMENTED"
def test01_clean(self):
HEADING("test01_clean")
self.inventory.clean()
def test02_add_Service(self):
HEADING("test02_add_Service")
now = datetime.now()
service = FabricService(
name='Euca',
date_start=now,
date_update=now,
date_stop=now
)
self.inventory.save(service)
def test03_add_Server(self):
HEADING("test03_add_Server")
now = datetime.now()
service = FabricService(
name='OpenStack',
date_start=now,
date_update=now,
date_stop=now
)
self.inventory.save(service)
server = FabricServer(
name='Hallo4',
date_start=now,
date_update=now,
date_stop=now,
services = [service]
)
self.inventory.save(server)
def test05_create(self):
HEADING("test05_create")
self.inventory.create("server","dynamic", "india[9-11].futuregrid.org,india[01-02].futuregrid.org")
print self.inventory.pprint()
assert self.inventory.exists("server", "india01.futuregrid.org")
def test06_loop_print(self):
HEADING("test06_loop_print")
for server in self.inventory.servers:
print server.data
def test07_exists(self):
HEADING("test07_exists")
assert self.inventory.exists("server", "india01.futuregrid.org") == True
def test08_print(self):
HEADING("test08_print")
self.inventory.pprint()
def test09_count(self):
HEADING("test09_count")
print self.inventory.servers.count(), self.inventory.services.count()
assert (self.inventory.servers.count() == 6) and (self.inventory.services.count() == 2)
def test10_set(self):
HEADING("test10_set")
self.inventory.clean()
self.inventory.create(
"server",
"dynamic",
"india01.futuregrid.org")
print self.inventory.pprint()
print self.inventory.exists("server", "india01.futuregrid.org")
self.inventory.set_service(
"india01-opensatck",
"india01.futuregrid.org",
"openstack")
self.inventory.pprint()
def test11_add(self):
HEADING("test11_add")
self.inventory.clean()
self.inventory.create(
"server",
"dynamic",
"india01.futuregrid.org")
print self.inventory.pprint()
print self.inventory.exists("server", "india01.futuregrid.org")
self.inventory.add_service(
"india01-opensatck",
"india01.futuregrid.org",
"openstack")
self.inventory.pprint()
def test12_logging(self):
self.test11_add()
HEADING("test12_logging")
s = self.inventory.get("server","india01.futuregrid.org")[0]
print s.data
s.stop()
s.start()
s.start()
######################################################################
from Inventory import Inventory
from Inventory import FabricService
from Inventory import FabricServer
inventory_db = server_config.get("mongo", "dbname")
if server_config.has_option("mongo", "host"):
inventory = Inventory(inventory_db,
server_config.get("mongo", "host"),
server_config.getint("mongo", "port"),
server_config.get("mongo", "user"),
server_config.get("mongo", "pass"))
else:
inventory = Inventory(inventory_db)
inventory.clean()
inventory.create_cluster("brave", "101.102.203.[11-26]", "b{0:03d}", 1, "b001", "b")
#inventory.create_cluster("delta", "102.202.204.[1-16]", "d-{0:03d}", 1, "d-001", "d")
#inventory.create_cluster("gamma", "302.202.204.[1-16]", "g-{0:03d}", 1, "g-001", "g")
#inventory.create_cluster("india", "402.202.204.[1-128]", "i-{0:03d}", 1, "i-001", "i")
#inventory.create_cluster("sierra", "502.202.204.[1-128]", "s-{0:03d}", 1, "s-001", "s")
centos = FabricImage(
name = "centos6",
osimage = '/path/to/centos0602v1-2013-06-11.squashfs',
os = 'centos6',
extension = 'squashfs',
partition_scheme = 'mbr',
method = 'put',
kernel = 'vmlinuz-2.6.32-279.19.1.el6.x86_64',
