OSF fixture generator

account_merge.py

from tests import factories
from website import models
from website.util import web_url_for
from main import app

def print_merge_url(user, token, prefix):
    with app.test_request_context():
        print(prefix + ": " +
            web_url_for(
                'confirm_email_get',
                uid=user._id,
                token=token,
                _absolute=True
            )
        )

if __name__ == "__main__":
    models.Node.remove()
    models.User.remove()

    one = factories.UserFactory(username='test@one.com')
    one.set_password('asdfasdf')
    one.save()

    two = factories.UserFactory(username='test@two.com')
    two.set_password('asdfasdf')
    two.save()

    token = one.add_unconfirmed_email('test@two.com')
    one.save()
    print_merge_url(one, token, "No addons")

    three = factories.UserFactory(username='test@three.com')
    three.set_password('asdfasdf')
    three.add_addon('dataverse')
    three.save()


    token = one.add_unconfirmed_email('test@three.com')
    one.save()
    print_merge_url(one, token, "Old-style addon")

    four = factories.UserFactory(username='test@four.com')
    four.set_password('adsfasdf')
    four.add_addon('mendeley')
    four.save()

    token = one.add_unconfirmed_email('test@four.com')
    one.save()
    print_merge_url(one, token, "New-style addon")