def set_up(self):
parse_result = urlparse(self._url)
domain = parse_result.netloc.split(':')[0] or self._url
if domain:
token = urandom(16).encode('hex')
self.result = Site(domain=domain, token=token)
self._future = self.result._put_async()
else:
self.add_error('domain', _('Domain should not be empty'))
class SaveSite(Command):
def __init__(self, user, url):
super(SaveSite, self).__init__()
self._user = user
self._url = url
self._to_commit = None
def set_up(self):
parse_result = urlparse(self._url)
domain = parse_result.netloc.split(':')[0] or self._url
if domain:
token = urandom(16).encode('hex')
self.result = Site(domain=domain, token=token)
self._future = self.result._put_async()
else:
self.add_error('domain', _('Domain should not be empty'))
def do_business(self, stop_on_error=False):
if not self.errors:
site_key = self._future.get_result()
self._to_commit = SiteOwner(origin=to_node_key(self._user), destination=site_key)
def commit(self):
if not self.errors:
return self._to_commit
def set_up(self):
try:
self.result = memcache(settings.APP_HOST)
except:
pass
if not self.result:
self._future = Site.find_by_domain(settings.APP_HOST).get_async()
class SaveSite(Command):
def __init__(self, user, url):
super(SaveSite, self).__init__()
self._user = user
self._url = url
self._to_commit = None
def set_up(self):
parse_result = urlparse(self._url)
domain = parse_result.netloc.split(':')[0] or self._url
if domain:
token = urandom(16).encode('hex')
self.result = Site(domain=domain, token=token)
self._future = self.result._put_async()
else:
self.add_error('domain', _('Domain should not be empty'))
def do_business(self, stop_on_error=False):
if not self.errors:
site_key = self._future.get_result()
self._to_commit = SiteOwner(origin=to_node_key(self._user),
destination=site_key)
def commit(self):
if not self.errors:
return self._to_commit
def set_up(self):
try:
self.result = memcache(settings.APP_HOST)
except:
pass
if not self.result:
self._future = Site.find_by_domain(settings.APP_HOST).get_async()
def set_up(self):
parse_result = urlparse(self._url)
domain = parse_result.netloc.split(':')[0] or self._url
if domain:
token = urandom(16).encode('hex')
self.result = Site(domain=domain, token=token)
self._future = self.result._put_async()
else:
self.add_error('domain', _('Domain should not be empty'))
def test_success(self):
# Mocks
site = Site(domain='pswdless.appspot.com', token='t')
email = '*****@*****.**'
user = mommy.make_one(MainUser, email=email)
login = Login(status=LOGIN_CALL,
hook='https://pswdless.appspot.com/foo')
ndb.put_multi([site, user, login])
# Arcs
ndb.put_multi([
LoginUser(origin=login.key, destination=user.key),
LoginSite(origin=login.key, destination=site.key)
])
callback_flag = Mock()
def callback(*args):
callback_flag()
zipped = zip([login, site, user], args)
for expected, current in zipped:
self.assertEqual(type(expected), type(current))
self.assertEqual(expected.key, current.key)
send_email = facade.send_login_email(str(login.key.id()), callback)
send_email.execute()
# Login changes
login_db = login.key.get()
self.assertEqual(LOGIN_EMAIL, login_db.status)
search = SingleLoginStatusArcSearch(login)
login_status = search()
self.assertIsInstance(login_status, LoginStatus)
self.assertEqual(LOGIN_EMAIL, login_status.label)
# Site User creation
user_search = SingleSiteUserSearch(site)
db_user = user_search()
self.assertEqual(user.key, db_user.key)
# Callback call
callback_flag.assert_called_once_with()
def test_success(self):
mail_stub = self.testbed.get_stub(testbed.MAIL_SERVICE_NAME)
# Mocks
email = '*****@*****.**'
site = Site(domain='pswdless.appspot.com', token='t')
user = mommy.save_one(MainUser, email=email)
login = Login(status=LOGIN_CALL,
hook='https://pswdless.appspot.com/foo')
ndb.put_multi([site, user, login])
task._ = lambda s: s
# Arcs
ndb.put_multi([
LoginUser(origin=login.key, destination=user.key),
LoginSite(origin=login.key, destination=site.key)
])
task.send_login_email(template.render, str(login.key.id()), 'pt_BR')
messages = mail_stub.get_sent_messages(to='*****@*****.**')
self.assertEqual(1, len(messages))
