def test_helper_no_profile(self):
u = User()
u.first_name = "John"
u.last_name = "Smith"
u.save()
self.assertEquals(False, helpers.user_has_profile(u))
p = Person()
p.user = u
p.save()
self.assertEquals(True, helpers.user_has_profile(u))
def test_helper_no_profile(self):
u = User()
u.first_name = "John"
u.last_name = "Smith"
u.save()
self.assertEquals(False, helpers.user_has_profile(u))
p = Person()
p.user = u
p.save()
self.assertEquals(True, helpers.user_has_profile(u))
def test_call_delete(self):
'''
Test deleting of a registration of a device from a channel
'''
u = User()
u.username = "******"
u.first_name = "sample_del"
u.email = "*****@*****.**"
u.set_password("123")
u.save()
ch1 = Channel()
ch1.owner = u
ch1.name = "DelSub"
ch1.image = 'http://www.google.com'
ch1.description = "A channel description"
ch1.kind = PUBLIC
ch1.hidden = False
ch1.subscriptions = 0
ch1.save()
sub1 = Subscriber()
sub1.sub_type = 'type2'
sub1.token = 'token2'
sub1.device_id = 'devid6'
sub1.save()
resp = ask_subscribe_channel(ch1, sub1.device_id)
self.assertEqual(resp, SubscribeResponse.SUBSCRIBED)
test_user = User.objects.create_superuser('test_user',
'*****@*****.**',
'password')
self.client.login(username='******', password='******')
response = self.client.delete(
reverse('browser-get-registration',
kwargs={
'device_id': sub1.device_id,
'channel_name': ch1.name
}))
self.assertEqual(response.status_code, 200)
channels = SubscriberManager().get_device_subscriptions(sub1.device_id)
sub_channel = next((x for x in channels if x == ch1.name.lower()),
None)
self.assertIsNone(sub_channel)
def test_call_register_channel(self):
'''
Test registration to a channel
'''
u = User()
u.username = "******"
u.first_name = "sample_post"
u.email = "*****@*****.**"
u.set_password("123")
u.save()
ch1 = Channel()
ch1.owner = u
ch1.name = "PostSub"
ch1.image = 'http://www.google.com'
ch1.description = "A channel description"
ch1.kind = PUBLIC
ch1.hidden = False
ch1.subscriptions = 0
ch1.save()
sub1 = Subscriber()
sub1.sub_type = 'type2'
sub1.token = 'token2'
sub1.device_id = 'devid5'
sub1.save()
resp = ask_subscribe_channel(ch1, sub1.device_id)
self.assertEqual(resp, SubscribeResponse.SUBSCRIBED)
test_user = User.objects.create_superuser('test_user',
'*****@*****.**',
'password')
self.client.login(username='******', password='******')
data = {
'channel': ch1.name,
'token': sub1.token,
'browser': 'chrome',
'device_id': sub1.device_id
}
response = self.client.post(reverse('browser-registration'),
json.dumps(data), sub1.token)
self.assertTrue(response.status_code, 200)
def test_call_register_channel(self):
'''
Test registration to a channel
'''
u = User()
u.username = "******"
u.first_name = "sample_post"
u.email = "*****@*****.**"
u.set_password("123")
u.save()
ch1 = Channel()
ch1.owner = u
ch1.name = "PostSub"
ch1.image = 'http://www.google.com'
ch1.description = "A channel description"
ch1.kind = PUBLIC
ch1.hidden = False
ch1.subscriptions = 0
ch1.save()
sub1 = Subscriber()
sub1.sub_type = 'type2'
sub1.token = 'token2'
sub1.device_id = 'devid5'
sub1.save()
resp = ask_subscribe_channel(ch1, sub1.device_id)
self.assertEqual(resp, SubscribeResponse.SUBSCRIBED)
test_user = User.objects.create_superuser('test_user', '*****@*****.**', 'password')
self.client.login(username='******', password='******')
data = {
'channel' : ch1.name,
'token' : sub1.token,
'browser' : 'chrome',
'device_id' : sub1.device_id
}
response = self.client.post(reverse('browser-registration'), json.dumps(data), sub1.token)
self.assertTrue(response.status_code, 200)
def test_call_delete(self):
'''
Test deleting of a registration of a device from a channel
'''
u = User()
u.username = "******"
u.first_name = "sample_del"
u.email = "*****@*****.**"
u.set_password("123")
u.save()
ch1 = Channel()
ch1.owner = u
ch1.name = "DelSub"
ch1.image = 'http://www.google.com'
ch1.description = "A channel description"
ch1.kind = PUBLIC
ch1.hidden = False
ch1.subscriptions = 0
ch1.save()
sub1 = Subscriber()
sub1.sub_type = 'type2'
sub1.token = 'token2'
sub1.device_id = 'devid6'
sub1.save()
resp = ask_subscribe_channel(ch1, sub1.device_id)
self.assertEqual(resp, SubscribeResponse.SUBSCRIBED)
test_user = User.objects.create_superuser('test_user', '*****@*****.**', 'password')
self.client.login(username='******', password='******')
response = self.client.delete(reverse('browser-get-registration', kwargs={'device_id': sub1.device_id,
'channel_name' : ch1.name}))
self.assertEqual(response.status_code, 200)
channels = SubscriberManager().get_device_subscriptions(sub1.device_id)
sub_channel = next((x for x in channels if x == ch1.name.lower()), None)
self.assertIsNone(sub_channel)
def test_call_get(self):
'''
Test check if a device is registered to a channel
'''
u = User()
u.username = "******"
u.first_name = "Sample_un"
u.email = "*****@*****.**"
u.set_password("123")
u.save()
ch1 = Channel()
ch1.owner = u
ch1.name = "GetSub"
ch1.image = 'http://www.google.com'
ch1.description = "A channel description"
ch1.kind = PUBLIC
ch1.hidden = False
ch1.subscriptions = 0
ch1.save()
sub1 = Subscriber()
sub1.sub_type = 'type2'
sub1.token = 'token2'
sub1.device_id = 'devid4'
sub1.save()
resp = ask_subscribe_channel(ch1, 'devid4')
self.assertEqual(resp, SubscribeResponse.SUBSCRIBED)
test_user = User.objects.create_superuser('test_user',
'*****@*****.**',
'password')
self.client.login(username='******', password='******')
response = self.client.get(
reverse('browser-get-registration',
kwargs={
'device_id': 'devid4',
'channel_name': 'GetSub'
}))
self.assertTrue(response.status_code, 200)
def test_call_get(self):
'''
Test check if a device is registered to a channel
'''
u = User()
u.username = "******"
u.first_name = "Sample_un"
u.email = "*****@*****.**"
u.set_password("123")
u.save()
ch1 = Channel()
ch1.owner = u
ch1.name = "GetSub"
ch1.image = 'http://www.google.com'
ch1.description = "A channel description"
ch1.kind = PUBLIC
ch1.hidden = False
ch1.subscriptions = 0
ch1.save()
sub1 = Subscriber()
sub1.sub_type = 'type2'
sub1.token = 'token2'
sub1.device_id = 'devid4'
sub1.save()
resp = ask_subscribe_channel(ch1, 'devid4')
self.assertEqual(resp, SubscribeResponse.SUBSCRIBED)
test_user = User.objects.create_superuser('test_user', '*****@*****.**', 'password')
self.client.login(username='******', password='******')
response = self.client.get(reverse('browser-get-registration', kwargs={'device_id': 'devid4', 'channel_name' : 'GetSub'}) )
self.assertTrue(response.status_code, 200)
def registration_approved_created_handler(sender, **kwargs):
registration = kwargs['registration']
user_doer = User()
user_doer.password = make_password(registration.password)
user_doer.username = registration.email
user_doer.email = registration.email
user_doer.first_name = registration.name
user_doer.last_Name = ''
user_doer.is_doer = True
user_doer.save()
doer = Doer()
doer.user = User.objects.filter(username=registration.email).first()
doer.email = registration.email
doer.nric_no = registration.nric_no
doer.name = registration.name
doer.birth_date = registration.birth_date
doer.state = registration.state
doer.city = registration.city
doer.gender = registration.gender
doer.race = registration.race
doer.level = Level.objects.filter(code='GANGSA').first()
doer.save()
def post(self, request):
from core.models import User, Employee
from employee_dashboard.models.Employee_Extra_Info import Employee_Extra_Info
from dateutil.parser import parse
from django.core.exceptions import ValidationError
from employee_dashboard.models.Section import Section
from django.contrib.sites.shortcuts import get_current_site
from django.template.loader import render_to_string
from django.utils.http import urlsafe_base64_encode
from django.utils.encoding import force_bytes
from core.tokens import account_activation_token
data = request.data
user = User()
user.email = data['email']
user.first_name = data['first_name']
user.last_name = data['last_name']
user.date_joined = data['hiring_date']
user.username = data['username']
user.position = data['position']
user.national_id = data['national_id']
user.phone_number = data['phone_number']
user.office_phone = data['office_number']
user.date_of_birth = data['date_of_birth']
user.is_employee = True
employee = Employee()
employee.employer = request.user.employer
employee.user = user
employee_extra = Employee_Extra_Info()
employee_extra.employee = employee
employee_extra.nationality = data['nationality']
employee_extra.job_title = data['job_title']
try:
employee_extra.direct_manager = Employee.objects.get(
user__email=data.get('direct_manager')) if data.get(
'direct_manager') else None
employee_extra.supervisor = Employee.objects.get(
user__email=data.get('supervisor')) if data.get(
'supervisor') else None
except Employee.DoesNotExist:
return Response({
'success':
False,
'message':
'Supervisor or Direct Manager Does Not Exist'
})
try:
employee_extra.section = Section.objects.get(
name=data.get('section')) if data.get('section') else None
except Section.DoesNotExist:
return Response({
'success':
False,
'message':
'The Section {} Does Not Exist'.format(data['section'])
})
employee_extra.birth_date = data['date_of_birth']
employee_extra.hiring_date = data['hiring_date']
employee_extra.end_of_contract = data['end_of_contract']
employee_extra.end_of_probation = data['end_of_probation']
employee_extra.marital_status = data['marital_status']
employee_extra.branch = data['branch']
employee_extra.sick_day_balance = 30
employee_extra.base_salary = data['base_salary']
employee_extra.GOSI_salary = data['GOSI_salary']
employee_extra.available_annual_vacation_balance = 30
employee_extra.gender = data['gender']
employee_extra.salutation = data['salutation']
employee_extra.status = 'Active'
try:
with transaction.atomic():
user.save()
employee.save()
employee_extra.save()
except ValidationError as e:
return Response({'success': False, 'message': e.message})
except IntegrityError:
return Response({
'success': False,
'message': 'Email or Username Already Exist'
})
current_site = get_current_site(request)
subject = 'Activate Employee Account'
message = render_to_string(
'core/account_activation_email.html', {
'user': employee,
'domain': current_site.domain,
'uid': urlsafe_base64_encode(force_bytes(employee.pk)),
'token': account_activation_token.make_token(user)
})
user.email_user(subject, message, from_email='*****@*****.**')
return Response({'success': True})
selangorCities = ['KLANG', 'HULU LANGAT', 'KUALA LANGAT', 'SEPANG', 'HULU SELANGOR', 'SABAK BERNAM', 'GOMBAK']
for i in range(len(selangorCities)):
city = City()
city.code = f'S{i}'
city.name = selangorCities[i]
city.state = State.objects.filter(code='SL').first()
city.save()
user_staff = User()
user_staff.password = '******'
user_staff.username = '******'
user_staff.email = '*****@*****.**'
user_staff.first_name = 'Staff'
user_staff.last_Name = '1'
user_staff.is_staff = True
user_staff.save()
staff = Staff()
staff.user = User.objects.filter(username='******').first()
staff.nric_no = '760607-12-4431'
staff.name = 'Rafizi Ramli'
staff.save()
user_staff = User()
user_staff.password = '******'
user_staff.username = '******'
user_staff.email = '*****@*****.**'
user_staff.first_name = 'Staff'
