def setUpClass(cls):
super(TestUser, cls).setUpClass()
cls.user = User(
id=cls.user_id,
username=cls.username,
password=cls.password,
first_name=cls.first_name,
last_name=cls.last_name,
)
session.add(cls.user)
session.flush()
def test_get_channel_by_id_for_value_error(self):
with self.assertRaises(ValueError) as err_wrong_param:
channel.get_channel_by_id(channel_id='wrong-id')
self.channel.deleted_on = '2016-04-04'
session.flush()
with self.assertRaises(ValueError) as err_deleted_param:
channel.get_channel_by_id(channel_id=self.channel_id)
with self.assertRaises(ValueError) as err_empty_string_param:
channel.get_channel_by_id(channel_id='')
self.assertEqual('INVALID-CHANNEL-ID', err_wrong_param.exception.message)
self.assertEqual('INVALID-CHANNEL-ID', err_deleted_param.exception.message)
self.assertEqual('REQ-CHANNEL-ID', err_empty_string_param.exception.message)
def test_get_user_by_id_user_for_value_error(self):
with self.assertRaises(ValueError) as err_wrong_param:
auth.get_user_by_id(user_id='wrong-id')
with self.assertRaises(ValueError) as err_empty_string_param:
auth.get_user_by_id(user_id='')
self.user.deleted_on = '2016-04-04'
session.flush()
with self.assertRaises(ValueError) as err_deleted_param:
auth.get_user_by_id(user_id=self.user_id)
self.assertEqual('BAD-USER-ID', err_wrong_param.exception.message)
self.assertEqual('REQ-USER-ID', err_empty_string_param.exception.message)
self.assertEqual('BAD-USER-ID', err_deleted_param.exception.message)
def test_publish_message_for_value_error(self):
with self.assertRaises(ValueError) as err_wrong_param:
chat.publish_message(channel_name=None, message=None, user_id=self.user_id)
with self.assertRaises(ValueError) as err_incomplete_param:
chat.publish_message(channel_name=None, message='some message', user_id=self.user_id)
self.chat.deleted_on = '2016-04-04'
session.flush()
with self.assertRaises(ValueError) as err_deleted_param:
chat.publish_message(channel_name=self.channel_name, message='some message', user_id=None)
with self.assertRaises(ValueError) as err_empty_string_param:
chat.publish_message(channel_name=None, message='some message', user_id=self.user_id)
self.assertEqual('PUB-REQ-MESSAGE-PAYLOAD', err_wrong_param.exception.message)
self.assertEqual('PUB-REQ-CHANNEL', err_incomplete_param.exception.message)
self.assertEqual('REQ-USER-ID', err_deleted_param.exception.message)
self.assertEqual('PUB-REQ-CHANNEL', err_empty_string_param.exception.message)
def signup(**kwargs):
log.info('Create user kwargs: {}'.format(kwargs))
if not get_user_by_username(username=kwargs['username']):
kwargs.update(dict(
id=str(uuid4()),
))
user = User(**kwargs)
session.add(user)
session.flush()
log.info("User added. id=%s", user.id)
else:
raise ValueError('SIGNUP-EXISTS-USERNAME')
return dict(
user_id=user.id,
username=user.username,
created=True
)
def setUpClass(cls):
super(TestChat, cls).setUpClass()
cls.user = User(
id=cls.user_id,
username=cls.username,
password=cls.password,
first_name=cls.first_name,
last_name=cls.last_name,
)
session.add(cls.user)
session.flush()
cls.channel = Channel(
id=cls.channel_id,
name=cls.channel_name,
type=cls.type,
created_by=cls.user.id,
)
session.add(cls.channel)
session.flush()
cls.chat = ChatHistory(
id=cls.message_id,
published_by=cls.user_id,
message=cls.message,
channel=cls.channel_id,
)
session.add(cls.chat)
session.flush()
def setUpClass(cls):
super(TestChannel, cls).setUpClass()
cls.user = User(
id=cls.user_id,
username=cls.username,
password=cls.password,
first_name=cls.first_name,
last_name=cls.last_name,
)
session.add(cls.user)
session.flush()
cls.channel = Channel(
id=cls.channel_id,
name=cls.channel_name,
type=cls.type,
created_by=cls.user.id,
)
session.add(cls.channel)
session.flush()
def test_authenticate_fail_because_user_deleted(self):
self.user.deleted_on = '2016-04-04'
session.flush()
self.assertEqual(auth.authenticate(self.username, self.password), False)
self.user.deleted_on = None
session.flush()
def test_get_user_by_username_user_deleted(self):
self.user.deleted_on = '2016-04-04'
session.flush()
self.assertIsNone(auth.get_user_by_username(username=self.username))
self.user.deleted_on = None
session.flush()
