@antv/x6/src/util/object/object.test.ts

JavaScript diagramming library that uses SVG and HTML for rendering.

import {
  getValue,
  getNumber,
  getBoolean,
  getByPath,
  unsetByPath,
  setByPath,
  flatten,
} from './object'

describe('Object', () => {
  const obj = {
    name: 'x6',
    age: 1,
    gender: null,
    tall: true,
  }

  describe('#getValue', () => {
    it('should return a value of a object', () => {
      expect(getValue<string>(obj, 'name')).toBe('x6')
      expect(getValue<number>(obj, 'age')).toBe(1)
      expect(getValue<any>(obj, 'gender')).toBe(null)
      expect(getValue<any>(obj, 'count')).toBe(undefined)
    })

    it('should return a defaultValue of a object', () => {
      expect(getValue(obj, 'gender', 'male')).toBe('male')
      expect(getValue(obj, 'count', 100)).toBe(100)
    })
  })

  describe('#getNumber', () => {
    it('should return a number value of a object', () => {
      expect(getNumber(obj, 'age', 2)).toBe(1)
    })

    it('should return a defaultValue of a object', () => {
      expect(getNumber(obj, 'name', 10)).toBe(10)
      expect(getNumber(obj, 'count', 20)).toBe(20)
    })
  })

  describe('#getBoolean', () => {
    it('should return a boolean value of a object', () => {
      expect(getBoolean(obj, 'tall', false)).toBe(true)
      expect(getBoolean(obj, 'name', false)).toBe(true)
      expect(getBoolean(obj, 'age', false)).toBe(true)
    })

    it('should return a defaultValue of a object', () => {
      expect(getBoolean(obj, 'gender', false)).toBe(false)
      expect(getBoolean(obj, 'count', true)).toBe(true)
    })
  })

  describe('#getByPath#setByPath', () => {
    const project = {
      name: 'x6',
      version: ['0.1', '0.2', '0.3'],
      attr: {
        node: {
          fontSize: 14,
        },
        edge: {
          color: 'red',
        },
      },
    }
    it('should set or get object value by path', () => {
      expect(getByPath(project, 'version/1')).toBe('0.2')
      expect(getByPath(project, 'attr/node/fontSize')).toBe(14)
      expect(getByPath(project, 'attr/node/color')).toBe(undefined)

      setByPath(project, 'version/1', '0.8')
      setByPath(project, 'attr/node/fontSize', 16)
      setByPath(project, 'attr/node/color', 'green')

      expect(getByPath(project, 'version/1')).toBe('0.8')
      expect(getByPath(project, 'attr/node/fontSize')).toBe(16)
      expect(getByPath(project, 'attr/node/color')).toBe('green')

      unsetByPath(project, 'version/1')
      unsetByPath(project, 'attr/node/fontSize')

      expect(getByPath(project, 'version/1')).toBe(undefined)
      expect(getByPath(project, 'attr/node/fontSize')).toBe(undefined)
    })
  })

  describe('#flatten', () => {
    const project = {
      name: 'x6',
      version: ['0.1', '0.2', '0.3'],
      attr: {
        node: {
          fontSize: 14,
        },
        edge: {
          color: 'red',
        },
      },
    }
    it('should return flatten object', () => {
      expect(flatten(project)).toEqual({
        name: 'x6',
        'version/0': '0.1',
        'version/1': '0.2',
        'version/2': '0.3',
        'attr/node/fontSize': 14,
        'attr/edge/color': 'red',
      })
    })
  })
})