5b405c4d51
but breaking functions and constants out into multiple modules. test_interface is now pretty much empty, as most of the tests there were useless. other modules that used to rely on constants from test_interface now look in constants, and the helper module now has a couple of broken out functions that are used through the tests.
78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
import mock
|
|
from artifice import interface, models
|
|
from .data_samples import RESOURCES, MAPPINGS
|
|
from .constants import config, TENANTS, DATACENTRE, AUTH_TOKEN
|
|
from datetime import timedelta
|
|
import json
|
|
|
|
|
|
@mock.patch("artifice.interface.keystone")
|
|
@mock.patch("sqlalchemy.create_engine")
|
|
def get_usage(sqlmock, keystone):
|
|
# At this point, we prime the ceilometer/requests response
|
|
# system, so that what we return to usage is what we expect
|
|
# to get in the usage system.
|
|
keystone.auth_token = AUTH_TOKEN
|
|
|
|
def get_meter(self, start, end, auth):
|
|
# Returns meter data from our data up above
|
|
data = MAPPINGS[self.link]
|
|
return data
|
|
|
|
interface.get_meter = get_meter
|
|
artifice = interface.Artifice(config)
|
|
artifice.auth.tenants.list.return_value = TENANTS
|
|
|
|
keystone.assert_called_with(
|
|
username=config["openstack"]["username"],
|
|
password=config["openstack"]["password"],
|
|
tenant_name=config["openstack"]["default_tenant"],
|
|
auth_url=config["openstack"]["authentication_url"]
|
|
)
|
|
tenants = None
|
|
tenants = artifice.tenants
|
|
|
|
t = tenants[0] # First tenant
|
|
t.resources = mock.create_autospec(interface.Resource, spec_set=True)
|
|
t.resources.return_value = (RESOURCES["networks"] + RESOURCES["vms"] +
|
|
RESOURCES["objects"] + RESOURCES["volumes"] +
|
|
RESOURCES["ips"])
|
|
|
|
# Replace the host_to_dc method with a mock that does what we need
|
|
# it to do, for the purposes of testing.
|
|
artifice.host_to_dc = mock.Mock()
|
|
artifice.host_to_dc.return_value = DATACENTRE
|
|
|
|
# because of mocking, start/end dates are not required here:
|
|
usage = t.usage(start=None, end=None)
|
|
|
|
# This is a fully qualified Usage object.
|
|
return usage
|
|
|
|
|
|
def fill_db(session, numb_tenants, numb_resources, now):
|
|
for i in range(numb_tenants):
|
|
session.add(models.Tenant(
|
|
id="tenant_id_" + str(i),
|
|
info="metadata",
|
|
name="tenant_name_" + str(i),
|
|
created=now
|
|
))
|
|
for ii in range(numb_resources):
|
|
session.add(models.Resource(
|
|
id="resource_id_" + str(ii),
|
|
info=json.dumps({"type": "Resource" + str(ii)}),
|
|
tenant_id="tenant_id_" + str(i),
|
|
created=now
|
|
))
|
|
session.add(models.UsageEntry(
|
|
service="service" + str(ii),
|
|
volume=5,
|
|
resource_id="resource_id_" + str(ii),
|
|
tenant_id="tenant_id_" + str(i),
|
|
start=(now - timedelta(days=30)),
|
|
end=now,
|
|
created=now
|
|
))
|
|
session.commit()
|