def test_add_topic_multiple_saved_topics(self):
with test_database(test_db, (Log, Topic), create_tables=True):
fixture = Topic.create(name='test/topic/test/xx')
fixture = Topic.create(name='test/topic/test/xxx')
topic = TopicsController()
result = topic.add_topic(self.msg)
parsedResponse = json.loads(result)
self.assertEqual(3, len(parsedResponse['topics']))
def test_add_topic_bad_pass(self):
with test_database(test_db, (Log, Topic), create_tables=True):
self.payload['password'] = "******"
self.msg.payload = json.dumps(self.payload)
topic = TopicsController()
result = topic.add_topic(self.msg)
parsedResponse = json.loads(result)
self.assertEqual('KO', parsedResponse['result'])
self.assertEqual('Bad Password', parsedResponse['error'])
def test_add_topic_missing_password(self):
with test_database(test_db, (Log, Topic), create_tables=True):
self.payload.pop('password', None)
self.msg.payload = json.dumps(self.payload)
topic = TopicsController()
result = topic.add_topic(self.msg)
parsedResponse = json.loads(result)
self.assertEqual('KO', parsedResponse['result'])
self.assertEqual('Missing key - password', parsedResponse['error'])
def test_remove_topic_delete_existing_topic(self):
with test_database(test_db, (Log, Topic), create_tables=True):
fixture = Topic.create(name='/test/topic/1')
fixture = Topic.create(name='/test/topic/2')
fixture = Topic.create(name='/test/topic/3')
fixture = Topic.create(name='/test/topic/4')
topic = TopicsController()
result = topic.remove_topic(self.msg)
parsedResponse = json.loads(result)
self.assertEqual(3, len(parsedResponse['topics']))
def test_list_topics(self):
with test_database(test_db, (Log, Topic), create_tables=True):
self.payload['password'] = Settings.QUERY_PASSWORD
self.msg.payload = json.dumps(self.payload)
fixture = Topic.create(name='/test/topic/2')
fixture = Topic.create(name='/test/topic/3')
fixture = Topic.create(name='/test/topic/4')
fixture = Topic.create(name='/test/topic/5')
topic = TopicsController()
result = topic.list_topics(self.msg)
parsedResponse = json.loads(result)
self.assertEqual('OK', parsedResponse['result'])
self.assertEqual(4, len(parsedResponse['topics']))
def test_missing_keys_missing_topic(self):
payload_dic = {}
payload_dic['client'] = "testClient"
payload_dic['password'] = "******"
topic = TopicsController()
self.assertEqual(
'topic',
Utils().missing_keys(payload_dic, ['password', 'client']))
def test_missing_keys_missing_client(self):
payload_dic = {}
payload_dic['password'] = "******"
payload_dic['topic'] = "/test/topic/1"
topic = TopicsController()
self.assertEqual(
'client',
Utils().missing_keys(payload_dic, ['password', 'client']))
def test_missing_keys_ignoring_topic(self):
payload_dic = {}
payload_dic['client'] = "testClient"
payload_dic['password'] = "******"
payload_dic['topic'] = "/test/topic/1"
topic = TopicsController()
self.assertEqual(
'',
Utils().missing_keys(payload_dic, ['password', 'client'],
topic=False))
def test_missing_keys_missing_options(self):
payload_dic = {}
payload_dic['client'] = "testClient"
payload_dic['password'] = "******"
payload_dic['topic'] = "/test/topic/1"
topic = TopicsController()
self.assertEqual(
'options',
Utils().missing_keys(payload_dic, ['password', 'client'],
options=True))
def test_add_topic_already_added(self):
with test_database(test_db, (Log, Topic), create_tables=True):
topic = TopicsController()
result = topic.add_topic(self.msg)
parsedResponse = json.loads(result)
self.assertIn(self.payload['topic'], parsedResponse['topics'])
def test_add_topic_correct_client(self):
with test_database(test_db, (Log, Topic), create_tables=True):
topic = TopicsController()
result = topic.add_topic(self.msg)
parsedResponse = json.loads(result)
self.assertEqual(self.payload['client'], parsedResponse['client'])
def test_add_topic_response_ok(self):
with test_database(test_db, (Log, Topic), create_tables=True):
topic = TopicsController()
result = topic.add_topic(self.msg)
parsedResponse = json.loads(result)
self.assertEqual('OK', parsedResponse['result'])