def test_can_activate_a_trial_subscription(self):
user = create_user(self.service, partner_id=self.partner['id'])
sub = self.service.create_subscription('wl_basic', user['id'],
{'trial': True})
self.assertTrue(sub['properties']['trial'])
activated_sub = self.service.activate_trial_subscription(sub['id'])
self.assertFalse(activated_sub['properties'].get('trial'))
def test_can_activate_a_trial_subscription(self):
user = create_user(self.service, partner_id=self.partner['id'])
sub = self.service.create_subscription(
'wl_basic', user['id'], {'trial': True})
self.assertTrue(sub['properties']['trial'])
activated_sub = self.service.activate_trial_subscription(sub['id'])
self.assertFalse(activated_sub['properties'].get('trial'))
def test_can_change_subscription_type(self):
user = create_user(self.service, partner_id=self.partner['id'])
sub = self.service.create_subscription('wl_basic', user['id'],
{'trial': True})
self.assertEqual(sub['type'], 'wl_basic')
modified_sub = self.service.change_subscription_type(
sub['id'], 'wl_lite')
self.assertEqual(modified_sub['type'], 'wl_lite')
def test_can_change_subscription_type(self):
user = create_user(self.service, partner_id=self.partner['id'])
sub = self.service.create_subscription(
'wl_basic', user['id'], {'trial': True})
self.assertEqual(sub['type'], 'wl_basic')
modified_sub = self.service.change_subscription_type(
sub['id'], 'wl_lite')
self.assertEqual(modified_sub['type'], 'wl_lite')
def setUpClass(cls):
super(TestYolaSubscription, cls).setUpClass()
cls.partner = create_partner(cls.service, properties={
'website': 'example.com',
'available_subscription_types': ['wl_basic', 'wl_lite']
})
user = create_user(cls.service, partner_id=cls.partner['id'])
cls.subscription = cls.service.create_subscription(
'wl_basic', user['id'], {'foo': 'bar'})
def test_set_site_url(self):
user = create_user(
self.service,
is_ws=True,
site_url='https://{}.yolasite.com'.format(uuid4().hex),
)
try:
new_site_url = 'https://{}.site.com'.format(uuid4().hex)
user = self.service.set_site_url(user['id'], new_site_url)
self.assertEqual(new_site_url, user['preferences']['site_url'])
finally:
self.service.delete_user(user['id'])
def setUpClass(cls):
super(TestYolaSubscription, cls).setUpClass()
cls.partner = create_partner(cls.service,
properties={
'website':
'example.com',
'available_subscription_types':
['wl_basic', 'wl_lite']
})
user = create_user(cls.service, partner_id=cls.partner['id'])
cls.subscription = cls.service.create_subscription(
'wl_basic', user['id'], {'foo': 'bar'})
def test_get_user_wsites(self):
expected_keys = ('id', 'owner_id', 'name', 'template_slug',
'created_at', 'deleted_at', 'updated_at')
ws_user = create_user(self.service,
is_ws=True,
site_url='https://{}.yolasite.com'.format(
uuid4().hex))
try:
user_wsites = self.service.get_user_wsites(ws_user['id'])
self.assertEqual(len(user_wsites), 1)
self.assertEqual(user_wsites[0]['owner_id'], ws_user['id'])
self.assertItemsEqual(user_wsites[0].keys(), expected_keys)
finally:
self.service.delete_user(ws_user['id'])
