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