Using WH datasets

src/lib/tree.test.ts

@@ -1,21 +1,40 @@
 import im from 'immutable'
-import { setPointsInTree } from './tree'
+import { 
+  setTalentPointsInTree, 
+  getTreePointCount 
+} from './tree'
 
-it('sets points on an empty tree', () => {
-  const tree = im.List()
-  expect(setPointsInTree(tree, 2, 5).toJS()).toEqual([0, 0, 5])
+
+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)
+  })
 })
 
-it('sets points in the end of the current range', () => {
-  const tree = im.List([0, 1])
-  expect(setPointsInTree(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(setPointsInTree(tree, 2, 5).toJS()).toEqual([0, 0, 5, 0, 0, 0, 5])
-})
+describe('getTreePointCount', () => {
+  it('returns proper count', () => {
+    const result = getTreePointCount(im.List([0, 0, 4, 5, 3, 0, 0]))
+    expect(result).toBe(12)
+  })
 
-it('does not mutate the tree for points already set', () => {
-  const tree = im.List([0, 3, 2, 0, 5])
-  expect(setPointsInTree(tree, 1, 3)).toStrictEqual(tree)
+  it('returns 0 for empty list', () => {
+    const result = getTreePointCount(im.List())
+    expect(result).toBe(0)
+  })
 })