status-im/status-go

package images

import (
    "image"
    "image/png"
    "os"
    "testing"

    "github.com/stretchr/testify/require"
)

func TestGenerateBannerImage_NoScaleUp(t *testing.T) {
    // Test image 256x256
    testImage := path + "status.png"
    identityImage, err := GenerateBannerImage(testImage, 50, 50, 150, 100)

    require.NoError(t, err)
    require.Exactly(t, identityImage.Name, BannerIdentityName)
    require.Positive(t, len(identityImage.Payload))
    // Ensure we don't scale it up. That will be done inefficiently in backend instead of frontend
    require.Exactly(t, identityImage.Width, 100)
    require.Exactly(t, identityImage.Height, 50)
    require.Exactly(t, identityImage.FileSize, len(identityImage.Payload))
    require.Exactly(t, identityImage.ResizeTarget, int(BannerDim))
}

func TestGenerateBannerImage_ShrinkOnly(t *testing.T) {
    // Generate test image bigger than BannerDim
    testImage := image.NewRGBA(image.Rect(0, 0, int(BannerDim)+10, int(BannerDim)+20))

    tmpTestFilePath := t.TempDir() + "/test.png"
    file, err := os.Create(tmpTestFilePath)
    require.NoError(t, err)
    defer file.Close()

    err = png.Encode(file, testImage)
    require.NoError(t, err)

    identityImage, error := GenerateBannerImage(tmpTestFilePath, 0, 0, int(BannerDim)+5, int(BannerDim)+10)

    require.NoError(t, error)
    require.Positive(t, len(identityImage.Payload))
    // Ensure we scale it down by the small side
    require.Exactly(t, identityImage.Width, int(BannerDim))
    require.Exactly(t, identityImage.Height, 805)
}