diff --git a/tokens/middleware_context_test.go b/tokens/middleware_context_test.go new file mode 100644 index 0000000..f160280 --- /dev/null +++ b/tokens/middleware_context_test.go @@ -0,0 +1,39 @@ +package tokens_test + +import ( + "context" + "gitea.deepak.science/deepak/gogmagog/tokens" + "github.com/stretchr/testify/assert" + "testing" +) + +func TestGoodContext(t *testing.T) { + assert := assert.New(t) + + idToUse := 3 + username := "username" + + ctx := tokens.GetContextForUserValues(idToUse, username) + + receivedID, err := tokens.GetUserID(ctx) + assert.Nil(err) + assert.EqualValues(idToUse, receivedID) + + receivedUsername, err := tokens.GetUsername(ctx) + assert.Nil(err) + assert.Equal(username, receivedUsername) + +} + +func TestBadContext(t *testing.T) { + assert := assert.New(t) + + ctx := context.Background() + + _, err := tokens.GetUserID(ctx) + assert.NotNil(err) + + _, err = tokens.GetUsername(ctx) + assert.NotNil(err) + +}