def test_increment_for_strings_with_suffix_and_start(self):
from model_bakery.recipe import seq # NoQA
fred_person = person_recipe.extend(
email=seq("fred", start=5, suffix="@example.com"))
person = fred_person.make()
assert person.email == "*****@*****.**"
person = fred_person.make()
assert person.email == "*****@*****.**"
person = fred_person.make()
assert person.email == "*****@*****.**"
def test_increment_for_numbers_with_suffix(self):
from model_bakery.recipe import seq # NoQA
with pytest.raises(TypeError) as exc:
baker.make_recipe(
"tests.generic.serial_numbers",
default_int_field=seq(1, suffix="will not work"),
)
assert (
str(exc.value)
== "Sequences with suffix can only be used with text values"
)
def populate_data(self):
people = baker.make(Person, name=seq('Name'), _quantity=5)
for person in people:
baker.make(HealthRecord,
person=person,
_quantity=random.randint(1, 3))
baker.make(PersonProfile,
person=person,
age=random.randint(5, 555))
baker.make(MailingListLog,
email=person.email,
_quantity=random.randint(1, 3))
person = Recipe(
Person,
name="John Doe",
nickname="joe",
age=18,
bio="Someone in the crowd",
blog="http://joe.blogspot.com",
days_since_last_login=4,
birthday=now().date(),
appointment=now(),
birth_time=now(),
)
serial_person = Recipe(
Person,
name=seq("joe"),
)
serial_numbers = Recipe(
DummyDefaultFieldsModel,
default_decimal_field=seq(Decimal("20.1")),
default_int_field=seq(10),
default_float_field=seq(1.23),
)
serial_numbers_by = Recipe(
DummyDefaultFieldsModel,
default_decimal_field=seq(Decimal("20.1"), increment_by=Decimal("2.4")),
default_int_field=seq(10, increment_by=3),
default_float_field=seq(1.23, increment_by=1.8),
)
from datetime import timedelta
person = Recipe(
Person,
name="John Doe",
nickname="joe",
age=18,
bio="Someone in the crowd",
blog="http://joe.blogspot.com",
wanted_games_qtd=4,
birthday=now().date(),
appointment=now(),
birth_time=now(),
)
serial_person = Recipe(Person, name=seq("joe"),)
serial_numbers = Recipe(
DummyDefaultFieldsModel,
default_decimal_field=seq(Decimal("20.1")),
default_int_field=seq(10),
default_float_field=seq(1.23),
)
serial_numbers_by = Recipe(
DummyDefaultFieldsModel,
default_decimal_field=seq(Decimal("20.1"), increment_by=Decimal("2.4")),
default_int_field=seq(10, increment_by=3),
default_float_field=seq(1.23, increment_by=1.8),
)
from dateutil.relativedelta import relativedelta
from django.contrib.sites.models import Site
from edc_constants.constants import YES, NO
from edc_utils import get_utcnow
from faker import Faker
from model_bakery.recipe import Recipe, seq
from .models import SubjectReconsent
from .models import SubjectConsent
fake = Faker()
subjectconsent = Recipe(
SubjectConsent,
assessment_score=YES,
confirm_identity=seq("12315678"),
consent_copy=YES,
consent_datetime=get_utcnow(),
consent_reviewed=YES,
consent_signature=YES,
dob=get_utcnow() - relativedelta(years=25),
first_name=fake.first_name,
gender="M",
identity=seq("12315678"),
identity_type="country_id",
initials="XX",
is_dob_estimated="-",
is_incarcerated=NO,
is_literate=YES,
last_name=fake.last_name,
screening_identifier=None,
def get_wmi_data():
with open(
os.path.join(settings.BASE_DIR, "tacticalrmm/test_data/wmi_python_agent.json")
) as f:
return json.load(f)
agent = Recipe(
"agents.Agent",
site=foreign_key(site),
hostname="DESKTOP-TEST123",
version="1.3.0",
monitoring_type=cycle(["workstation", "server"]),
agent_id=seq("asdkj3h4234-1234hg3h4g34-234jjh34|DESKTOP-TEST123"),
)
server_agent = agent.extend(
monitoring_type="server",
)
workstation_agent = agent.extend(
monitoring_type="workstation",
)
online_agent = agent.extend(last_seen=djangotime.now())
offline_agent = agent.extend(
last_seen=djangotime.now() - djangotime.timedelta(minutes=7)
)
from django.contrib.auth.models import User
from model_bakery.recipe import Recipe, seq
from allauth.socialaccount.models import SocialApp
user = Recipe(User,
username=seq("test_user"),
password="******",
email="*****@*****.**",
)
fb_app = Recipe(SocialApp, provider='facebook')
from .models import Agent
from model_bakery.recipe import Recipe, seq
from itertools import cycle
from django.utils import timezone as djangotime
agent = Recipe(
Agent,
client="Default",
site="Default",
hostname=seq("TestHostname"),
monitoring_type=cycle(["workstation", "server"]),
)
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,
from faker import Faker
from model_bakery.recipe import Recipe, seq
from .models import Container
fake = Faker()
container = Recipe(
Container,
size_class=seq(fake.word(ext_word_list=None)),
standard=fake.pybool(),
volume=fake.pyfloat(left_digits=None,
right_digits=None,
positive=False,
min_value=1,
max_value=200),
deposit=fake.pydecimal(left_digits=None,
right_digits=None,
positive=False,
min_value=1,
max_value=100),
delivery_fee=fake.pydecimal(left_digits=None,
right_digits=None,
positive=False,
min_value=1,
max_value=50),
image_url=fake.image_url(width=None, height=None),
)
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")
from datetime import datetime as dt
from itertools import cycle
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",
from model_bakery.recipe import Recipe, seq
from .models import User
email_seq = seq("*****@*****.**")
user = Recipe(User,
first_name="Johnny",
last_name=seq("User"),
email=email_seq)
person = Recipe(
Person,
name="John Doe",
nickname="joe",
age=18,
bio="Someone in the crowd",
blog="http://joe.blogspot.com",
wanted_games_qtd=4,
birthday=now().date(),
appointment=now(),
birth_time=now(),
)
serial_person = Recipe(
Person,
name=seq("joe"),
)
serial_numbers = Recipe(
DummyDefaultFieldsModel,
default_decimal_field=seq(Decimal("20.1")),
default_int_field=seq(10),
default_float_field=seq(1.23),
)
serial_numbers_by = Recipe(
DummyDefaultFieldsModel,
default_decimal_field=seq(Decimal("20.1"), increment_by=Decimal("2.4")),
default_int_field=seq(10, increment_by=3),
default_float_field=seq(1.23, increment_by=1.8),
)
