planetstack/tests/slivertest.py

"""
    Basic Sliver Test

    1) Create a slice1
    2) Create sliver1 on slice1
"""

import os
import json
import sys
import time

sys.path.append("/opt/planetstack")

os.environ.setdefault("DJANGO_SETTINGS_MODULE", "planetstack.settings")
from openstack.manager import OpenStackManager
from core.models import Slice, Sliver, ServiceClass, Reservation, Tag, Network, User, Node, Image, Deployment, Site, NetworkTemplate, NetworkSlice

from planetstacktest import PlanetStackTest, fail_unless

class SliverTest(PlanetStackTest):
    def __init__(self):
        PlanetStackTest.__init__(self)

    def run_sliver1(self):
        slice1Name = self.make_slice_name()
        slice1 = Slice(name = slice1Name,
                       omf_friendly=True,
                       site=self.testSite,
                       creator=self.testUser)
        slice1=self.save_and_wait_for_enacted(slice1, nonempty_fields=["tenant_id"])

        sliver1 = Sliver(image = self.testImage,
                         creator=self.testUser,
                         slice=slice1,
                         node=self.testNode,
                         deploymentNetwork=self.testDeployment)
        sliver1=self.save_and_wait_for_enacted(sliver1, nonempty_fields=["instance_id", "ip"])

    def run(self):
        self.setup()
        try:
             self.run_sliver1()
        finally:
             self.cleanup()

def main():
    SliverTest().run()

if __name__=="__main__":
    main()