async def setup_testcase(self):
self.stubs = stubout.StubOutForTesting()
self.addCleanup(self.stubs.SmartUnsetAll)
# Setup fake filesystem.
self.fs = fake_filesystem.FakeFilesystem()
self.fake_os = fake_filesystem.FakeOsModule(self.fs)
self.fake_open = fake_filesystem.FakeFileOpen(self.fs)
self.stubs.SmartSet(storage, 'os', self.fake_os)
self.stubs.SmartSet(fileutil, 'os', self.fake_os)
self.stubs.SmartSet(project_registry, 'os', self.fake_os)
self.stubs.SmartSet(shutil, 'os', self.fake_os)
self.stubs.SmartSet(builtins, 'open', self.fake_open)
self.music_dir = self.fake_os.path.expanduser('~/Music/Noisicaä')
self.fake_os.makedirs(self.music_dir)
storage.ProjectStorage.create(
self.fake_os.path.join(self.music_dir, 'proj1')).close()
storage.ProjectStorage.create(
self.fake_os.path.join(self.music_dir, 'proj2')).close()
storage.ProjectStorage.create(
self.fake_os.path.join(self.music_dir, 'proj3')).close()
self.__registry = project_registry.ProjectRegistry(
context=self.context)
await self.__registry.setup()
def setUp(self):
super(WebSocketRequestHandlerTestCase, self).setUp()
self.stubs = stubout.StubOutForTesting()
self.tmpdir = tempfile.mkdtemp('-websockify-tests')
# Mock this out cause it screws tests up
self.stubs.Set(os, 'chdir', lambda *args, **kwargs: None)
self.stubs.Set(SimpleHTTPRequestHandler, 'send_response',
lambda *args, **kwargs: None)
def setUp(self):
super(MoxStubout, self).setUp()
self.mox = mox.Mox()
self.stubs = stubout.StubOutForTesting()
self.addCleanup(self.mox.UnsetStubs)
self.addCleanup(self.stubs.UnsetAll)
self.addCleanup(self.stubs.SmartUnsetAll)
self.addCleanup(self.mox.VerifyAll)
def setUp(self):
super(ProxyRequestHandlerTestCase, self).setUp()
self.stubs = stubout.StubOutForTesting()
self.handler = websocketproxy.ProxyRequestHandler(
FakeSocket(''), "127.0.0.1", FakeServer())
self.handler.path = "https://localhost:6080/websockify?token=blah"
self.handler.headers = None
self.stubs.Set(websocket.WebSocketServer, 'socket',
staticmethod(lambda *args, **kwargs: None))
def setup_testcase(self):
self.stubs = stubout.StubOutForTesting()
self.addCleanup(self.stubs.SmartUnsetAll)
# Setup fake filesystem.
self.fs = fake_filesystem.FakeFilesystem()
self.fake_os = fake_filesystem.FakeOsModule(self.fs)
self.fake_open = fake_filesystem.FakeFileOpen(self.fs)
self.stubs.SmartSet(builtins, 'open', self.fake_open)
def testSmartSetOnModule(self):
mock_function = self.mox.CreateMockAnything()
mock_function()
stubber = stubout.StubOutForTesting()
stubber.SmartSet(stubout_helper, 'SampleFunction', mock_function)
self.mox.ReplayAll()
stubout_helper.SampleFunction()
self.mox.VerifyAll()
def setUp(self):
super(WebSockifyRequestHandlerTestCase, self).setUp()
self.stubs = stubout.StubOutForTesting()
self.tmpdir = tempfile.mkdtemp('-websockify-tests')
# Mock this out cause it screws tests up
self.stubs.Set(os, 'chdir', lambda *args, **kwargs: None)
self.stubs.Set(BaseHTTPRequestHandler, 'send_response',
lambda *args, **kwargs: None)
def fake_send_error(self, code, message=None, explain=None):
self.last_code = code
self.stubs.Set(BaseHTTPRequestHandler, 'send_error', fake_send_error)
async def setup_testcase(self):
self.stubs = stubout.StubOutForTesting()
self.addCleanup(self.stubs.SmartUnsetAll)
# Setup fake filesystem.
self.fs = fake_filesystem.FakeFilesystem()
self.fake_os = fake_filesystem.FakeOsModule(self.fs)
self.fake_open = fake_filesystem.FakeFileOpen(self.fs)
self.stubs.SmartSet(storage, 'os', self.fake_os)
self.stubs.SmartSet(fileutil, 'os', self.fake_os)
self.stubs.SmartSet(builtins, 'open', self.fake_open)
self.setup_writer_process(inline=True)
create_process_response = editor_main_pb2.CreateProcessResponse()
await self.process_manager_client.call('CREATE_WRITER_PROCESS', None,
create_process_response)
self.writer_address = create_process_response.address
self.writer_client = writer_client.WriterClient(event_loop=self.loop)
await self.writer_client.setup()
await self.writer_client.connect(self.writer_address)
def setUp(self):
"""Patch os.environ to avoid required auth info."""
super(ShellTests, self).setUp()
self.stubs = stubout.StubOutForTesting()
self.fake_client = fakes.FakeHTTPClient()
self.stubs.Set(
httpclient.HTTPClient, "_cs_request",
lambda ign_self, *args, **kwargs: self.fake_client._cs_request(
*args, **kwargs))
self.stubs.Set(httpclient.HTTPClient, "authenticate",
lambda cl_obj: self.fake_client.authenticate(cl_obj))
self.old_environment = os.environ.copy()
os.environ = {
'OS_USERNAME': DEFAULT_USERNAME,
'OS_PASSWORD': DEFAULT_PASSWORD,
'OS_TENANT_ID': DEFAULT_TENANT_ID,
'OS_TENANT_NAME': DEFAULT_TENANT_NAME,
'OS_AUTH_URL': DEFAULT_AUTH_URL,
}
import keystoneclient.shell
self.shell = keystoneclient.shell.OpenStackIdentityShell()
def setUp(self):
super(WebSockifyServerTestCase, self).setUp()
self.stubs = stubout.StubOutForTesting()
self.tmpdir = tempfile.mkdtemp('-websockify-tests')
# Mock this out cause it screws tests up
self.stubs.Set(os, 'chdir', lambda *args, **kwargs: None)
