def test_edit_alert(self):
"""Can we edit the alert and see the message about it being edited?"""
user = User.objects.get(username='******')
alert = Alert(
user=user,
name=self.alert_params['name'],
query=self.alert_params['query'],
rate=self.alert_params['rate'],
)
alert.save()
# Pan tries to edit their alert
self.browser.get(
'{url}{path}?{q}&edit_alert={pk}'.format(
url=self.server_url,
path=reverse('show_results'),
q=alert.query,
pk=alert.pk,
))
# But winds up at the sign in form
self.assertIn(reverse('sign-in'), self.browser.current_url)
# So Pan signs in.
self.browser.find_element_by_id('username').send_keys('pandora')
self.browser.find_element_by_id('password').send_keys('password' + '\n')
# And gets redirected to the SERP where they see a notice about their
# alert being edited.
self.assert_text_in_body("editing your alert")
def test_edit_alert(self):
"""Can we edit the alert and see the message about it being edited?"""
user = User.objects.get(username="******")
alert = Alert(
user=user, name=self.alert_params["name"], query=self.alert_params["query"], rate=self.alert_params["rate"]
)
alert.save()
# Pan tries to edit their alert
self.browser.get(
"{url}{path}?{q}&edit_alert={pk}".format(
url=self.server_url, path=reverse("show_results"), q=alert.query, pk=alert.pk
)
)
# But winds up at the sign in form
self.assertIn(reverse("sign-in"), self.browser.current_url)
# So Pan signs in.
self.browser.find_element_by_id("username").send_keys("pandora")
self.browser.find_element_by_id("password").send_keys("password" + "\n")
# And gets redirected to the SERP where they see a notice about their
# alert being edited.
self.assert_text_in_body("editing your alert")
def test_edit_alert(self):
"""Can we edit the alert and see the message about it being edited?"""
user = User.objects.get(username='******')
alert = Alert(
user=user,
name=self.alert_params['name'],
query=self.alert_params['query'],
rate=self.alert_params['rate'],
)
alert.save()
# Pan tries to edit their alert
self.browser.get(
'{url}{path}?{q}&edit_alert={pk}'.format(
url=self.server_url,
path=reverse('show_results'),
q=alert.query,
pk=alert.pk,
))
# But winds up at the sign in form
self.assertIn(reverse('sign-in'), self.browser.current_url)
# So Pan signs in.
self.browser.find_element_by_id('username').send_keys('pandora')
self.browser.find_element_by_id('password').send_keys('password' + '\n')
# And gets redirected to the SERP where they see a notice about their
# alert being edited.
self.assert_text_in_body("editing your alert")
def test_edit_alert(self) -> None:
"""Can we edit the alert and see the message about it being edited?"""
user = User.objects.get(username="******")
alert = Alert(
user=user,
name=self.alert_params["name"],
query=self.alert_params["query"],
rate=self.alert_params["rate"],
)
alert.save()
# Pan tries to edit their alert
url = "{url}{path}?{q}&edit_alert={pk}".format(
url=self.live_server_url,
path=reverse("show_results"),
q=alert.query,
pk=alert.pk,
)
self.browser.get(url)
# But winds up at the sign in form
self.assertIn(reverse("sign-in"), self.browser.current_url)
# So Pan signs in.
self.browser.find_element(By.ID, "username").send_keys("pandora")
self.browser.find_element(By.ID, "password").send_keys("password")
self.browser.find_element(By.ID, "password").submit()
# And gets redirected to the SERP where they see a notice about their
# alert being edited.
self.assert_text_in_node("editing your alert", "body")
def migrate_users_profiles_alerts_favorites_and_donations(self):
self.stdout.write("Migrating users, profiles, alerts, favorites, and "
"donations to the new database...")
old_users = User.objects.using('old').all()
num_users = old_users.count()
progress = 0
self._print_progress(progress, num_users)
for old_user in old_users:
old_profile = old_user.profile_legacy
old_alerts = old_profile.alert.all()
old_favorites = old_profile.favorite.all()
old_donations = old_profile.donation.all()
new_user = User(
pk=old_user.pk,
username=old_user.username,
first_name=old_user.first_name,
last_name=old_user.last_name,
email=old_user.email,
is_staff=old_user.is_staff,
is_active=old_user.is_active,
is_superuser=old_user.is_superuser,
date_joined=old_user.date_joined,
last_login=old_user.last_login,
password=old_user.password,
)
new_user.save(using='default')
new_profile = UserProfileNew(
pk=old_profile.pk,
user=new_user,
stub_account=old_profile.stub_account,
employer=old_profile.employer,
address1=old_profile.address1,
address2=old_profile.address2,
city=old_profile.city,
state=old_profile.state,
zip_code=old_profile.zip_code,
avatar=old_profile.avatar,
wants_newsletter=old_profile.wants_newsletter,
plaintext_preferred=old_profile.plaintext_preferred,
activation_key=old_profile.activation_key,
key_expires=old_profile.key_expires,
email_confirmed=old_profile.email_confirmed,
)
new_profile.save(using='default')
new_profile.barmembership.add(
*[membership.pk for membership in
old_profile.barmembership.all()]
)
for old_alert in old_alerts:
new_alert = AlertNew(
pk=old_alert.pk,
user=new_user,
date_created=self.the_beginning_of_time,
date_modified=self.the_beginning_of_time,
name=old_alert.name,
query=old_alert.query,
rate=old_alert.rate,
always_send_email=old_alert.always_send_email,
date_last_hit=old_alert.date_last_hit,
)
new_alert.save(using='default')
for old_favorite in old_favorites:
opinion_fave_pk = getattr(old_favorite.doc_id, 'pk', None)
audio_fave_pk = getattr(old_favorite.audio_id, 'pk', None)
if opinion_fave_pk is not None:
cluster = OpinionClusterNew.objects.get(
pk=opinion_fave_pk)
audio = None
else:
cluster = None
audio = AudioNew.objects.get(pk=audio_fave_pk)
new_favorite = FavoriteNew(
pk=old_favorite.pk,
user=new_user,
cluster_id=cluster,
audio_id=audio,
date_created=old_favorite.date_modified or now(),
date_modified=old_favorite.date_modified or now(),
name=old_favorite.name,
notes=old_favorite.notes,
)
new_favorite.save(using='default')
for old_donation in old_donations:
new_donation = DonationNew(
pk=old_donation.pk,
donor=new_user,
date_modified=old_donation.date_modified,
date_created=old_donation.date_created,
clearing_date=old_donation.clearing_date,
send_annual_reminder=old_donation.send_annual_reminder,
amount=old_donation.amount,
payment_provider=old_donation.payment_provider,
payment_id=old_donation.payment_id,
transaction_id=old_donation.transaction_id,
status=old_donation.status,
referrer=old_donation.referrer,
)
new_donation.save(using='default')
progress += 1
self._print_progress(progress, num_users)
self.stdout.write(u'') # Do a newline...
