messageformat_test.go

// @TODO(gotnospirit) add test on SetCulture, Format, getNamedKey, getFormatter

package messageformat

import (
	"fmt"
	"testing"
)

// doTest(t, Test{
// "You have {NUM_TASKS, plural, zero {no task} one {one task} two {two tasks} few{few tasks} many {many tasks} other {# tasks} =42 {the answer to the life, the universe and everything tasks}} remaining.",
// []Expectation{
// {map[string]interface{}{"NUM_TASKS": -1}, "You have -1 tasks remaining."},
// {map[string]interface{}{"NUM_TASKS": 0}, "You have no task remaining."},
// {map[string]interface{}{"NUM_TASKS": 1}, "You have one task remaining."},
// {map[string]interface{}{"NUM_TASKS": 2}, "You have two tasks remaining."},
// {map[string]interface{}{"NUM_TASKS": 3}, "You have few tasks remaining."},
// {map[string]interface{}{"NUM_TASKS": 6}, "You have many tasks remaining."},
// {map[string]interface{}{"NUM_TASKS": 15}, "You have 15 tasks remaining."},
// {map[string]interface{}{"NUM_TASKS": 42}, "You have the answer to the life, the universe and everything tasks remaining."},
// },
// })

func doParse(input string) (*MessageFormat, error) {
	o, err := New()
	if err != nil {
		return nil, err
	}
	mf, err := o.Parse(input)
	if err != nil {
		return nil, err
	}
	return mf, nil
}

func TestSetPluralFunction(t *testing.T) {
	mf, err := doParse(`{N,plural,one{1}other{2}}`)
	if err != nil {
		t.Errorf("Unexpected parse failure: `%s`", err.Error())
	} else {
		// checks we can't unset the default plural function
		err := mf.SetPluralFunction(nil)
		doTestError(t, "PluralFunctionRequired", err)

		data := map[string]interface{}{"N": 1}
		result, err := mf.FormatMap(data)
		if err != nil {
			t.Errorf("Unexpected error : `%s`", err.Error())
		} else if result != "1" {
			t.Errorf("Unexpected format result : `%s`", result)
		} else {
			// checks we can set a new plural function and get a different result
			err := mf.SetPluralFunction(func(interface{}, bool) string {
				return "other"
			})

			if err != nil {
				t.Errorf("Unexpected error <%s>", err)
			} else {
				result, err := mf.FormatMap(data)

				if err != nil {
					t.Errorf("Unexpected error : `%s`", err.Error())
				} else if result != "2" {
					t.Errorf("Unexpected format result : `%s`", result)
				} else if testing.Verbose() {
					fmt.Printf("- Got expected value <%s>\n", result)
				}
			}
		}
	}
}