'use strict';

/*
 * This file tests the FSM test framework.
 */

load('jstests/concurrency/fsm_libs/fsm.js');

(function() {
    var getWeightedRandomChoice = fsm._getWeightedRandomChoice;

    var doc = {a: 0.25, b: 0.5, c: 0.25};

    // NOTE: getWeightedRandomChoice calls assert internally, so it will print stack traces
    // when assert.throws executes
    assert.throws(function() {
        getWeightedRandomChoice(doc, -1);
    }, [], 'should reject negative values');
    assert.throws(function() {
        getWeightedRandomChoice(doc, 1);
    }, [], 'should reject values == 1');
    assert.throws(function() {
        getWeightedRandomChoice(doc, 2);
    }, [], 'should reject values > 1');

    assert.throws(function() {
        getWeightedRandomChoice({}, 0.0);
    }, [], 'cannot choose from zero states');
    assert.throws(function() {
        getWeightedRandomChoice({}, 0.5);
    }, [], 'cannot choose from zero states');
    assert.throws(function() {
        getWeightedRandomChoice({}, 0.99);
    }, [], 'cannot choose from zero states');

    assert.eq('a', getWeightedRandomChoice(doc, 0.00), '0');
    assert.eq('a', getWeightedRandomChoice(doc, 0.24), '1');
    assert.eq('b', getWeightedRandomChoice(doc, 0.25), '2');
    assert.eq('b', getWeightedRandomChoice(doc, 0.50), '3');
    assert.eq('b', getWeightedRandomChoice(doc, 0.74), '4');
    assert.eq('c', getWeightedRandomChoice(doc, 0.75), '5');
    assert.eq('c', getWeightedRandomChoice(doc, 0.99), '6');
})();