28 lines
1003 B
Python
28 lines
1003 B
Python
from test_plus.test import TestCase
|
|
from mhackspace.users.models import Membership
|
|
from mhackspace.users.models import User
|
|
from django.contrib.auth.models import Group
|
|
|
|
from mhackspace.subscriptions.management.commands.update_membership_status import update_subscriptions
|
|
|
|
# this needs mocking
|
|
class TestTasks(TestCase):
|
|
def setUp(self):
|
|
self.user1 = self.make_user('u1')
|
|
self.user2 = self.make_user('u2')
|
|
self.group = Group(name='members')
|
|
self.group.save()
|
|
|
|
def test_refresh_subscriptions(self):
|
|
membership_count = Membership.objects.all().delete()
|
|
user_count = User.objects.all().count()
|
|
membership_count = Membership.objects.all().count()
|
|
self.assertEquals(0, membership_count)
|
|
self.assertEquals(2, user_count)
|
|
|
|
update_subscriptions(provider_name='gocardless')
|
|
|
|
membership_count = Membership.objects.all().count()
|
|
self.assertEquals(2, membership_count)
|
|
self.assertEquals(2, user_count)
|