def setUp(self):
self.notifier = FakeProvisionerNotifier()
self.store = ProvisionerStore()
self.ctx = FakeContextClient()
self.driver = FakeNodeDriver()
self.dtrs = FakeDTRS()
drivers = {'fake': self.driver}
self.core = ProvisionerCore(store=self.store,
notifier=self.notifier,
dtrs=self.dtrs,
site_drivers=drivers,
context=self.ctx)
def setUp(self):
self.notifier = FakeProvisionerNotifier()
self.context_client = FakeContextClient()
self.store = ProvisionerStore()
self.site_drivers = {'fake-site1': FakeNodeDriver()}
yield self._start_container()
yield self.spawn_procs()
self.fakecore = TerminateAllFakeCore()
self.patch(self.provisioner, "core", self.fakecore)
pId = yield self.procRegistry.get("provisioner")
self.client = ProvisionerClient(pid=pId)
def get_provisioner_store(conf):
cassandra_host = conf.getValue('cassandra_hostname')
if cassandra_host:
log.info("Using cassandra store. host: %s", cassandra_host)
try:
store = get_cassandra_store(cassandra_host,
conf['cassandra_username'],
conf['cassandra_password'],
conf.getValue('cassandra_keyspace'),
conf.getValue('cassandra_port'))
except KeyError, e:
raise KeyError("Provisioner config missing: " + str(e))
else:
log.info("Using in-memory Provisioner store")
store = ProvisionerStore()
return store
def get_context_client(conf):
try:
return ProvisionerContextClient(conf['context_uri'],
conf['context_key'],
conf['context_secret'])
except KeyError, e:
raise KeyError("Provisioner config missing: " + str(e))
def get_site_drivers(site_config):
"""Loads a dict of IaaS drivers from a config block
"""
def setup_store(self):
return defer.succeed(ProvisionerStore())
def setUp(self):
self.store = ProvisionerStore()
