diff --git a/src/style_manager/index.js b/src/style_manager/index.js index 979da314e..aa6203915 100644 --- a/src/style_manager/index.js +++ b/src/style_manager/index.js @@ -177,12 +177,16 @@ export default () => { /** * Get all sectors - * @returns {Array<[Sector]>} Collection of sectors + * @returns {Collection<[Sector]>} Collection of sectors * @example * const sectors = styleManager.getSectors(); * */ - getSectors() { - return sectors && sectors.models ? [...sectors.models] : []; + getSectors(opts = {}) { + return sectors && sectors.models + ? opts.array + ? [...sectors.models] + : sectors + : []; }, /** diff --git a/test/specs/style_manager/index.js b/test/specs/style_manager/index.js index 0e1e0141f..6e5926c33 100644 --- a/test/specs/style_manager/index.js +++ b/test/specs/style_manager/index.js @@ -27,7 +27,7 @@ describe('StyleManager', () => { obj.addSector('test', { name: 'Test name' }); - var sector = obj.getSectors()[0]; + var sector = obj.getSectors({ array: true })[0]; expect(obj.getSectors().length).toEqual(1); expect(sector.get('id')).toEqual('test'); expect(sector.get('name')).toEqual('Test name');