def servers(session, clusters):
cluster1, *_ = clusters
server1 = Server(title="test1", cluster_id=cluster1.id)
server2 = Server(title="test2", cluster_id=cluster1.id)
session.add_all([server1, server2])
session.commit()
return Server.query.with_entities(Server.id, Server.title)
def metrics(session, services, services_groups, billing_batch_stories):
billing_batch_story1, billing_batch_story2 = billing_batch_stories
batch_stories = [billing_batch_story1, billing_batch_story2] * 40
metrics = []
for i in zip(services, services_groups):
metrics.append(
MetricBilling(
value=random.randint(1, 30),
batch_id=batch_stories.pop().id,
service_id=i[0].id,
services_group_id=i[1].id,
type="user_cpu_percent",
))
for i in zip(services, services_groups):
metrics.append(
MetricBilling(
value=random.randint(1, 30),
batch_id=batch_stories.pop().id,
service_id=i[0].id,
services_group_id=i[1].id,
type="vsize",
))
session.add_all(metrics)
session.commit()
return MetricBilling.query.with_entities(MetricBilling.id,
MetricBilling.type,
MetricBilling.value)
def services_groups(session):
services_group1 = ServicesGroup(title="test1")
services_group2 = ServicesGroup(title="test2")
session.add_all([services_group1, services_group2])
session.commit()
return ServicesGroup.query.with_entities(ServicesGroup.id,
ServicesGroup.title)
def servers(session, clusters):
servers = []
for i, cluster in enumerate(clusters, start=1):
servers.append(Server(title="server_" + str(i), cluster_id=cluster.id))
session.add_all(servers)
session.commit()
return Server.query.with_entities(Server.id, Server.title)
def teams(session):
"""save Teams instances to db"""
teams = [Team(title="team1"), Team(title="team2"), Team(title="team3")]
session.add_all(teams)
session.commit()
return Team.query.with_entities(Team.id, Team.title)
def create_project_settings_test_data(session):
session.add_all(
[
Project(id=1, title="test_project_1"),
Project(id=2, title="test_project_2"),
Project(id=3, title="test_project_3"),
]
)
session.commit()
def create_services_group_test_data(session):
session.add_all(
[
ServicesGroup(id=1, title="test_services_group_1"),
ServicesGroup(id=2, title="test_services_group_2"),
ServicesGroup(id=3, title="test_services_group_3"),
]
)
session.commit()
def billing_batch_stories(session):
time1 = datetime.now() - timedelta(days=8)
time2 = datetime.now()
batches_story1 = BatchStoryBilling(time=time1)
batches_story2 = BatchStoryBilling(time=time2)
session.add_all([batches_story1, batches_story2])
session.commit()
return BatchStoryBilling.query.with_entities(BatchStoryBilling.id,
BatchStoryBilling.time)
def create_team_settings_test_data(session):
session.add_all(
[
Team(id=1, title="test_team_1"),
Team(id=2, title="test_team_2"),
Team(id=3, title="test_team_3"),
]
)
session.commit()
def create_server_cluster_test_data(session):
session.add_all([
Server(id=1, title="test_server_1", cpu_price=2.4, memory_price=6),
Server(id=2, title="test_server_2"),
])
session.add_all([
Cluster(id=1, title="test_cluster_1", cpu_price=3, memory_price=2),
Cluster(id=2, title="test_cluster_2"),
])
session.commit()
def projects(session):
"""save Projects instances to db"""
projects = [
Project(title="project1"),
Project(title="project2"),
Project(title="project3"),
]
session.add_all(projects)
session.commit()
return Project.query.with_entities(Project.id, Project.title)
def create_server_settings_test_data(session, create_cluster_settings_test_data):
session.add_all(
[
Server(id=1, title="test_server_1", cluster_id=2),
Server(id=2, title="test_server_2", cluster_id=1),
Server(
id=3, title="test_server_3", cluster_id=3, cpu_price=2.2, memory_price=1.8
),
]
)
session.commit()
def create_cluster_settings_test_data(session):
session.add_all(
[
Cluster(id=1, title="test_cluster_1"),
Cluster(id=2, title="test_cluster_2"),
Cluster(
id=3, title="test_cluster_3", cpu_price=1.4, memory_price=2.7
),
]
)
session.commit()
def clusters(session):
"""save Cluster instances to db"""
clusters = [
Cluster(title="cluster1"),
Cluster(title="cluster2"),
Cluster(title="cluster3"),
Cluster(title="cluster4"),
]
session.add_all(clusters)
session.commit()
return Cluster.query.with_entities(Cluster.id, Cluster.title)
def services_groups(session, teams, projects):
services_groups = []
for i, t in enumerate(zip(projects, teams), start=1):
services_groups.append(
ServicesGroup(
title="services_group_" + str(i),
project_id=t[0].id,
team_id=t[1].id,
))
session.add_all(services_groups)
session.commit()
return ServicesGroup.query.with_entities(ServicesGroup.id,
ServicesGroup.title)
def services(session, servers, services_groups):
services = []
for i, t in enumerate(zip(servers, services_groups), start=1):
services.append(
Service(
title="service_" + str(i),
instance="test_instance",
server_id=t[0].id,
services_group_id=t[1].id,
))
session.add_all(services)
session.commit()
return Service.query.with_entities(Service.id, Service.title)
def create_project_services_group_test_data(session):
session.add_all(
[
Project(id=1, title="test_project_1"),
Project(id=2, title="test_project_2"),
]
)
session.commit()
session.add_all(
[
ServicesGroup(id=1, title="test_services_group_1", project_id=1),
ServicesGroup(id=2, title="test_services_group_2", project_id=2),
]
)
session.commit()
def servers(session, clusters: tuple):
"""save Servers instances to db"""
(cluster1_id, _), (cluster2_id, _), *_ = clusters
servers = [
Server(title="Server1", cluster_id=cluster1_id),
Server(title="Server2", cluster_id=cluster1_id),
Server(title="Server3", cluster_id=cluster2_id),
]
session.add_all(servers)
session.commit()
return Server.query.with_entities(Server.id, Server.title,
Server.cluster_id).all()
def services_groups_with_servers_and_clusters(session, servers, clusters):
"""
save ServicesGroups instances with
server_id and cluster_id foreign keys
ServicesGroups attached to servers through services
"""
(server1_id, *_), (server2_id, *_), (server3_id, *_), *_ = servers
session.add_all(
[ServicesGroup(title="group1"),
ServicesGroup(title="group2")])
session.commit()
session.add_all([
Service(
title="service1",
server_id=server1_id,
services_group_id=1,
instance=0,
),
Service(
title="service2",
server_id=server1_id,
services_group_id=1,
instance=0,
),
Service(
title="service3",
server_id=server2_id,
services_group_id=1,
instance=0,
),
Service(
title="service4",
server_id=server3_id,
services_group_id=1,
instance=0,
),
Service(
title="service5",
server_id=server3_id,
services_group_id=2,
instance=0,
),
])
session.commit()
def services(session, servers, services_groups):
services_group1, services_group2, *_ = services_groups
server1, server2, *_ = servers
service1 = Service(
title="service1",
instance="test",
server_id=server1.id,
services_group_id=services_group1.id,
)
service2 = Service(
title="service2",
instance="test",
server_id=server1.id,
services_group_id=services_group2.id,
)
service3 = Service(
title="service3",
instance="test",
server_id=server2.id,
services_group_id=services_group2.id,
)
session.add_all([service1, service2, service3])
session.commit()
return Service.query.with_entities(Service.id, Service.title)
def metrics(session, services, services_groups, batch_stories):
batches_story1, batches_story2, *_ = batch_stories
service1, service2, service3, *_ = services
services_group1, services_group2, *_ = services_groups
metric1 = MetricPeak(
value=2,
batch_id=batches_story1.id,
service_id=service1.id,
services_group_id=services_group1.id,
type="user_cpu_percent",
)
metric2 = MetricPeak(
value=2,
batch_id=batches_story2.id,
service_id=service2.id,
services_group_id=services_group2.id,
type="user_cpu_percent",
)
metric3 = MetricPeak(
value=3,
batch_id=batches_story2.id,
service_id=service3.id,
services_group_id=services_group2.id,
type="user_cpu_percent",
)
metric4 = MetricPeak(
value=4,
batch_id=batches_story1.id,
service_id=service3.id,
services_group_id=services_group2.id,
type="user_cpu_percent",
)
session.add_all([metric1, metric2, metric3, metric4])
session.commit()
return MetricPeak.query.with_entities(MetricPeak.id, MetricPeak.value)
def create_project_team_test_data(session):
session.add_all([
Project(id=1, title="test_project_1"),
Project(id=2, title="test_project_2"),
Project(id=3, title="test_project_3"),
])
session.commit()
session.add_all([
Team(id=1, title="test_team_1"),
Team(id=2, title="test_team_2"),
Team(id=3, title="test_team_3"),
])
session.commit()
session.add_all([
ServicesGroup(id=1, title="test_services_group_1"),
ServicesGroup(id=2, title="test_services_group_2"),
ServicesGroup(id=3, title="test_services_group_3"),
ServicesGroup(id=4, title="test_services_group_4", project_id=2),
])
session.commit()
def clusters(session):
clusters = [Cluster(title="test1")]
session.add_all(clusters)
session.commit()
return Cluster.query.with_entities(Cluster.id, Cluster.title)
def teams(session):
teams = [Team(title="team_" + str(i)) for i in range(1, 21)]
session.add_all(teams)
session.commit()
return Team.query.with_entities(Team.id, Team.title)
def clusters(session):
clusters = [Cluster(title="cluster_" + str(i)) for i in range(1, 21)]
session.add_all(clusters)
session.commit()
return Cluster.query.with_entities(Cluster.id, Cluster.title)
def projects(session):
projects = [Project(title="project_" + str(i)) for i in range(1, 21)]
session.add_all(projects)
session.commit()
return Project.query.with_entities(Project.id, Project.title)
