pool_test.go

package faker_test

import (
	"fmt"
	"testing"

	"github.com/pioz/faker"
	"github.com/stretchr/testify/assert"
)

func ExampleGetData() {
	faker.SetSeed(801)
	faker.SetPool("address", "city", faker.Pool{"New York", "Rome"})
	value, err := faker.GetData("address", "city")
	if err != nil {
		panic(err)
	}
	fmt.Println(value)
	// Output: New York
}

func TestGetAndSetPool(t *testing.T) {
	faker.SetSeed(802)
	faker.SetPool("ns1", "grp1", faker.Pool{"foo", "bar"})
	value, err := faker.GetData("ns1", "grp1")
	assert.Nil(t, err)
	t.Log(value)
	assert.Equal(t, "foo", value)

	faker.SetPool("ns2", "grp1", faker.Pool{"foo", "bar"})
	value, err = faker.GetData("ns2", "grp1")
	assert.Nil(t, err)
	t.Log(value)
	assert.Equal(t, "bar", value)

	faker.SetPool("ns2", "grp2", faker.Pool{"foo", "bar"})
	value, err = faker.GetData("ns2", "grp2")
	assert.Nil(t, err)
	t.Log(value)
	assert.Equal(t, "foo", value)

	faker.SetPoolGroup("ns3", faker.PoolGroup{"grp1": faker.Pool{"foo", "bar"}})
	value, err = faker.GetData("ns3", "grp1")
	assert.Nil(t, err)
	t.Log(value)
	assert.Equal(t, "bar", value)
}

func TestGetDataWithNamespaceError(t *testing.T) {
	faker.SetPool("ns4", "grp1", faker.Pool{"foo", "bar"})
	_, err := faker.GetData("not-exist", "grp1")
	assert.NotNil(t, err)
	assert.Equal(t, "the namespace 'not-exist' does not exist", err.Error())
}

func TestGetDataWithGroupError(t *testing.T) {
	faker.SetPool("ns5", "grp1", faker.Pool{"foo", "bar"})
	_, err := faker.GetData("ns5", "not-exist")
	assert.NotNil(t, err)
	assert.Equal(t, "the group 'not-exist' in namespace 'ns5' does not exist", err.Error())
}