def test_get_capabilities(self, connection): """ Test get_capabilities """ socket_path = "little red riding hood" module = MagicMock(spec=[ 'fail_json', ]) module._socket_path = socket_path module.fail_json.side_effect = TypeError capabilities = {'network_api': 'cliconf'} connection().get_capabilities.return_value = json.dumps(capabilities) capabilities_returned = slxos.get_capabilities(module) self.assertEqual(capabilities, capabilities_returned)