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 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'])