package claims

import (
	"bytes"
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestPermStorage_Set(t *testing.T) {
	ps := NewPermStorage()
	ps.Set("mjwt:test")
	if _, ok := ps.values["mjwt:test"]; !ok {
		assert.Fail(t, "perm not set")
	}
}

func TestPermStorage_Clear(t *testing.T) {
	ps := NewPermStorage()
	ps.values["mjwt:test"] = struct{}{}
	ps.Clear("mjwt:test")
	if _, ok := ps.values["mjwt:test"]; ok {
		assert.Fail(t, "perm not cleared")
	}
}

func TestPermStorage_Has(t *testing.T) {
	ps := NewPermStorage()
	assert.False(t, ps.Has("mjwt:test"))
	ps.values["mjwt:test"] = struct{}{}
	assert.True(t, ps.Has("mjwt:test"))
}

func TestPermStorage_OneOf(t *testing.T) {
	o := NewPermStorage()
	o.Set("mjwt:test")
	o.Set("mjwt:test2")

	ps := NewPermStorage()
	assert.False(t, ps.OneOf(o))
	ps.values["mjwt:test"] = struct{}{}
	assert.True(t, ps.OneOf(o))
	ps.values["mjwt:test2"] = struct{}{}
	assert.True(t, ps.OneOf(o))
	delete(ps.values, "mjwt:test")
	assert.True(t, ps.OneOf(o))
	delete(ps.values, "mjwt:test2")
	assert.False(t, ps.OneOf(o))
}

func TestPermStorage_MarshalJSON(t *testing.T) {
	ps := NewPermStorage()
	ps.Set("mjwt:test")
	ps.Set("mjwt:test2")
	b, err := ps.MarshalJSON()
	assert.NoError(t, err)
	assert.Equal(t, 0, bytes.Compare([]byte(`["mjwt:test","mjwt:test2"]`), b))
}

func TestPermStorage_MarshalYAML(t *testing.T) {
	ps := NewPermStorage()
	ps.Set("mjwt:test")
	ps.Set("mjwt:test2")
	b, err := ps.MarshalYAML()
	assert.NoError(t, err)
	assert.Equal(t, 0, bytes.Compare([]byte("- mjwt:test\n- mjwt:test2\n"), b.([]byte)))
}