def test_init_no_id(self):
user = User(userid='user1', client='client1')
self.assertIsNotNone(user)
self.assertIsNone(user.id)
self.assertEqual('user1', user.userid)
self.assertEqual('client1', user.client)
self.assertEqual({
'client': 'client1',
'userid': 'user1'
}, user.to_document())
def test_from_document(self):
user1 = User.from_document({'client': 'client1', 'userid': 'user1'})
self.assertIsNotNone(user1)
self.assertIsNone(user1.id)
self.assertEqual('user1', user1.userid)
self.assertEqual('client1', user1.client)
user2 = User.from_document({
'_id': '666',
'client': 'client1',
'userid': 'user1'
})
self.assertIsNotNone(user2)
self.assertIsNotNone(user2.id)
self.assertEqual('666', user2.id)
self.assertEqual('user1', user2.userid)
self.assertEqual('client1', user2.client)
def test_repr_with_id(self):
user2 = User.from_document({
'_id': '666',
'client': 'client1',
'userid': 'user1'
})
self.assertEquals(
"<User(id='666', userid='user1', clientid='client1')>", str(user2))
def add_user(self, userid, client):
YLogger.info(self, "Adding user [%s] for client [%s]", userid, client)
user = User(userid, client)
self.add_document(user)
return True
def test_repr_no_id(self):
user1 = User.from_document({'client': 'client1', 'userid': 'user1'})
self.assertEquals(
"<User(id='n/a', userid='user1', clientid='client1')>", str(user1))
def test_from_document_no_id(self):
user1 = User.from_document({'client': 'client1', 'userid': 'user1'})
self.assertIsNotNone(user1)
self.assertIsNone(user1.id)
self.assertEqual('user1', user1.userid)
self.assertEqual('client1', user1.client)