Melvin Valster
40 lines
1.1 KiB
TypeScript
40 lines
1.1 KiB
TypeScript
import im from 'immutable'
|
|
import {
|
|
setTalentPointsInTree,
|
|
getTreePointCount
|
|
} from './tree'
|
|
|
|
|
|
describe('setTalentPointsInTree', () => {
|
|
it('sets points on an empty tree', () => {
|
|
const tree = im.List()
|
|
expect(setTalentPointsInTree(tree, 2, 5).toJS()).toEqual([0, 0, 5])
|
|
})
|
|
|
|
it('sets points in the end of the current range', () => {
|
|
const tree = im.List([0, 1])
|
|
expect(setTalentPointsInTree(tree, 2, 5).toJS()).toEqual([0, 1, 5])
|
|
})
|
|
|
|
it('sets points in the middle of the current range', () => {
|
|
const tree = im.List([0, 0, 0, 0, 0, 0, 5])
|
|
expect(setTalentPointsInTree(tree, 2, 5).toJS()).toEqual([0, 0, 5, 0, 0, 0, 5])
|
|
})
|
|
|
|
it('does not mutate the tree for points already set', () => {
|
|
const tree = im.List([0, 3, 2, 0, 5])
|
|
expect(setTalentPointsInTree(tree, 1, 3)).toStrictEqual(tree)
|
|
})
|
|
})
|
|
|
|
describe('getTreePointCount', () => {
|
|
it('returns proper count', () => {
|
|
const result = getTreePointCount(im.List([0, 0, 4, 5, 3, 0, 0]))
|
|
expect(result).toBe(12)
|
|
})
|
|
|
|
it('returns 0 for empty list', () => {
|
|
const result = getTreePointCount(im.List())
|
|
expect(result).toBe(0)
|
|
})
|
|
}) |