default_date_time_field=seq(TEST_TIME, timedelta(hours=3)),
default_time_field=seq(TEST_TIME.time(),
timedelta(seconds=15),
start="xpto"),
)
dog = Recipe(Dog, breed="Pug", owner=foreign_key(person))
homeless_dog = Recipe(
Dog,
breed="Pug",
)
other_dog = Recipe(Dog, breed="Basset", owner=foreign_key("person"))
dog_with_friends = dog.extend(friends_with=related(dog, dog), )
dog_with_more_friends = dog.extend(friends_with=related(dog_with_friends), )
extended_dog = dog.extend(breed="Super basset", )
class SmallDogRecipe(Recipe):
pass
small_dog = SmallDogRecipe(Dog)
pug = small_dog.extend(breed="Pug", )
other_dog_unicode = Recipe(Dog, breed="Basset", owner=foreign_key("person"))
with open(
os.path.join(settings.BASE_DIR,
"tacticalrmm/test_data/wmi_python_agent.json")) as f:
return json.load(f)
agent = Recipe(
Agent,
hostname="DESKTOP-TEST123",
version="1.3.0",
monitoring_type=cycle(["workstation", "server"]),
salt_id=generate_agent_id("DESKTOP-TEST123"),
agent_id="71AHC-AA813-HH1BC-AAHH5-00013|DESKTOP-TEST123",
)
server_agent = agent.extend(monitoring_type="server", )
workstation_agent = agent.extend(monitoring_type="workstation", )
online_agent = agent.extend(last_seen=djangotime.now())
overdue_agent = agent.extend(last_seen=djangotime.now() -
djangotime.timedelta(minutes=6))
agent_with_services = agent.extend(services=[
{
"pid": 880,
"name": "AeLookupSvc",
"status": "stopped",
"binpath": "C:\\Windows\\system32\\svchost.exe -k netsvcs",
"username": "******",
registration_early_deadline=now.add(days=30).datetime,
registration_deadline=now.add(days=60).datetime,
start_date=now.add(days=60).date,
end_date=now.add(days=65).date,
)
roadshow = Recipe("hipeac.Roadshow", start_date=now.add(days=30).date, end_date=now.add(days=35).date)
session_type = Recipe("hipeac.Metadata", type=Metadata.SESSION_TYPE)
session = Recipe(
"hipeac.Session", session_type=foreign_key(session_type), date=now.add(days=62).date, event=foreign_key(event)
)
# users
user = Recipe("auth.User", is_staff=False, profile__institution=foreign_key(institution))
staff = user.extend(is_staff=True)
member = user.extend(profile__membership_tags="member")
# jobs
employment_type = Recipe("hipeac.Metadata", type=Metadata.EMPLOYMENT)
job = Recipe(
"hipeac.Job",
employment_type=foreign_key(employment_type),
institution=foreign_key(institution),
email="*****@*****.**",
project=foreign_key(project),
deadline=now.add(months=1).datetime,
created_at=now.datetime,
created_by=foreign_key(user),
from model_bakery.recipe import Recipe
check = Recipe("checks.Check")
diskspace_check = check.extend(check_type="diskspace",
disk="C:",
warning_threshold=30,
error_threshold=10)
cpuload_check = check.extend(check_type="cpuload",
warning_threshold=30,
error_threshold=75)
ping_check = check.extend(check_type="ping", ip="10.10.10.10")
memory_check = check.extend(check_type="memory",
warning_threshold=60,
error_threshold=75)
winsvc_check = check.extend(
check_type="winsvc",
svc_name="ServiceName",
svc_display_name="ServiceName",
svc_policy_mode="manual",
pass_if_svc_not_exist=False,
)
eventlog_check = check.extend(check_type="eventlog",
event_id=5000,
event_type="application")
from .models import Check
from model_bakery.recipe import Recipe, seq
from model_bakery import baker
from itertools import cycle
check = Recipe(Check)
diskspace_check = check.extend(check_type="diskspace", disk="C:")
cpuload_check = check.extend(check_type="cpuload", threshold="75")
ping_check = check.extend(check_type="ping", ip="10.10.10.10")
memory_check = check.extend(check_type="memory", threshold="75")
winsvc_check = check.extend(check_type="winsvc",
svc_name=seq("ServiceName"),
svc_display_name=seq("ServiceName"),
svc_policy_mode="manual")
eventlog_check = check.extend(check_type="eventlog",
event_id=seq(0),
event_type="application")
script_check = check.extend(name="Script Name",
check_type="script",
script__name="Script Name")
mimetype = Recipe(MimeType, mime_type="image/png")
active_wms_service_metadata = Recipe(
Metadata,
title=seq("metadata_wms_"),
identifier=seq("metadata_wms"),
is_active=True,
metadata_type=MetadataEnum.SERVICE.value,
created_by=foreign_key(superadmin_group),
formats=related(mimetype),
)
active_wms_layer_metadata = active_wms_service_metadata.extend(
metadata_type=MetadataEnum.LAYER.value,
identifier=seq("metadata_wms_layer"),
formats=related(mimetype),
)
active_wfs_service_metadata = Recipe(
Metadata,
title=seq("metadata_wfs_"),
identifier=seq("metadata_wfs"),
is_active=True,
metadata_type=MetadataEnum.SERVICE.value,
created_by=foreign_key(superadmin_group),
formats=related(mimetype),
)
active_wfs_featuretype_metadata = active_wfs_service_metadata.extend(
metadata_type=MetadataEnum.FEATURETYPE.value,
guest_group = Recipe(
MrMapGroup,
name=seq("GuestGroup", increment_by=int(random.random() * 100)),
description=seq("Description"),
#created_by=foreign_key(god_user),
)
active_testuser = Recipe(MrMapUser,
username="******",
email="*****@*****.**",
salt=salt,
password=make_password(PASSWORD, salt=salt),
is_active=True,
groups=related(guest_group))
inactive_testuser = active_testuser.extend(is_active=False, )
publish_request = Recipe(
PublishRequest,
group=foreign_key(superadmin_group),
organization=foreign_key(non_autogenerated_orga),
)
group_invitation_request = Recipe(
GroupInvitationRequest,
to_group=foreign_key(superadmin_group),
invited_user=foreign_key(active_testuser),
message="Test",
)
pending_task = Recipe(TaskResult, status=states.STARTED, task_id=1)
from learning.models import Category, Tutorial, TutorialComment
def get_random_confirm_status():
return random.choice(
[
ConfirmStatusChoices.CONFIRMED,
ConfirmStatusChoices.WAITING_FOR_CONFIRM,
ConfirmStatusChoices.DISPROVED,
]
)
# Category recipes
category = Recipe(Category)
active_category = category.extend(is_active=True)
inactive_category = category.extend(is_active=False)
# Tutorial recipes
tutorial = Recipe(
Tutorial,
short_description=lambda: gen_string(150),
body=lambda: gen_string(150),
confirm_status=get_random_confirm_status,
_fill_optional=["author"],
)
confirmed_tutorial = tutorial.extend(
confirm_status=ConfirmStatusChoices.CONFIRMED
)
waiting_for_confirm_tutorial = tutorial.extend(
confirm_status=ConfirmStatusChoices.WAITING_FOR_CONFIRM
import pytz
from model_bakery.recipe import Recipe, seq
from .models import WinUpdate, WinUpdatePolicy
timezone = pytz.timezone("America/Los_Angeles")
severity = ["Critical", "Important", "Moderate", "Low", ""]
winupdate = Recipe(
WinUpdate,
kb=seq("kb0000000"),
guid=seq("12312331-123232-123-123-1123123"),
severity=cycle(severity),
)
approved_winupdate = winupdate.extend(action="approve")
winupdate_policy = Recipe(
WinUpdatePolicy,
run_time_hour=dt.now(timezone).hour,
run_time_frequency="daily",
run_time_days=[dt.now(timezone).weekday()],
)
winupdate_approve = winupdate_policy.extend(
critical="approve",
important="approve",
moderate="approve",
low="approve",
other="approve",
)
from .models import Script
from model_bakery.recipe import Recipe, seq
from model_bakery import baker
from itertools import cycle
script = Recipe(
Script,
name="Test Script",
description="Test Desc",
shell="cmd",
filename="test.bat",
script_type="userdefined",
)
builtin_script = script.extend(script_type="builtin")
from model_bakery.recipe import Recipe, related, foreign_key
from app.api.models import Artist, Album
from itertools import cycle
from datetime import date, timedelta
kari_job = Recipe(Artist, name="Kari Jobe")
comethazine = Recipe(Artist, name="Comethazine")
lee_brice = Recipe(Artist, name="Lee Brice")
fiji_blue = Recipe(Artist, name="Fiji Blue")
album_fixture = Recipe(Album)
blessing = album_fixture.extend(title="The Blessing (Live)",
album_type="album",
artists=related("kari_job"),
available_in_france=False,
release_date=date.today() - timedelta(days=3))
bawskee = album_fixture.extend(title="Bawskee 4",
album_type="single",
artists=related("comethazine"),
release_date=date.today() - timedelta(days=7))
memory = album_fixture.extend(title="The Blessing (Live)",
album_type="single",
artists=related("lee_brice"),
release_date=date.today() - timedelta(days=3))
affection = album_fixture.extend(title="Affection",
album_type="album",
artists=related("fiji_blue"),
