def setUp(self):
self.persistence_helper = self.add_helper(
PersistenceHelper(use_riak=True))
riak_manager = self.persistence_helper.get_riak_manager()
self.store = AccountStore(riak_manager)
# Some old stores for testing migrations.
self.store_v2 = AccountStoreV2(riak_manager)
self.store_v1 = AccountStoreV1(riak_manager)
self.store_vnone = AccountStoreVNone(riak_manager)
def setUp(self):
self.vumi_helper = yield self.add_helper(
VumiApiHelper(is_sync=self.is_sync))
self.vumi_api = self.vumi_helper.get_vumi_api()
self.user_helper = yield self.vumi_helper.make_user(u'Buster')
self.user_api = self.user_helper.user_api
# Some stores for old versions to test migrations.
self.account_store_vnone = AccountStoreVNone(self.vumi_api.manager)
self.account_store_v1 = AccountStoreV1(self.vumi_api.manager)
class TestUserAccount(VumiTestCase):
def setUp(self):
self.persistence_helper = self.add_helper(
PersistenceHelper(use_riak=True))
riak_manager = self.persistence_helper.get_riak_manager()
self.store = AccountStore(riak_manager)
# Some old stores for testing migrations.
self.store_v2 = AccountStoreV2(riak_manager)
self.store_v1 = AccountStoreV1(riak_manager)
self.store_vnone = AccountStoreVNone(riak_manager)
def assert_user(self, user, **fields):
def assert_field(value, name, default):
self.assertEqual(fields.get(name, default), value, name)
assert_field(user.username, 'username', u'testuser')
assert_field(user.tagpools.keys(), 'tagpools', [])
assert_field(user.applications.keys(), 'applications', [])
assert_field(user.created_at, 'created_at', UTCNearNow())
assert_field(user.event_handler_config, 'event_handler_config', [])
assert_field(user.msisdn, 'msisdn', None)
assert_field(user.confirm_start_conversation,
'confirm_start_conversation', False)
assert_field(user.tags, 'tags', [])
assert_field(user.routing_table, 'routing_table', RoutingTable({}))
def assert_user_v2(self, user, **fields):
def assert_field(value, name, default):
self.assertEqual(fields.get(name, default), value, name)
assert_field(user.username, 'username', u'testuser')
assert_field(user.tagpools.keys(), 'tagpools', [])
assert_field(user.applications.keys(), 'applications', [])
assert_field(user.created_at, 'created_at', UTCNearNow())
assert_field(user.event_handler_config, 'event_handler_config', [])
assert_field(user.msisdn, 'msisdn', None)
assert_field(user.confirm_start_conversation,
'confirm_start_conversation', False)
assert_field(user.tags, 'tags', [])
assert_field(user.routing_table, 'routing_table', {})
def assert_user_v1(self, user, **fields):
def assert_field(value, name, default):
self.assertEqual(fields.get(name, default), value, name)
assert_field(user.username, 'username', u'testuser')
assert_field(user.tagpools.keys(), 'tagpools', [])
assert_field(user.applications.keys(), 'applications', [])
assert_field(user.created_at, 'created_at', UTCNearNow())
assert_field(user.event_handler_config, 'event_handler_config', [])
assert_field(user.msisdn, 'msisdn', None)
assert_field(user.confirm_start_conversation,
'confirm_start_conversation', False)
assert_field(user.tags, 'tags', [])
def assert_user_vnone(self, user, **fields):
def assert_field(value, name, default):
self.assertEqual(fields.get(name, default), value, name)
assert_field(user.username, 'username', u'testuser')
assert_field(user.tagpools.keys(), 'tagpools', [])
assert_field(user.applications.keys(), 'applications', [])
assert_field(user.created_at, 'created_at', UTCNearNow())
assert_field(user.event_handler_config, 'event_handler_config', None)
assert_field(user.msisdn, 'msisdn', None)
assert_field(user.confirm_start_conversation,
'confirm_start_conversation', False)
@inlineCallbacks
def test_new_account(self):
user = yield self.store.new_user(u'testuser')
self.assert_user(user)
@inlineCallbacks
def test_migrate_new_from_v2(self):
user_v2 = yield self.store_v2.new_user(u'testuser')
self.assert_user_v2(user_v2)
user = yield self.store.get_user(user_v2.key)
self.assert_user(user)
@inlineCallbacks
def test_migrate_new_from_v1(self):
user_v1 = yield self.store_v1.new_user(u'testuser')
self.assert_user_v1(user_v1)
user = yield self.store.get_user(user_v1.key)
self.assert_user(user)
@inlineCallbacks
def test_migrate_new_from_vnone(self):
user_vnone = yield self.store_vnone.new_user(u'testuser')
self.assert_user_vnone(user_vnone)
user = yield self.store.get_user(user_vnone.key)
self.assert_user(user)
