def setUp(self):
self.user = UserFactory()
self.account = AccountFactory(user=self.user)
self.income_category = IncomeCategoryFactory(user=self.user)
self.income = IncomeFactory(user=self.user,
account=self.account,
category=self.income_category)
def setUp(self):
self.user = UserFactory()
self.account = AccountFactory(user=self.user)
self.expense_category = ExpenseCategoryFactory(user=self.user)
self.expense = ExpenseFactory(user=self.user,
account=self.account,
category=self.expense_category)
def test_get_incomes_by_user(self):
COUNT = random.randint(1, 10)
user = UserFactory()
IncomeFactory.create_batch(COUNT, user=user)
incomes = services.get_incomes_by_user(user.id)
self.assertEqual(COUNT, incomes.count())
def test_reverse_link_from_expense_category_model(self):
"""Test reverse link from ExpenseCategory model."""
user = UserFactory()
account = AccountFactory(user=user)
expense_category = ExpenseCategoryFactory(user=user)
ExpenseFactory.create_batch(
7,
user=user,
account=account,
category=expense_category
)
self.assertEqual(expense_category.expenses.count(), 7)
def test_reverse_link_from_income_category_model(self):
"""Test reverse link from IncomeCategory model."""
user = UserFactory()
account = AccountFactory(user=user)
income_category = IncomeCategoryFactory(user=user)
IncomeFactory.create_batch(
6,
user=user,
account=account,
category=income_category
)
self.assertEqual(income_category.incomes.count(), 6)
from apps.tournament.management.commands.ping_snakes import Command
from apps.tournament.factories import TournamentFactory
from apps.authentication.factories import UserFactory
from apps.core.factories import SnakeFactory
from apps.core.models import Snake
from mock import patch
tournament_factory = TournamentFactory()
snake_factory = SnakeFactory()
user_factory = UserFactory()
command = Command()
@patch.object(Snake, "make_ping_request")
def test_all_healthy(mock_make_ping_request):
mock_make_ping_request.return_value = 200
user = user_factory.basic(email="*****@*****.**", commit=True)
snake_factory.basic(profile=user.profile, commit=True)
tournament_factory.basic(url="https://dedsnek.herokuapp.com")
command.test_all_snakes(only_tournament_snakes=False)
assert 41 == Snake.objects.filter(healthy=True).count()
@patch.object(Snake, "make_ping_request")
def test_healthy(mock_make_ping_request):
mock_make_ping_request.return_value = 200
tournament_factory.basic(url="https://dedsnek.herokuapp.com")
command.test_all_snakes(only_tournament_snakes=True)
assert 40 == Snake.objects.filter(healthy=True).count()
def setUp(self):
self.user = UserFactory()
