def setUp(self):
self.temp_dir = tempfile.mkdtemp()
self.config_file_path = os.path.join(self.temp_dir, "foo")
with open(self.config_file_path, "w") as config_file:
config_file.write("")
self.conn = Mock()
self.conn.__enter__ = Mock(return_value=(Mock(), None))
self.conn.__exit__ = Mock(return_value=None)
self.cursor = FakeCursor()
self.db_singleton = Mock()
self.os_env = dict(USER="******")
self.os_env["MASTER_DATA_DIRECTORY"] = self.temp_dir
self.os_env["GPHOME"] = self.temp_dir
self.gparray = self._create_gparray_with_2_primary_2_mirrors()
self.pool = Mock()
self.pool.getCompletedItems.return_value = []
self.pgconf_dict = gucdict()
self.pgconf_dict["port"] = setting("port", "123", None, None, None)
self.pgconf_dict["max_connection"] = setting("max_connections", "1", None, None, None)
self.config_provider_mock = MagicMock(spec=GpConfigurationProvider)
self.config_provider_mock.initializeProvider.return_value = self.config_provider_mock
self.gpArrayMock = MagicMock(spec=GpArray)
self.gpArrayMock.getDbList.side_effect = [[self.primary0], [self.primary0], [self.primary0]]
self.gpArrayMock.hasMirrors = True
self.gpArrayMock.isStandardArray.return_value = (True, None)
self.gpArrayMock.master = self.gparray.master
self.config_provider_mock.loadSystemConfig.return_value = self.gpArrayMock
self.mirror_to_build = GpMirrorToBuild(self.mirror0, self.primary0, None, False)
self.apply_patches([
patch('os.environ', new=self.os_env),
patch('gppylib.db.dbconn.connect', return_value=self.conn),
patch('gppylib.db.dbconn.execSQL', return_value=self.cursor),
patch('gppylib.db.dbconn.execSQLForSingletonRow', return_value=["foo"]),
patch('gppylib.pgconf.readfile', return_value=self.pgconf_dict),
patch('gppylib.commands.gp.GpVersion'),
patch('gppylib.system.faultProberInterface.getFaultProber'),
patch('gppylib.system.configurationInterface.getConfigurationProvider', return_value=self.config_provider_mock),
patch('gppylib.commands.base.WorkerPool', return_value=self.pool),
patch('gppylib.gparray.GpArray.getSegmentsByHostName', return_value={}),
patch('gppylib.gplog.get_default_logger'),
patch.object(GpMirrorListToBuild, "__init__", return_value=None),
patch.object(GpMirrorListToBuild, "buildMirrors"),
patch.object(GpMirrorListToBuild, "getAdditionalWarnings"),
patch.object(GpMirrorListToBuild, "getMirrorsToBuild"),
patch.object(HeapChecksum, "check_segment_consistency"),
patch.object(HeapChecksum, "get_segments_checksum_settings"),
])
self.call_count = 0
self.return_one = True
self.mock_build_mirrors = self.get_mock_from_apply_patch("buildMirrors")
self.mock_get_mirrors_to_build = self.get_mock_from_apply_patch('getMirrorsToBuild')
self.mock_heap_checksum_init = self.get_mock_from_apply_patch("__init__")
self.mock_check_segment_consistency = self.get_mock_from_apply_patch('check_segment_consistency')
self.mock_get_segments_checksum_settings = self.get_mock_from_apply_patch('get_segments_checksum_settings')
sys.argv = ["gprecoverseg"] # reset to relatively empty args list
options = Options()
options.masterDataDirectory = self.temp_dir
options.spareDataDirectoryFile = self.config_file_path
# import HERE so that patches are already in place!
from programs.clsRecoverSegment import GpRecoverSegmentProgram
self.subject = GpRecoverSegmentProgram(options)
self.subject.logger = Mock(spec=['log', 'warn', 'info', 'debug', 'error', 'warning', 'fatal'])
faultProberInterface.gFaultProber = Mock()
def setUp(self):
self.temp_dir = tempfile.mkdtemp()
self.config_file_path = os.path.join(self.temp_dir, "foo")
with open(self.config_file_path, "w") as config_file:
config_file.write("")
self.conn = Mock()
self.conn.__enter__ = Mock(return_value=(Mock(), None))
self.conn.__exit__ = Mock(return_value=None)
self.cursor = FakeCursor()
self.db_singleton = Mock()
self.os_env = dict(USER="******")
self.os_env["COORDINATOR_DATA_DIRECTORY"] = self.temp_dir
self.os_env["GPHOME"] = self.temp_dir
self.gparray = self._create_gparray_with_2_primary_2_mirrors()
self.pool = Mock()
self.pool.getCompletedItems.return_value = []
self.pgconf_dict = gucdict()
self.pgconf_dict["port"] = setting("port", "123", None, None, None)
self.pgconf_dict["max_connection"] = setting("max_connections", "1",
None, None, None)
self.config_provider_mock = MagicMock(spec=GpConfigurationProvider)
self.config_provider_mock.initializeProvider.return_value = self.config_provider_mock
self.gpArrayMock = MagicMock(spec=GpArray)
self.gpArrayMock.getDbList.side_effect = [[self.primary0],
[self.primary0],
[self.primary0]]
self.gpArrayMock.segmentPairs = []
self.gpArrayMock.hasMirrors = True
self.gpArrayMock.isStandardArray.return_value = (True, None)
self.gpArrayMock.coordinator = self.gparray.coordinator
self.config_provider_mock.loadSystemConfig.return_value = self.gpArrayMock
self.mirror_to_build = GpMirrorToBuild(self.mirror0, self.primary0,
None, False)
self.apply_patches([
patch('os.environ', new=self.os_env),
patch('gppylib.db.dbconn.connect', return_value=self.conn),
patch('gppylib.db.dbconn.query', return_value=self.cursor),
patch('gppylib.db.dbconn.queryRow', return_value=["foo"]),
patch('gppylib.pgconf.readfile', return_value=self.pgconf_dict),
patch('gppylib.commands.gp.GpVersion'),
patch('gppylib.system.faultProberInterface.getFaultProber'),
patch(
'gppylib.system.configurationInterface.getConfigurationProvider',
return_value=self.config_provider_mock),
patch('gppylib.commands.base.WorkerPool', return_value=self.pool),
patch('gppylib.gparray.GpArray.getSegmentsByHostName',
return_value={}),
patch('gppylib.gplog.get_default_logger'),
patch.object(GpMirrorListToBuild, "__init__", return_value=None),
patch.object(GpMirrorListToBuild, "buildMirrors"),
patch.object(GpMirrorListToBuild, "getAdditionalWarnings"),
patch.object(GpMirrorListToBuild, "getMirrorsToBuild"),
patch.object(HeapChecksum, "check_segment_consistency"),
patch.object(HeapChecksum, "get_segments_checksum_settings"),
])
self.call_count = 0
self.return_one = True
self.mock_build_mirrors = self.get_mock_from_apply_patch(
"buildMirrors")
self.mock_get_mirrors_to_build = self.get_mock_from_apply_patch(
'getMirrorsToBuild')
self.mock_heap_checksum_init = self.get_mock_from_apply_patch(
"__init__")
self.mock_check_segment_consistency = self.get_mock_from_apply_patch(
'check_segment_consistency')
self.mock_get_segments_checksum_settings = self.get_mock_from_apply_patch(
'get_segments_checksum_settings')
sys.argv = ["gprecoverseg"] # reset to relatively empty args list
options = Options()
options.coordinatorDataDirectory = self.temp_dir
options.spareDataDirectoryFile = self.config_file_path
options.showProgress = True
options.showProgressInplace = True
# import HERE so that patches are already in place!
from gppylib.programs.clsRecoverSegment import GpRecoverSegmentProgram
self.subject = GpRecoverSegmentProgram(options)
self.subject.logger = Mock(
spec=['log', 'warn', 'info', 'debug', 'error', 'warning', 'fatal'])
faultProberInterface.gFaultProber = Mock()
