| /** |
| * @fileoverview Tests for storage.js. |
| */ |
| goog.module('protobuf.runtime.BinaryStorageTest'); |
| |
| goog.setTestOnly(); |
| |
| const BinaryStorage = goog.require('protobuf.runtime.BinaryStorage'); |
| const {Field} = goog.require('protobuf.binary.field'); |
| |
| /** |
| * @type {number} |
| */ |
| const DEFAULT_PIVOT = 24; |
| |
| const /** !Field */ field1 = |
| Field.fromDecodedValue(/* decodedValue= */ 1, /* encoder= */ () => {}); |
| const /** !Field */ field2 = |
| Field.fromDecodedValue(/* decodedValue= */ 2, /* encoder= */ () => {}); |
| const /** !Field */ field3 = |
| Field.fromDecodedValue(/* decodedValue= */ 3, /* encoder= */ () => {}); |
| const /** !Field */ field4 = |
| Field.fromDecodedValue(/* decodedValue= */ 4, /* encoder= */ () => {}); |
| |
| /** |
| * Returns the number of fields stored. |
| * |
| * @param {!BinaryStorage} storage |
| * @return {number} |
| */ |
| function getStorageSize(storage) { |
| let size = 0; |
| storage.forEach(() => void size++); |
| return size; |
| } |
| |
| describe('BinaryStorage', () => { |
| it('sets and gets a field not greater than the pivot', () => { |
| const storage = new BinaryStorage(DEFAULT_PIVOT); |
| |
| storage.set(1, field1); |
| storage.set(DEFAULT_PIVOT, field2); |
| |
| expect(storage.getPivot()).toBe(DEFAULT_PIVOT); |
| expect(storage.get(1)).toBe(field1); |
| expect(storage.get(DEFAULT_PIVOT)).toBe(field2); |
| }); |
| |
| it('sets and gets a field greater than the pivot', () => { |
| const storage = new BinaryStorage(DEFAULT_PIVOT); |
| |
| storage.set(DEFAULT_PIVOT + 1, field1); |
| storage.set(100000, field2); |
| |
| expect(storage.get(DEFAULT_PIVOT + 1)).toBe(field1); |
| expect(storage.get(100000)).toBe(field2); |
| }); |
| |
| it('sets and gets a field when pivot is zero', () => { |
| const storage = new BinaryStorage(0); |
| |
| storage.set(0, field1); |
| storage.set(100000, field2); |
| |
| expect(storage.getPivot()).toBe(0); |
| expect(storage.get(0)).toBe(field1); |
| expect(storage.get(100000)).toBe(field2); |
| }); |
| |
| it('sets and gets a field when pivot is undefined', () => { |
| const storage = new BinaryStorage(); |
| |
| storage.set(0, field1); |
| storage.set(DEFAULT_PIVOT, field2); |
| storage.set(DEFAULT_PIVOT + 1, field3); |
| |
| expect(storage.getPivot()).toBe(DEFAULT_PIVOT); |
| expect(storage.get(0)).toBe(field1); |
| expect(storage.get(DEFAULT_PIVOT)).toBe(field2); |
| expect(storage.get(DEFAULT_PIVOT + 1)).toBe(field3); |
| }); |
| |
| it('returns undefined for nonexistent fields', () => { |
| const storage = new BinaryStorage(DEFAULT_PIVOT); |
| |
| expect(storage.get(1)).toBeUndefined(); |
| expect(storage.get(DEFAULT_PIVOT)).toBeUndefined(); |
| expect(storage.get(DEFAULT_PIVOT + 1)).toBeUndefined(); |
| expect(storage.get(100000)).toBeUndefined(); |
| }); |
| |
| it('returns undefined for nonexistent fields after map initialization', |
| () => { |
| const storage = new BinaryStorage(DEFAULT_PIVOT); |
| storage.set(100001, field1); |
| |
| expect(storage.get(1)).toBeUndefined(); |
| expect(storage.get(DEFAULT_PIVOT)).toBeUndefined(); |
| expect(storage.get(DEFAULT_PIVOT + 1)).toBeUndefined(); |
| expect(storage.get(100000)).toBeUndefined(); |
| }); |
| |
| it('deletes a field in delete() when values are only in array', () => { |
| const storage = new BinaryStorage(DEFAULT_PIVOT); |
| storage.set(1, field1); |
| |
| storage.delete(1); |
| |
| expect(storage.get(1)).toBeUndefined(); |
| }); |
| |
| it('deletes a field in delete() when values are both in array and map', |
| () => { |
| const storage = new BinaryStorage(DEFAULT_PIVOT); |
| storage.set(DEFAULT_PIVOT, field2); |
| storage.set(DEFAULT_PIVOT + 1, field3); |
| |
| storage.delete(DEFAULT_PIVOT); |
| storage.delete(DEFAULT_PIVOT + 1); |
| |
| expect(storage.get(DEFAULT_PIVOT)).toBeUndefined(); |
| expect(storage.get(DEFAULT_PIVOT + 1)).toBeUndefined(); |
| }); |
| |
| it('deletes a field in delete() when values are only in map', () => { |
| const storage = new BinaryStorage(DEFAULT_PIVOT); |
| storage.set(100000, field4); |
| |
| storage.delete(100000); |
| |
| expect(storage.get(100000)).toBeUndefined(); |
| }); |
| |
| it('loops over all the elements in forEach()', () => { |
| const storage = new BinaryStorage(DEFAULT_PIVOT); |
| storage.set(1, field1); |
| storage.set(DEFAULT_PIVOT, field2); |
| storage.set(DEFAULT_PIVOT + 1, field3); |
| storage.set(100000, field4); |
| |
| const fields = new Map(); |
| storage.forEach( |
| (field, fieldNumber) => void fields.set(fieldNumber, field)); |
| |
| expect(fields.size).toEqual(4); |
| expect(fields.get(1)).toBe(field1); |
| expect(storage.get(DEFAULT_PIVOT)).toBe(field2); |
| expect(storage.get(DEFAULT_PIVOT + 1)).toBe(field3); |
| expect(fields.get(100000)).toBe(field4); |
| }); |
| |
| it('creates a shallow copy of the storage in shallowCopy()', () => { |
| const storage = new BinaryStorage(DEFAULT_PIVOT); |
| storage.set(1, field1); |
| storage.set(100000, field2); |
| |
| const copy = storage.shallowCopy(); |
| |
| expect(getStorageSize(copy)).toEqual(2); |
| expect(copy.get(1)).not.toBe(field1); |
| expect(copy.get(1).getDecodedValue()).toEqual(1); |
| expect(copy.get(100000)).not.toBe(field1); |
| expect(copy.get(100000).getDecodedValue()).toEqual(2); |
| }); |
| }); |