def setUp(self):
self.client = APIClient()
self.owner_1 = 'owner_1'
self.owner_2 = 'owner_2'
self.accounts = (
Account(id='acc_1',
owner=self.owner_1,
balance=0,
currency=CURRENCY_PHP),
Account(id='acc_2',
owner=self.owner_1,
balance=100,
currency=CURRENCY_PHP),
Account(id='acc_3',
owner=self.owner_2,
balance=200,
currency=CURRENCY_PHP),
Account(id='acc_4',
owner=self.owner_2,
balance=200,
currency=CURRENCY_USD),
)
Account.objects.bulk_create(self.accounts)
self.empty_account, self.to_account, self.from_account, self.usd_account = self.accounts
def setUp(self):
self.client = APIClient()
self.owner_1 = 'owner_1'
self.owner_2 = 'owner_2'
self.accounts = (
Account(id='acc_1',
owner=self.owner_1,
balance=0,
currency=CURRENCY_PHP),
Account(id='acc_2',
owner=self.owner_2,
balance=200,
currency=CURRENCY_PHP),
)
Account.objects.bulk_create(self.accounts)
self.accounts[1].pay(self.accounts[0], Decimal(50), CURRENCY_PHP)
self.payments = list(Payment.objects.all())
self.creation_data_json = {
'from_account': self.accounts[1].pk,
'to_account': self.accounts[0].pk,
'amount': '1.5005',
'currency': CURRENCY_PHP,
}
def test_account_can_subscribe() -> None:
"""
Organization accounts that are missing exemptions can subscribe.
GitHub user accounts and accounts marked as exempt cannot subscribe.
"""
assert (Account(github_account_type=AccountType.user).can_subscribe() is
False), "user accounts cannot subscribe"
assert (
Account(github_account_type=AccountType.organization).can_subscribe()
is True), "organization accounts can subscribe"
assert (Account(github_account_type=AccountType.organization,
subscription_exempt=True).can_subscribe() is
False), "organization accounts with an exemption cannot subscribe"
def test_get_active_users_in_last_30_days() -> None:
account = Account(github_installation_id=52324234)
create_user_activity(
account=account,
user_id=333777,
pr_number=953,
date=timezone.now() - datetime.timedelta(days=5),
)
create_user_activity(
account=account, user_id=333777, pr_number=953, date=timezone.now()
)
create_user_activity(
account=account,
user_id=90322322,
pr_number=883,
date=timezone.now() - datetime.timedelta(days=10),
)
create_user_activity(
account=account, user_id=90322322, pr_number=883, date=timezone.now()
)
active_users = UserPullRequestActivity.get_active_users_in_last_30_days(account)
assert len(active_users) == 2
assert active_users[0].github_id == 90322322
assert active_users[1].github_id == 333777
def create(self, request, *args, **kwargs):
crypto_obj = get_data(request)
plain_text = crypto_obj.decrypt_text(request.data['data']).replace(
'\'', '\"')
data = json.loads(plain_text)
data = data if isinstance(data, dict) else {}
try:
serializer = self.get_serializer(data=data)
serializer.is_valid(raise_exception=True)
data = serializer.validated_data
password = data.pop('password')
user = Account(**data)
user.set_password(password)
user.save()
response = crypto_obj.encrypt_text("{}".format(
{"response": "user created."}))
return Response({'response': response}, status.HTTP_201_CREATED)
except IntegrityError as e:
response = crypto_obj.encrypt_text("{}".format(
{"response": "This username exist."}))
return Response({"response": response},
status.HTTP_400_BAD_REQUEST)
except Exception as e:
response = crypto_obj.encrypt_text("{}".format({'response': e}))
return Response({'response': response},
status=status.HTTP_400_BAD_REQUEST)
def save(self):
account = Account(wealth=self.wealth,
type=self.cleaned_data['type'],
name=self.cleaned_data['name'],
balance=self.cleaned_data['opening_balance'],
modified_date=datetime.utcnow(),
currency=self.cleaned_data['currency'],
opening_balance=self.cleaned_data['opening_balance'],
opened_date=self.cleaned_data['opened_date'])
account.save()
return account
def seed_db():
from core.models import Account, Server, ConnectionType
local_c = ConnectionType(name='local', connection_class='local')
local_c.save()
local = Server(ip_address='127.0.0.1',
name='localhost',
connection_type_id=local_c.id)
local.save()
act = Account(username='******',
password='******',
base_dir='/',
server_id=local.id)
act.save()
def form_valid(self, form):
# save client
form.save()
# generate data
num_card = utils.generateNumCard()
num_pin = utils.generateNumPin()
getClient = form.cleaned_data['document_number']
# add user
user = User(username=num_card, client=Client.objects.get(document_number=getClient))
user.set_password(num_pin)
user.save()
# add card
card = CreditCard(num_card=num_card, num_pin=num_pin,
expiration_date=utils.generateExpDate(), num_cvv=utils.generateNumCvv())
card.save()
# add account
account = Account(
num_account=utils.generateNumAccount(), client=Client.objects.get(document_number=getClient), card=CreditCard.objects.get(num_card=num_card))
account.save()
return super(SignUpView, self).form_valid(form)
from core.models import Account, IncomeCategory, ExpenseCategory
import json
with open('data.json', 'r') as file:
data = json.load(file)
for acc in data['account']:
account = Account(name=acc['name'])
account.save()
for in_c in data['incomeCategory']:
incomeCategory = IncomeCategory(name=in_c['name'])
incomeCategory.save()
for ex_c in data['expenseCategory']:
expenseCategory = ExpenseCategory(name=ex_c['name'])
expenseCategory.save()
