mocks = mock.Mock() mocks.attach_mock(mock_init, 'init') mocks.attach_mock(mock_set, 'set') mocks.attach_mock(mock_parser, 'parser') mocks.attach_mock(mock_exists, 'exists') mock_exists.side_effect = [False, True] instance_config.InstanceConfig() expected_calls = [ mock.call.init( config_file='template', config_header='/tmp/test.py template'), mock.call.exists('config'), mock.call.exists('distro'), mock.call.parser.SafeConfigParser(), mock.call.parser.SafeConfigParser().read('distro'), mock.call.parser.SafeConfigParser().sections(), mock.call.set('a', 'key: a', 'value: a', overwrite=False), mock.call.set('b', 'key: b', 'value: b', overwrite=False), ] self.assertEqual(mocks.mock_calls, expected_calls) @mock.patch('google_compute_engine.instance_setup.instance_config.config_manager.ConfigManager.WriteConfig') def testWriteConfig(self, mock_write): mock_config = instance_config.InstanceConfig() instance_config.InstanceConfig.WriteConfig(mock_config) mock_write.assert_called_once_with(config_file='config') if __name__ == '__main__': unittest.main()
def testGetMetadata(self): mock_response = mock.Mock() mock_response.return_value = {} self.mock_watcher._HandleMetadataUpdate = mock_response self.assertEqual(self.mock_watcher.GetMetadata(), {}) mock_response.assert_called_once_with( metadata_key='', recursive=True, wait=False, timeout=None, retry=True) self.mock_watcher.logger.exception.assert_not_called() def testGetMetadataArgs(self): mock_response = mock.Mock() mock_response.return_value = {} self.mock_watcher._HandleMetadataUpdate = mock_response metadata_key = 'instance/id' recursive = False retry = False response = self.mock_watcher.GetMetadata( metadata_key=metadata_key, recursive=recursive, timeout=60, retry=retry) self.assertEqual(response, {}) mock_response.assert_called_once_with( metadata_key=metadata_key, recursive=False, wait=False, timeout=60, retry=False) self.mock_watcher.logger.exception.assert_not_called() if __name__ == '__main__': unittest.main()