class RegistrationStoreTestCase(unittest.TestCase): @defer.inlineCallbacks def setUp(self): hs = yield setup_test_homeserver() self.store = RegistrationStore(hs) self.user_id = "@my-user:test" self.tokens = ["AbCdEfGhIjKlMnOpQrStUvWxYz", "BcDeFgHiJkLmNoPqRsTuVwXyZa"] self.pwhash = "{xx1}123456789" @defer.inlineCallbacks def test_register(self): yield self.store.register(self.user_id, self.tokens[0], self.pwhash) self.assertEquals( # TODO(paul): Surely this field should be 'user_id', not 'name' # Additionally surely it shouldn't come in a 1-element list {"name": self.user_id, "password_hash": self.pwhash}, (yield self.store.get_user_by_id(self.user_id)) ) result = yield self.store.get_user_by_token(self.tokens[0]) self.assertDictContainsSubset( { "admin": 0, "device_id": None, "name": self.user_id, }, result ) self.assertTrue("token_id" in result) @defer.inlineCallbacks def test_add_tokens(self): yield self.store.register(self.user_id, self.tokens[0], self.pwhash) yield self.store.add_access_token_to_user(self.user_id, self.tokens[1]) result = yield self.store.get_user_by_token(self.tokens[1]) self.assertDictContainsSubset( { "admin": 0, "device_id": None, "name": self.user_id, }, result ) self.assertTrue("token_id" in result)
class RegistrationStoreTestCase(unittest.TestCase): @defer.inlineCallbacks def setUp(self): hs = yield setup_test_homeserver() self.store = RegistrationStore(hs) self.user_id = "@my-user:test" self.tokens = ["AbCdEfGhIjKlMnOpQrStUvWxYz", "BcDeFgHiJkLmNoPqRsTuVwXyZa"] self.pwhash = "{xx1}123456789" @defer.inlineCallbacks def test_register(self): yield self.store.register(self.user_id, self.tokens[0], self.pwhash) self.assertEquals( # TODO(paul): Surely this field should be 'user_id', not 'name' # Additionally surely it shouldn't come in a 1-element list {"name": self.user_id, "password_hash": self.pwhash}, (yield self.store.get_user_by_id(self.user_id)) ) result = yield self.store.get_user_by_token(self.tokens[1]) self.assertDictContainsSubset( { "admin": 0, "device_id": None, "name": self.user_id, }, result ) self.assertTrue("token_id" in result) @defer.inlineCallbacks def test_add_tokens(self): yield self.store.register(self.user_id, self.tokens[0], self.pwhash) yield self.store.add_access_token_to_user(self.user_id, self.tokens[1]) result = yield self.store.get_user_by_token(self.tokens[1]) self.assertDictContainsSubset( { "admin": 0, "device_id": None, "name": self.user_id, }, result ) self.assertTrue("token_id" in result)
class RegistrationStoreTestCase(unittest.TestCase): @defer.inlineCallbacks def setUp(self): db_pool = SQLiteMemoryDbPool() yield db_pool.prepare() hs = HomeServer( "test", db_pool=db_pool, ) self.store = RegistrationStore(hs) self.user_id = "@my-user:test" self.tokens = [ "AbCdEfGhIjKlMnOpQrStUvWxYz", "BcDeFgHiJkLmNoPqRsTuVwXyZa" ] self.pwhash = "{xx1}123456789" @defer.inlineCallbacks def test_register(self): yield self.store.register(self.user_id, self.tokens[0], self.pwhash) self.assertEquals( # TODO(paul): Surely this field should be 'user_id', not 'name' # Additionally surely it shouldn't come in a 1-element list [{ "name": self.user_id, "password_hash": self.pwhash }], (yield self.store.get_user_by_id(self.user_id))) self.assertEquals({ "admin": 0, "device_id": None, "name": self.user_id }, (yield self.store.get_user_by_token(self.tokens[0]))) @defer.inlineCallbacks def test_add_tokens(self): yield self.store.register(self.user_id, self.tokens[0], self.pwhash) yield self.store.add_access_token_to_user(self.user_id, self.tokens[1]) self.assertEquals({ "admin": 0, "device_id": None, "name": self.user_id }, (yield self.store.get_user_by_token(self.tokens[1])))
class RegistrationStoreTestCase(unittest.TestCase): @defer.inlineCallbacks def setUp(self): db_pool = SQLiteMemoryDbPool() yield db_pool.prepare() hs = HomeServer("test", db_pool=db_pool, ) self.store = RegistrationStore(hs) self.user_id = "@my-user:test" self.tokens = ["AbCdEfGhIjKlMnOpQrStUvWxYz", "BcDeFgHiJkLmNoPqRsTuVwXyZa"] self.pwhash = "{xx1}123456789" @defer.inlineCallbacks def test_register(self): yield self.store.register(self.user_id, self.tokens[0], self.pwhash) self.assertEquals( # TODO(paul): Surely this field should be 'user_id', not 'name' # Additionally surely it shouldn't come in a 1-element list [{"name": self.user_id, "password_hash": self.pwhash}], (yield self.store.get_user_by_id(self.user_id)) ) self.assertEquals( self.user_id, (yield self.store.get_user_by_token(self.tokens[0])) ) @defer.inlineCallbacks def test_add_tokens(self): yield self.store.register(self.user_id, self.tokens[0], self.pwhash) yield self.store.add_access_token_to_user(self.user_id, self.tokens[1]) self.assertEquals( self.user_id, (yield self.store.get_user_by_token(self.tokens[1])) )