mocha.setup('bdd');

const expect = chai.expect;

class Todo {

  constructor(title="", text="", completed=false) {

    this.id = (+new Date() + Math.floor(Math.random() * 999999)).toString(36);

    this.title = title;

    this.text = text;

    this.completed = completed;

  }

}

function addTodo(todos, todo) {

  return todos.set(todo.id, todo);

}

function retrieveFinalPair(todos) {

  return todos.slice(todos.size-2, todos.size);

  // Alernatively, you can use this terser syntax

  //return todos.slice(-2);

}

function removeLastEntry(todos) {

  return todos.slice(0, -1);

}

function removeFirstEntry(todos) {

  return todos.slice(1, todos.size);

}

function removeFirstFive(todos) {

  return todos.skip(5);

}

function findMeMonkey(todos) {

  return todos.skipUntil(todo => todo.text === "monkey" );

}

function stopAtMonkey(todos) {

  return todos.skipWhile(todo => todo.text === "monkey" );

}

describe('Working with Subsets of an Immutable.js Map()', () => {

  it('should retrieve last two entries using slice()', () => {

    var todos = Immutable.Map();

    _.each(_.range(10), (index) => {

      todos = addTodo(todos, new Todo("Todo" + index, "I'm a todo!", false));

    });

    const lastTwoTodos = retrieveFinalPair(todos);

    expect(lastTwoTodos.size).to.equal(2);

    todos.takeLast(2).forEach(todo => {

      expect(lastTwoTodos.get(todo.id)).to.equal(todo);

    });

  });

  it('should remove last entry using negative slice()', () => {

    var todos = Immutable.Map();

    _.each(_.range(10), (index) => {

      todos = addTodo(todos, new Todo("Todo" + index, "I'm a todo!", false));

    });

    const todosWithoutLast = removeLastEntry(todos);

    todos.butLast().forEach(todo => {

      expect(todosWithoutLast.get(todo.id)).to.equal(todo);

    });

  });

  it('should remove first entry using slice()', () => {

    var todos = Immutable.Map();

    _.each(_.range(10), (index) => {

      todos = addTodo(todos, new Todo("Todo" + index, "I'm a todo!", false));

    });

    const todosWithoutFirst = removeFirstEntry(todos);

    todos.rest().forEach(todo => {

      expect(todosWithoutFirst.get(todo.id)).to.equal(todo);

    });

  });

  it('should return last 5 todos using skip()', () => {

    var todos = Immutable.Map();

    _.each(_.range(10), (index) => {

      todos = addTodo(todos, new Todo("Todo" + index, "I'm a todo!", false));

    });

    const lastFive = removeFirstFive(todos);

    todos.takeLast(5).forEach(todo => {

      expect(lastFive.get(todo.id)).to.equal(todo);

    });

  });  

  it('should return todos after reaching \"monkey\" using skipUntil()', () => {

    var texts = ["dog", "cat", "frog", "monkey", "octopus", "horse", "orangutan"];

    var todos = Immutable.Map();

    _.each(_.range(texts.length), (index) => {

      todos = addTodo(todos, new Todo("Todo" + index, texts[index], false));

    });

    const monkeyAndAfter = findMeMonkey(todos);

    todos.takeLast(4).forEach(todo => {

      expect(monkeyAndAfter.get(todo.id)).to.equal(todo);

    });

  });  

  it('should return todos up to reaching \"monkey\" using skipWhile()', () => {

    var texts = ["dog", "cat", "frog", "monkey", "octopus", "horse", "orangutan"];

    var todos = Immutable.Map();

    _.each(_.range(texts.length), (index) => {

      todos = addTodo(todos, new Todo("Todo" + index, texts[index], false));

    });

    const upToMonkey = stopAtMonkey(todos);

    todos.take(4).forEach(todo => {

      expect(upToMonkey.get(todo.id)).to.equal(todo);

    });

  });

});

mocha.run();