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)