def set_up(self):
google_user = users.get_current_user()
if google_user:
if users.is_current_user_admin():
find_user = FindOrCreateUser(google_user.email())
find_user.execute()
self._save_site = SaveSite(find_user.result, settings.APP_HOST)
self._save_site.set_up()
else:
self.add_error('user', _('You have no admin permission'))
else:
self.add_error('user', _('Login with your Google account'))
def set_up(self):
google_user = users.get_current_user()
if google_user:
if users.is_current_user_admin():
find_user = FindOrCreateUser(google_user.email())
find_user.execute()
self._save_site = SaveSite(find_user.result, settings.APP_HOST)
self._save_site.set_up()
else:
self.add_error('user', _('You have no admin permission'))
else:
self.add_error('user', _('Login with your Google account'))
def test_success(self,
url='http://www.mydomain.com',
domain='www.mydomain.com'):
find_user = FindOrCreateUser('*****@*****.**')
find_user.execute()
user = find_user.result
save_site = SaveSite(user, url)
save_site.execute()
site = save_site.result
self.assertEqual(domain, site.domain)
self.assertIsNotNone(site.token)
search = SitesSearch(user)
search.execute()
self.assertListEqual([site.key], [s.key for s in search.result])
def test_success(self):
users_mock = Mock()
google_user = Mock()
user_email = '*****@*****.**'
google_user.email = Mock(return_value=user_email)
users_mock.get_current_user = Mock(return_value=google_user)
users_mock.is_current_user_admin = Mock(return_value=True)
sites.users = users_mock
setup = facade.initial_setup()
setup.execute()
site = setup.result
self.assertEqual(settings.APP_HOST, site.domain)
find_user = FindOrCreateUser(user_email)
find_user.execute()
user = find_user.result
search = SitesSearch(to_node_key(user))
search.execute()
user_sites = search.result
self.assertListEqual([site.key], [s.key for s in user_sites])
