size_test.go

package ptt

import (
	"testing"
)

func TestSizeGroupDetection(t *testing.T) {
	cases := []struct {
		releaseName string
		expectedSize *string
	}{
		{
			releaseName: "www.1TamilBlasters.lat - Thuritham (2023) [Tamil - 2K QHD AVC UNTOUCHED - x264 - AAC - 3.4GB - ESub].mkv",
			expectedSize: strPtr("3.4GB"),
		},
		{
			releaseName: "www.1TamilMV.world - Raja Vikramarka (2024) Tamil HQ HDRip - 400MB - x264 - AAC - ESub.mkv",
			expectedSize: strPtr("400MB"),
		},
		{
			releaseName: "www.1TamilMV.cz - Maharaja (2024) TRUE WEB-DL - 1080p HQ - AVC - (DD+5.1 - 640Kbps) [Tam + Tel + Hin + Mal + Kan] - 8.4GB - ESub.mkv",
			expectedSize: strPtr("8.4GB"),
		},
		{
			releaseName: "The.Walking.Dead.S06E07.SUBFRENCH.HDTV.x264-AMB3R.mkv",
			expectedSize: nil,
		},
	}

	p := newTestParser()
	for _, tt := range cases {
		result := p.Parse(tt.releaseName)
		if tt.expectedSize != nil {
			if result.Size != *tt.expectedSize {
				t.Fatalf("%s: Size = %q, want %q", tt.releaseName, result.Size, *tt.expectedSize)
			}
		} else {
			if result.Size != "" {
				t.Fatalf("%s: expected Size to be empty, got %q", tt.releaseName, result.Size)
			}
		}
	}
}