aboutsummaryrefslogtreecommitdiffstats
path: root/src/dashboard/testing_utils.py
diff options
context:
space:
mode:
authorParker Berberian <pberberian@iol.unh.edu>2019-07-08 16:39:54 -0400
committerParker Berberian <pberberian@iol.unh.edu>2019-07-10 09:12:20 -0400
commit38b4412f8b6cf21371b67e8de917f5a24b2bf72d (patch)
treeb19adcaf27fce5d161b226017b62ce7e91a83be2 /src/dashboard/testing_utils.py
parent35c8640a32b7c6b3b35d68367668898fdfd7d423 (diff)
Fixes and adds Tests
Fixes old tests and adds some more. Also fixes linting errors Change-Id: If84f0ed2dab5f9ffa7ce2deb92e058b0076d9154 Signed-off-by: Parker Berberian <pberberian@iol.unh.edu>
Diffstat (limited to 'src/dashboard/testing_utils.py')
-rw-r--r--src/dashboard/testing_utils.py25
1 files changed, 16 insertions, 9 deletions
diff --git a/src/dashboard/testing_utils.py b/src/dashboard/testing_utils.py
index a96b6d0..0f52daa 100644
--- a/src/dashboard/testing_utils.py
+++ b/src/dashboard/testing_utils.py
@@ -103,12 +103,16 @@ def make_config_bundle(grb, owner, topology={}, host_set={},
cb = ConfigBundle.objects.create(
owner=owner,
name="config bundle " + str(ConfigBundle.objects.count()),
- description="cb generated by make_config_bundle() method"
+ description="cb generated by make_config_bundle() method",
+ bundle=grb
)
+ scen = scenario or Scenario.objects.first() or make_scenario()
+ inst = installer or Installer.objects.first() or make_installer([scen])
+
opnfv_config = OPNFVConfig.objects.create(
- installer=installer,
- scenario=scenario,
+ installer=inst,
+ scenario=scen,
bundle=cb
)
@@ -194,7 +198,8 @@ def make_generic_host(grb, host_profile, hostname):
def make_user(is_superuser=False, username="testuser",
password="testpassword", email="default_email@user.com"):
- user = User.objects.create_user(username=username, email=email, password=password)
+ user = User.objects.get_or_create(username=username, email=email, password=password)[0]
+
user.is_superuser = is_superuser
user.save()
@@ -204,14 +209,14 @@ def make_user(is_superuser=False, username="testuser",
def make_user_profile(user=None, email_addr="email@email.com",
company="company", full_name="John Doe",
booking_privledge=True, ssh_file=None):
- user = user or User.objects.first() or make_user()
- profile = UserProfile.objects.create(
+ user = user or make_user()
+ profile = UserProfile.objects.get_or_create(
email_addr=email_addr,
company=company,
full_name=full_name,
booking_privledge=booking_privledge,
user=user
- )
+ )[0]
profile.ssh_public_key.save("user_ssh_key", ssh_file if ssh_file else ContentFile("public key content string"))
return profile
@@ -238,7 +243,7 @@ def make_lab(user=None, name="Test_Lab_Instance",
vlan_manager = make_vlan_manager()
if not user:
- user = make_user()
+ user = make_user(username=name + " user")
lab = Lab.objects.create(
lab_user=user,
@@ -354,7 +359,9 @@ def make_installer(scenarios, name="test installer"):
return installer
-def make_os(installers, name="test OS"):
+def make_os(installers=None, name="test OS"):
+ if not installers:
+ installers = [make_installer([make_scenario()])]
os = Opsys.objects.create(name=name)
for installer in installers:
os.sup_installers.add(installer)