def test_key_present(): """ Test to ensure key pair is present. """ name = "mykeypair" upublic = "salt://mybase/public_key.pub" ret = {"name": name, "result": True, "changes": {}, "comment": ""} mock = MagicMock(side_effect=[True, False, False]) mock_bool = MagicMock(side_effect=[IOError, True]) with patch.dict( boto_ec2.__salt__, {"boto_ec2.get_key": mock, "cp.get_file_str": mock_bool} ): comt = "The key name {} already exists".format(name) ret.update({"comment": comt}) assert boto_ec2.key_present(name) == ret comt = "File {} not found.".format(upublic) ret.update({"comment": comt, "result": False}) assert boto_ec2.key_present(name, upload_public=upublic) == ret with patch.dict(boto_ec2.__opts__, {"test": True}): comt = "The key {} is set to be created.".format(name) ret.update({"comment": comt, "result": None}) assert boto_ec2.key_present(name, upload_public=upublic) == ret
def test_key_present(self): ''' Test to ensure key pair is present. ''' name = 'mykeypair' upublic = 'salt://mybase/public_key.pub' ret = {'name': name, 'result': True, 'changes': {}, 'comment': ''} mock = MagicMock(side_effect=[True, False, False]) mock_bool = MagicMock(side_effect=[IOError, True]) with patch.dict(boto_ec2.__salt__, {'boto_ec2.get_key': mock, 'cp.get_file_str': mock_bool}): comt = ('The key name {0} already exists'.format(name)) ret.update({'comment': comt}) self.assertDictEqual(boto_ec2.key_present(name), ret) comt = ('File {0} not found.'.format(upublic)) ret.update({'comment': comt, 'result': False}) self.assertDictEqual(boto_ec2.key_present(name, upload_public=upublic), ret) with patch.dict(boto_ec2.__opts__, {'test': True}): comt = ('The key {0} is set to be created.'.format(name)) ret.update({'comment': comt, 'result': None}) self.assertDictEqual(boto_ec2.key_present (name, upload_public=upublic), ret)
def test_key_present(self): ''' Test to ensure key pair is present. ''' name = 'mykeypair' upublic = 'salt://mybase/public_key.pub' ret = {'name': name, 'result': True, 'changes': {}, 'comment': ''} mock = MagicMock(side_effect=[True, False, False]) mock_bool = MagicMock(side_effect=[IOError, True]) with patch.dict(boto_ec2.__salt__, {'boto_ec2.get_key': mock, 'cp.get_file_str': mock_bool}): comt = ('The key name {0} already exists'.format(name)) ret.update({'comment': comt}) self.assertDictEqual(boto_ec2.key_present(name), ret) comt = ('File {0} not found.'.format(upublic)) ret.update({'comment': comt, 'result': False}) self.assertDictEqual(boto_ec2.key_present(name, upload_public=upublic), ret) with patch.dict(boto_ec2.__opts__, {'test': True}): comt = ('The key {0} is set to be created.'.format(name)) ret.update({'comment': comt, 'result': None}) self.assertDictEqual(boto_ec2.key_present (name, upload_public=upublic), ret)