From e7df8cc7399171c9df7aefed54528116767ceca9 Mon Sep 17 00:00:00 2001 From: Artur Arseniev Date: Fri, 29 Oct 2021 13:13:53 +0200 Subject: [PATCH] Update getSectors --- src/style_manager/index.js | 10 +++++++--- test/specs/style_manager/index.js | 2 +- 2 files changed, 8 insertions(+), 4 deletions(-) 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');