def test_commit_refresh_separate_stores(self): """Refresh-commit cycle from separate stores restores HostInfo.""" info = host_info.HostInfo(labels=['labels'], attributes={'attrib': 'value'}) store = file_store.FileStore(self._store_file) store.commit(info) read_store = file_store.FileStore(self._store_file) got = read_store.get() self.assertEqual(info, got)
def test_commit_to_bad_path_raises(self): """Commit to a non-writable path raises StoreError.""" # file_lock_timeout of 0 forces no retries (speeds up the test) store = file_store.FileStore('/rooty/non-writable/path/mostly', file_lock_timeout_seconds=0) with self.assertRaises(host_info.StoreError): store.commit(host_info.HostInfo())
def test_commit_refresh_round_trip(self): """Refresh-commit cycle from a single store restores HostInfo.""" info = host_info.HostInfo(labels=['labels'], attributes={'attrib': 'value'}) store = file_store.FileStore(self._store_file) store.commit(info) got = store.get(force_refresh=True) self.assertEqual(info, got)
def test_refresh_from_non_existent_path_raises(self): """Refresh from a non-existent backing file raises StoreError.""" # file_lock_timeout of 0 forces no retries (speeds up the test) store = file_store.FileStore(self._store_file, file_lock_timeout_seconds=0) store.commit(host_info.HostInfo()) os.unlink(self._store_file) with self.assertRaises(host_info.StoreError): store.get(force_refresh=True)
def test_refresh_from_unreadable_path_raises(self): """Refresh from an unreadable backing file raises StoreError.""" # file_lock_timeout of 0 forces no retries (speeds up the test) store = file_store.FileStore(self._store_file, file_lock_timeout_seconds=0) store.commit(host_info.HostInfo()) old_mode = os.stat(self._store_file).st_mode os.chmod(self._store_file, old_mode & ~stat.S_IRUSR) self.addCleanup(os.chmod, self._store_file, old_mode) with self.assertRaises(host_info.StoreError): store.get(force_refresh=True)
def test_commit_blocks_for_locked_file(self): """Commit blocks when the backing file is locked. This is a greybox test. We artificially lock the backing file. This test intentionally uses a real locking.FileLock to ensure that locking API is used correctly. """ # file_lock_timeout of 0 forces no retries (speeds up the test) store = file_store.FileStore(self._store_file, file_lock_timeout_seconds=0) file_lock = locking.FileLock(store._lock_path, locktype=locking.FLOCK) with file_lock.lock(), self.assertRaises(host_info.StoreError): store.commit(host_info.HostInfo())
def test_commit_with_negative_timeout_clips(self, mock_file_lock_class): """Commit request with negative timeout is same as 0 timeout. @param mock_file_lock_class: A patched version of the locking.FileLock class. """ mock_file_lock = mock_file_lock_class.return_value mock_file_lock.__enter__.return_value = mock_file_lock mock_file_lock.write_lock.side_effect = ( locking.LockNotAcquiredError('Testing error')) store = file_store.FileStore(self._store_file, file_lock_timeout_seconds=-1) with self.assertRaises(host_info.StoreError): store.commit(host_info.HostInfo()) self.assertEqual(1, mock_file_lock.write_lock.call_count)
def test_commit_succeeds_after_lock_retry(self, mock_file_lock_class): """Tests that commit succeeds when locking requires retries. @param mock_file_lock_class: A patched version of the locking.FileLock class. """ mock_file_lock = mock_file_lock_class.return_value mock_file_lock.__enter__.return_value = mock_file_lock mock_file_lock.write_lock.side_effect = [ locking.LockNotAcquiredError('Testing error'), True, ] store = file_store.FileStore(self._store_file, file_lock_timeout_seconds=0.1) store.commit(host_info.HostInfo()) self.assertEqual(2, mock_file_lock.write_lock.call_count)
def _parse_hostinfo_keyval(job_dir, hostname): """ Parse host keyvals from hostinfo. @param job_dir: The string directory name of the associated job. @param hostname: The string hostname. @return A dictionary representing the host keyvals. """ # The hostinfo path looks like: # host_info_store/chromeos6-row4-rack11-host6.store # # TODO(ayatane): We should pass hostinfo path explicitly. subdir = 'host_info_store' hostinfo_path = os.path.join(job_dir, subdir, hostname + '.store') store = file_store.FileStore(hostinfo_path) hostinfo = store.get() # TODO(ayatane): Investigate if urllib.quote is better. label_string = ','.join( label.replace(':', '%3A') for label in hostinfo.labels) return {'labels': label_string, 'platform': hostinfo.model}
def _read_store(path): """Read a HostInfo from a file at path.""" store = file_store.FileStore(path) return store.get()
def test_empty_store_raises_on_get(self): """Refresh from store before commit raises StoreError""" store = file_store.FileStore(self._store_file) with self.assertRaises(host_info.StoreError): store.get()
def test_str(self): """Sanity tests the __str__ implementaiton""" store = file_store.FileStore('/foo/path') self.assertEqual(str(store), 'FileStore[/foo/path]')