def test_pgdump(self): db = self.PartialMockDb(None, None) self.mox.StubOutWithMock(db, 'get_db_args_env') self.mox.StubOutWithMock(External, 'run') pgdumpargs = ['-d', 'name', '-h', 'host', '-U', 'user', '-w', 'arg1', 'arg2'] db.get_db_args_env().AndReturn(self.create_db_test_params()) External.run('pg_dump', pgdumpargs, capturestd=True, env={'PGPASSWORD': '******'}).AndReturn(('', '')) self.mox.ReplayAll() db.pgdump('arg1', 'arg2') self.mox.VerifyAll()
def test_pgdump(self): db = self.PartialMockDb(None, None) self.mox.StubOutWithMock(db, 'get_db_args_env') self.mox.StubOutWithMock(External, 'run') pgdumpargs = [ '-d', 'name', '-h', 'host', '-U', 'user', '-w', 'arg1', 'arg2' ] db.get_db_args_env().AndReturn(self.create_db_test_params()) External.run('pg_dump', pgdumpargs, capturestd=True, env={ 'PGPASSWORD': '******' }).AndReturn(('', '')) self.mox.ReplayAll() db.pgdump('arg1', 'arg2') self.mox.VerifyAll()
def test_dump(self, dumpfile, dryrun): args = self.Args({'dry_run': dryrun, 'dumpfile': dumpfile}) db = self.PartialMockDb(args, None) self.mox.StubOutWithMock(omego.fileutils, 'timestamp_filename') self.mox.StubOutWithMock(db, 'get_db_args_env') self.mox.StubOutWithMock(db, 'pgdump') if not dumpfile: db.get_db_args_env().AndReturn(self.create_db_test_params()) dumpfile = 'omero-database-name-00000000-000000-000000.pgdump' omego.fileutils.timestamp_filename('omero-database-name', 'pgdump').AndReturn(dumpfile) if not dryrun: db.pgdump('-Fc', '-f', dumpfile).AndReturn('') self.mox.ReplayAll() db.dump() self.mox.VerifyAll()
def test_dump(self, dumpfile, dryrun): args = self.Args({'dry_run': dryrun, 'dumpfile': dumpfile}) db = self.PartialMockDb(args, None) self.mox.StubOutWithMock(omego.fileutils, 'timestamp_filename') self.mox.StubOutWithMock(db, 'get_db_args_env') self.mox.StubOutWithMock(db, 'pgdump') if not dumpfile: db.get_db_args_env().AndReturn(self.create_db_test_params()) dumpfile = 'omero-database-name-00000000-000000-000000.pgdump' omego.fileutils.timestamp_filename( 'omero-database-name', 'pgdump').AndReturn(dumpfile) if not dryrun: db.pgdump('-Fc', '-f', dumpfile).AndReturn('') self.mox.ReplayAll() db.dump() self.mox.VerifyAll()