import logging import unittest import re import urlparse from sleepwalker.datarep import ListDataRep from sim_server import SimServer, UnknownUsername, BadPassword, MissingAuthHeader from service_loader import ServiceDefLoader, SERVICE_MANAGER, CONNECTION_MANAGER, TEST_SERVER_MANAGER logger = logging.getLogger(__name__) TEST_PATH = os.path.abspath(os.path.dirname(__file__)) ServiceDefLoader.register_servicedef( "http://support.riverbed.com/apis/crossref.foo/1.0", os.path.join(TEST_PATH, "service_crossref_foo.yml") ) ServiceDefLoader.register_servicedef( "http://support.riverbed.com/apis/crossref.bar/1.0", os.path.join(TEST_PATH, "service_crossref_bar.yml") ) class CrossRefFooServer(SimServer): def __init__(self, *args, **kwargs): SimServer.__init__(self, *args, **kwargs) self.add_collection("foos", "foo") def embed_bar_links_get(self, link, method, uri, data, params, headers): return {"name": "embed_bar", "bar": {"id": 4, "name": "bar"}}
import copy from sleepwalker.datarep import DataRep from sleepwalker.exceptions import DataPullError from sim_server import SimServer, BadPassword from service_loader import \ ServiceDefLoader, SERVICE_MANAGER, TEST_SERVER_MANAGER logger = logging.getLogger(__name__) TEST_PATH = os.path.abspath(os.path.dirname(__file__)) ServiceDefLoader.register_servicedef( 'http://support.riverbed.com/apis/basic/1.0', os.path.join(TEST_PATH, "service_basic.yml")) class BasicServer(SimServer): def __init__(self, *args, **kwargs): SimServer.__init__(self, *args, **kwargs) self.add_collection('items', 'item') self.add_collection('categories', 'category') self.add_collection('nullints', 'nullint') self._x = 5 def x_links_get(self, link, method, uri, data, params, headers): return self._x