fcfa0f077d
Lots of general rename changes, and a few pieces of minor tidy up. Change-Id: Ia88b75af0e2d294cfc57164ac42155234d8ba71a
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
import mock
|
|
from distil import models
|
|
from datetime import timedelta
|
|
import json
|
|
|
|
|
|
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,
|
|
unit='gigabyte',
|
|
resource_id="resource_id_" + str(ii),
|
|
tenant_id="tenant_id_" + str(i),
|
|
start=(now - timedelta(days=20)),
|
|
end=now,
|
|
created=now
|
|
))
|
|
session.commit()
|
|
|
|
|
|
def create_usage_entries(num_resources, num_services, volume):
|
|
entries = []
|
|
for i in range(num_resources):
|
|
for ii in range(num_services):
|
|
entry = mock.MagicMock()
|
|
entry.volume = volume
|
|
entry.service = "service" + str(ii)
|
|
entry.resource_id = "resource_id_" + str(i)
|
|
entries.append(entry)
|
|
|
|
return entries
|