class TestSkillManagementPlugin(unittest.TestCase):
@classmethod
def setUpClass(cls):
pass
@classmethod
def tearDownClass(cls):
pass
def setUp(self):
""" setup any state tied to the execution of the given method in a
class. setup_method is invoked for every test method of a class.
"""
args = {"transaction_management":"999"}
args_string = shlex.quote(json.dumps(args))
self.test_subject = SkillManagementPlugin(123,456,None,args_string)
self.test_subject.send_response = MagicMock()
def tearDown(self):
""" teardown any state that was previously setup with a setup_method
call.
"""
client = MongoClient()
client.drop_database(settings.MONGO_DBNAME)
self.test_subject = None
def test_constructor(self):
"""
SkillManagementPlugin.__init__() Test plan:
-ensure name, logger set as parameters
-ensure that mongo is an instance of mongo client
-ensure that a cache db is set up
"""
args = {"transaction_management":"999"}
args_string = shlex.quote(json.dumps(args))
test_subject = SkillManagementPlugin(123,456,None,args_string)
test_subject.logger.should.equal(None)
isinstance(test_subject.mongo,MongoClient).should.equal(True)
isinstance(test_subject.db,Collection).should.equal(True)
def test_get_skill_name_callback(self):
"""
SkillManagementPlugin.get_skill_name_callback() Test plan:
- pass in message without id, should respond with error
- pass in message with bogus id, should respond with error
- pass in message with good id, should respond with name
"""
real_id = self.test_subject.db.insert({"skill_name":"addition"})
msg= {"message_id":"1","skill_id":real_id}
self.test_subject.get_skill_name_callback(msg)
self.test_subject.send_response.assert_called_with("1",{
"skill_name":"addition",
"skill_id": str(real_id),
})
def test_get_skill_name_callback_no_skill_id(self):
"""
SkillManagementPlugin.get_skill_name_callback() No Skill Id
"""
msg = {"message_id":"1"}
self.test_subject.get_skill_name_callback(msg)
self.test_subject.send_response.assert_called_with("1",{
"error":"Message must contain a 'skill_id'",
})
def test_get_skill_name_callback_invalid_id(self):
"""
SkillManagementPlugin.get_skill_name_callback() Invalid Skill Id
"""
msg= {"message_id":"1","skill_id":"foo"}
self.test_subject.get_skill_name_callback(msg)
self.test_subject.send_response.assert_called_with("1",{
"error":"'skill_id' is not a valid ObjectId"
})
def test_get_skill_name_callback_no_exist_skill(self):
"""
SkillManagementPlugin.get_skill_name_callback() Skill Id Does Not Exist
"""
bogus_id = ObjectId()
msg= {"message_id":"1","skill_id":bogus_id}
self.test_subject.get_skill_name_callback(msg)
self.test_subject.send_response.assert_called_with("1",{
"error":"Skill with id " + str(bogus_id) + " does not exist."
})
def test_batch_get_skill_ids(self):
"""
SkillManagementPlugin.batch_get_skill_ids() Test plan:
- send without skill_names, should reply error
- send with skill_names as string, should reply error
- with new skills, should return new ids, should be in db
- with old skills, should returns same ids, should be in db
"""
#no skill_names
msg = {"message_id":"1"}
self.test_subject.batch_get_skill_ids_callback(msg)
self.test_subject.send_response.assert_called_with("1",{
"error":"Message must contain a 'skill_names'"
})
self.test_subject.send_response.reset_mock()
#skill_names not a list
msg["skill_names"]="some_skill"
self.test_subject.batch_get_skill_ids_callback(msg)
self.test_subject.send_response.assert_called_with("1",{
"error":"skill_names must be a list."
})
self.test_subject.send_response.reset_mock()
#new skills
msg["skill_names"] = ["skill_one","skill_two"]
self.test_subject.batch_get_skill_ids_callback(msg)
skill_one = self.test_subject.db.find_one({"skill_name":"skill_one"})
skill_two = self.test_subject.db.find_one({"skill_name":"skill_two"})
skill_one.should_not.equal(None)
skill_two.should_not.equal(None)
self.test_subject.send_response.assert_called_with("1",{
"skill_names": ["skill_one","skill_two"],
"skill_model":"Default",
"skill_ids": {
"skill_one":str(skill_one["_id"]),
"skill_two":str(skill_two["_id"]),
},
"cached":False
})
self.test_subject.send_response.reset_mock()
#existing skills
self.test_subject.batch_get_skill_ids_callback(msg)
self.test_subject.send_response.assert_called_with("1",{
"skill_names": ["skill_one","skill_two"],
"skill_model":"Default",
"skill_ids": {
"skill_one":str(skill_one["_id"]),
"skill_two":str(skill_two["_id"]),
},
"cached":False
})
self.test_subject.send_response.reset_mock()
def test_get_skill_id_callback_no_skill_name(self):
"""
SkillManagementPlugin.get_skill_id_callback() Test plan:
- pass in message without name, should respond with error
- pass in message with name for non existent, should create one
- pass in message with existing name, should return proper id
"""
msg = {"message_id":"1"}
self.test_subject.get_skill_id_callback(msg)
self.test_subject.send_response.assert_called_with("1",{
"error":"Message must contain a 'skill_name'",
})
def test_get_skill_id_callback_new_skill(self):
"""
SkillManagementPlugin.get_skill_id_callback() New Skill
"""
msg = {"message_id":"1","skill_name":"new_addition"}
self.test_subject.get_skill_id_callback(msg)
self.test_subject.db.find({"skill_name":"new_addition"}).count().should.equal(1)
added_id = self.test_subject.db.find_one({"skill_name":"new_addition"})["_id"]
self.test_subject.send_response.assert_called_with("1",{
"skill_name":"new_addition",
"skill_id":str(added_id),
"skill_model":"Default",
"cached":False
})
def test_get_skill_id_callback_existing_skill(self):
"""
SkillManagementPlugin.get_skill_id_callback() Existing Skill
"""
skill_id = self.test_subject.db.insert({"skill_name":"addition","skill_model":"Default"})
msg = {"message_id":"1","skill_name":"addition"}
self.test_subject.get_skill_id_callback(msg)
self.test_subject.send_response.assert_called_with("1",{
"skill_name":"addition",
"skill_id":str(skill_id),
"skill_model":"Default",
"cached":False #this would be true if cached
})
def test_transaction_callback_method(self):
"""
SkillManagementPlugin.transaction_callback_method() Test plan:
- if skill names not present, should respond with error
- if skill_ids not present, should respond with error
- if skill names or skill ids not dict, should respond with error
- if empty, should reply empty
- otherwise, for new skill names, ids should be set
- calling again should render same ids
"""
def mock_send(message_name,payload,callback):
callback({"responder":["downstream"]})
self.test_subject.send = MagicMock(side_effect = mock_send)
self.test_subject.send_response = MagicMock()
#access denied
msg = {"message_id":"1","orig_sender_id":"3","sender_entity_id":"888"}
self.test_subject.transaction_callback_method(msg)
self.test_subject.send_response.assert_called_with("1",{ "error" : "Access denied","responder":"skill",})
self.test_subject.send_response.reset_mock()
#no args
msg = {"message_id":"1","orig_sender_id":"2","sender_entity_id":"999"}
self.test_subject.transaction_callback_method(msg)
self.test_subject.send_response.assert_called_with("1",{
"error":"skill manager transactions requires 'skill_ids' and 'skill_names'",
"responder":"skill",
})
self.test_subject.send_response.reset_mock()
#skill names only
msg["skill_names"] = {"Default":"skill_name"}
self.test_subject.transaction_callback_method(msg)
self.test_subject.send_response.assert_called_with("1",{
"error":"skill manager transactions requires 'skill_ids' and 'skill_names'",
"responder":"skill",
})
self.test_subject.send_response.reset_mock()
#skill_ids not dict
msg["skill_ids"] = "not dict"
self.test_subject.transaction_callback_method(msg)
self.test_subject.send_response.assert_called_with("1",{
"error":"invalid skill_names or skill_ids for skill manager transaction",
"responder":"skill",
})
self.test_subject.send_response.reset_mock()
#skill names not dict
msg["skill_ids"] = {}
msg["skill_names"] = "not dict"
self.test_subject.transaction_callback_method(msg)
self.test_subject.send_response.assert_called_with("1",{
"error":"invalid skill_names or skill_ids for skill manager transaction",
"responder":"skill",
})
self.test_subject.send_response.reset_mock()
#empty skills
msg["skill_ids"] = {}
msg["skill_names"] = {}
self.test_subject.transaction_callback_method(msg)
self.test_subject.send_response.assert_called_with("1",{
"skill_ids" : {},
"responder":"skill",
})
self.test_subject.send_response.reset_mock()
#create new ids
msg["skill_ids"] = {}
msg["skill_names"] = {"Default":"skill_name"}
self.test_subject.transaction_callback_method(msg)
skill = self.test_subject.db.find_one({"skill_name":"skill_name"})
skill.should_not.equal(None)
self.test_subject.send_response.assert_called_with("1",{
"skill_ids":{"skill_name" :str(skill["_id"])},
"responder":"skill",
})
self.test_subject.send_response.reset_mock()
#should be the same id
self.test_subject.transaction_callback_method(msg)
self.test_subject.send_response.assert_called_with("1",{
"skill_ids":{"skill_name" :str(skill["_id"])},
"responder":"skill",
})
self.test_subject.send_response.reset_mock()