// Copyright (c) 2021 - 2023, Ludvig Lundgren and the autobrr contributors. // SPDX-License-Identifier: GPL-2.0-or-later package domain import ( "fmt" "net/http" "net/http/httptest" "os" "strings" "testing" "time" "github.com/rs/zerolog" "github.com/stretchr/testify/assert" ) func TestRelease_Parse(t *testing.T) { tests := []struct { name string fields Release want Release }{ { name: "parse_1", fields: Release{ TorrentName: "Servant S01 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-FLUX", }, want: Release{ TorrentName: "Servant S01 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-FLUX", Title: "Servant", Season: 1, Episode: 0, Resolution: "2160p", Source: "WEB-DL", Codec: []string{"HEVC"}, Audio: []string{"DDP", "Atmos"}, AudioChannels: "5.1", HDR: []string{"DV"}, Group: "FLUX", //Website: "ATVP", }, }, { name: "parse_2", fields: Release{ TorrentName: "Servant.S01.2160p.ATVP.WEB-DL.DDP.5.1.Atmos.DV.HEVC-FLUX", }, want: Release{ TorrentName: "Servant.S01.2160p.ATVP.WEB-DL.DDP.5.1.Atmos.DV.HEVC-FLUX", Title: "Servant", Season: 1, Episode: 0, Resolution: "2160p", Source: "WEB-DL", Codec: []string{"HEVC"}, Audio: []string{"DDP", "Atmos"}, AudioChannels: "5.1", HDR: []string{"DV"}, Group: "FLUX", }, }, { name: "parse_3", fields: Release{ TorrentName: "Servant.S01.2160p.ATVP.WEB-DL.DDP.5.1.Atmos.DV.HEVC-FLUX", ReleaseTags: "MKV / 2160p / WEB-DL", }, want: Release{ TorrentName: "Servant.S01.2160p.ATVP.WEB-DL.DDP.5.1.Atmos.DV.HEVC-FLUX", ReleaseTags: "MKV / 2160p / WEB-DL", Title: "Servant", Season: 1, Episode: 0, Resolution: "2160p", Source: "WEB-DL", Container: "mkv", Codec: []string{"HEVC"}, Audio: []string{"DDP", "Atmos"}, AudioChannels: "5.1", HDR: []string{"DV"}, Group: "FLUX", }, }, { name: "parse_4", fields: Release{ TorrentName: "Servant.S01.2160p.ATVP.WEB-DL.DDP.5.1.Atmos.DV.HEVC-FLUX", ReleaseTags: "MKV | 2160p | WEB-DL", }, want: Release{ TorrentName: "Servant.S01.2160p.ATVP.WEB-DL.DDP.5.1.Atmos.DV.HEVC-FLUX", ReleaseTags: "MKV | 2160p | WEB-DL", Title: "Servant", Season: 1, Episode: 0, Resolution: "2160p", Source: "WEB-DL", Container: "mkv", Codec: []string{"HEVC"}, Audio: []string{"DDP", "Atmos"}, AudioChannels: "5.1", HDR: []string{"DV"}, Group: "FLUX", }, }, { name: "parse_5", fields: Release{ TorrentName: "Servant.S01.2160p.ATVP.WEB-DL.DDP.5.1.Atmos.DV.HEVC-FLUX", ReleaseTags: "MP4 | 2160p | WEB-DL", }, want: Release{ TorrentName: "Servant.S01.2160p.ATVP.WEB-DL.DDP.5.1.Atmos.DV.HEVC-FLUX", ReleaseTags: "MP4 | 2160p | WEB-DL", Title: "Servant", Season: 1, Episode: 0, Resolution: "2160p", Source: "WEB-DL", Container: "mp4", Codec: []string{"HEVC"}, Audio: []string{"DDP", "Atmos"}, AudioChannels: "5.1", HDR: []string{"DV"}, Group: "FLUX", }, }, { name: "parse_6", fields: Release{ TorrentName: "Servant.S01.2160p.ATVP.WEB-DL.DDP.5.1.Atmos.DV.HEVC-FLUX", ReleaseTags: "MP4 | 2160p | WEB-DL | Freeleech!", }, want: Release{ TorrentName: "Servant.S01.2160p.ATVP.WEB-DL.DDP.5.1.Atmos.DV.HEVC-FLUX", ReleaseTags: "MP4 | 2160p | WEB-DL | Freeleech!", Title: "Servant", Season: 1, Episode: 0, Resolution: "2160p", Source: "WEB-DL", Container: "mp4", Codec: []string{"HEVC"}, Audio: []string{"DDP", "Atmos"}, AudioChannels: "5.1", HDR: []string{"DV"}, Group: "FLUX", Freeleech: true, Bonus: []string{"Freeleech"}, }, }, { name: "parse_music_1", fields: Release{ TorrentName: "Artist - Albumname", ReleaseTags: "FLAC / Lossless / Log / 100% / Cue / CD", }, want: Release{ TorrentName: "Artist - Albumname", ReleaseTags: "FLAC / Lossless / Log / 100% / Cue / CD", Title: "Artist", Group: "Albumname", Audio: []string{"Cue", "FLAC", "Lossless", "Log100", "Log"}, Source: "CD", }, }, { name: "parse_music_2", fields: Release{ TorrentName: "Various Artists - Music '21", Tags: []string{"house, techno, tech.house, electro.house, future.house, bass.house, melodic.house"}, ReleaseTags: "MP3 / 320 / Cassette", }, want: Release{ TorrentName: "Various Artists - Music '21", Tags: []string{"house, techno, tech.house, electro.house, future.house, bass.house, melodic.house"}, ReleaseTags: "MP3 / 320 / Cassette", Title: "Various Artists - Music '21", Source: "Cassette", Audio: []string{"320", "MP3"}, }, }, { name: "parse_music_3", fields: Release{ TorrentName: "The artist (ザ・フリーダムユニティ) - Long album name", ReleaseTags: "MP3 / V0 (VBR) / CD", }, want: Release{ TorrentName: "The artist (ザ・フリーダムユニティ) - Long album name", ReleaseTags: "MP3 / V0 (VBR) / CD", Title: "The artist", Group: "name", Source: "CD", Audio: []string{"MP3", "VBR"}, }, }, { name: "parse_music_4", fields: Release{ TorrentName: "Artist - Albumname", ReleaseTags: "FLAC / Lossless / Log / 100% / Cue / CD", }, want: Release{ TorrentName: "Artist - Albumname", ReleaseTags: "FLAC / Lossless / Log / 100% / Cue / CD", Title: "Artist", Group: "Albumname", Audio: []string{"Cue", "FLAC", "Lossless", "Log100", "Log"}, Source: "CD", }, }, { name: "parse_music_5", fields: Release{ TorrentName: "Artist - Albumname", ReleaseTags: "FLAC / 24bit Lossless / Log / 100% / Cue / CD", }, want: Release{ TorrentName: "Artist - Albumname", ReleaseTags: "FLAC / 24bit Lossless / Log / 100% / Cue / CD", Title: "Artist", Group: "Albumname", Audio: []string{"24BIT Lossless", "Cue", "FLAC", "Log100", "Log"}, Source: "CD", }, }, { name: "parse_movies_case_1", fields: Release{ TorrentName: "I Am Movie 2007 Theatrical UHD BluRay 2160p DTS-HD MA 5.1 DV HEVC HYBRID REMUX-GROUP1", }, want: Release{ TorrentName: "I Am Movie 2007 Theatrical UHD BluRay 2160p DTS-HD MA 5.1 DV HEVC HYBRID REMUX-GROUP1", Title: "I Am Movie", Resolution: "2160p", Source: "UHD.BluRay", Codec: []string{"HEVC"}, HDR: []string{"DV"}, Audio: []string{"DTS-HD.MA"}, AudioChannels: "5.1", Year: 2007, Group: "GROUP1", Other: []string{"HYBRiD", "REMUX"}, }, }, { name: "parse_title_1", fields: Release{ TorrentName: "The Peripheral (2022) S01 (2160p AMZN WEB-DL H265 HDR10+ DDP 5.1 English - GROUP1)", }, want: Release{ TorrentName: "The Peripheral (2022) S01 (2160p AMZN WEB-DL H265 HDR10+ DDP 5.1 English - GROUP1)", Title: "The Peripheral", Resolution: "2160p", Source: "WEB-DL", Codec: []string{"H.265"}, HDR: []string{"HDR10+"}, Audio: []string{"DDP"}, AudioChannels: "5.1", Year: 2022, Group: "GROUP1", Season: 1, Language: []string{"ENGLiSH"}, }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { r := tt.fields r.ParseString(tt.fields.TorrentName) assert.Equal(t, tt.want, r) }) } } func TestRelease_MapVars(t *testing.T) { type args struct { varMap map[string]string definition IndexerDefinition } tests := []struct { name string fields *Release want *Release args args }{ { name: "1", fields: &Release{}, want: &Release{TorrentName: "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2"}, args: args{varMap: map[string]string{ "torrentName": "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2", }}, }, { name: "2", fields: &Release{}, want: &Release{ TorrentName: "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2", Category: "tv", Freeleech: true, FreeleechPercent: 100, Bonus: []string{"Freeleech"}, Uploader: "Anon", Size: uint64(10000000000), }, args: args{ varMap: map[string]string{ "torrentName": "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2", "category": "tv", "freeleech": "freeleech", "uploader": "Anon", "torrentSize": "10GB", }, }, }, { name: "3", fields: &Release{}, want: &Release{ TorrentName: "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2", Category: "tv", Freeleech: true, FreeleechPercent: 100, Bonus: []string{"Freeleech", "Freeleech100"}, Uploader: "Anon", Size: uint64(10000000000), }, args: args{varMap: map[string]string{ "torrentName": "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2", "category": "tv", "freeleechPercent": "100%", "uploader": "Anon", "torrentSize": "10GB", }}, }, { name: "4", fields: &Release{}, want: &Release{ TorrentName: "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2", Category: "tv", Freeleech: true, FreeleechPercent: 50, Bonus: []string{"Freeleech", "Freeleech50"}, Uploader: "Anon", Size: uint64(10000000000), Tags: []string{"foreign", "tv"}, }, args: args{varMap: map[string]string{ "torrentName": "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2", "category": "tv", "freeleechPercent": "50%", "uploader": "Anon", "torrentSize": "10GB", "tags": "foreign,tv", }}, }, { name: "5", fields: &Release{}, want: &Release{ TorrentName: "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2", Category: "tv", Freeleech: true, FreeleechPercent: 100, Bonus: []string{"Freeleech", "Freeleech100"}, Uploader: "Anon", Size: uint64(10000000000), Tags: []string{"foreign", "tv"}, }, args: args{varMap: map[string]string{ "torrentName": "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2", "category": "tv", "freeleechPercent": "100%", "uploader": "Anon", "torrentSize": "10GB", "tags": "foreign,tv", }}, }, { name: "6", fields: &Release{}, want: &Release{ TorrentName: "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2", Category: "tv", Year: 2020, Freeleech: true, FreeleechPercent: 100, Bonus: []string{"Freeleech", "Freeleech100"}, Uploader: "Anon", Size: uint64(10000000000), Tags: []string{"foreign", "tv"}, }, args: args{varMap: map[string]string{ "torrentName": "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2", "category": "tv", "year": "2020", "freeleechPercent": "100%", "uploader": "Anon", "torrentSize": "10GB", "tags": "foreign, tv", }}, }, { name: "7", fields: &Release{}, want: &Release{ TorrentName: "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2", Category: "tv", Year: 2020, Freeleech: true, FreeleechPercent: 25, Bonus: []string{"Freeleech", "Freeleech25"}, Uploader: "Anon", Size: uint64(10000000000), Tags: []string{"hip.hop", "rhythm.and.blues", "2000s"}, }, args: args{varMap: map[string]string{ "torrentName": "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2", "category": "tv", "year": "2020", "freeleechPercent": "25%", "uploader": "Anon", "torrentSize": "10GB", "tags": "hip.hop,rhythm.and.blues, 2000s", }}, }, { name: "8", fields: &Release{}, want: &Release{ TorrentName: "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2", Category: "tv", Year: 2020, Freeleech: true, FreeleechPercent: 100, Bonus: []string{"Freeleech", "Freeleech100"}, Uploader: "Anon", Size: uint64(10000000000), Tags: []string{"hip.hop", "rhythm.and.blues", "2000s"}, }, args: args{ varMap: map[string]string{ "torrentName": "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2", "category": "tv", "year": "2020", "freeleechPercent": "100%", "uploader": "Anon", "torrentSize": "10000", "tags": "hip.hop,rhythm.and.blues, 2000s", }, definition: IndexerDefinition{IRC: &IndexerIRC{Parse: &IndexerIRCParse{ForceSizeUnit: "MB"}}}, }, }, { name: "9", fields: &Release{}, want: &Release{ TorrentName: "Greatest Anime Ever", Year: 2022, Group: "GROUP1", Tags: []string{"comedy", "fantasy", "school.life", "shounen", "slice.of.life"}, Uploader: "Tester", }, args: args{varMap: map[string]string{ "torrentName": "Greatest Anime Ever", "year": "2022", "releaseGroup": "GROUP1", "tags": "comedy, fantasy, school.life, shounen, slice.of.life", "uploader": "Tester", }}, }, { name: "10", fields: &Release{}, want: &Release{ TorrentName: "Greatest Anime Ever", Year: 2022, Group: "GROUP1", Tags: []string{"comedy", "fantasy", "school.life", "shounen", "slice.of.life"}, Uploader: "Tester", Freeleech: true, FreeleechPercent: 100, Bonus: []string{"Freeleech"}, }, args: args{varMap: map[string]string{ "torrentName": "Greatest Anime Ever", "year": "2022", "releaseGroup": "GROUP1", "tags": "comedy, fantasy, school.life, shounen, slice.of.life", "uploader": "Tester", "freeleech": "VIP", }}, }, { name: "11", fields: &Release{}, want: &Release{ TorrentName: "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2", Category: "tv", Freeleech: true, FreeleechPercent: 100, Bonus: []string{"Freeleech"}, Uploader: "Anon", Size: uint64(10000000000), Tags: []string{"comedy", "science fiction", "fantasy", "school.life", "shounen", "slice.of.life"}, }, args: args{ varMap: map[string]string{ "torrentName": "Good show S02 2160p ATVP WEB-DL DDP 5.1 Atmos DV HEVC-GROUP2", "category": "tv", "tags": "comedy, science fiction, fantasy, school.life, shounen, slice.of.life", "freeleech": "freeleech", "uploader": "Anon", "torrentSize": "10GB", }, }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { r := tt.fields _ = r.MapVars(&tt.args.definition, tt.args.varMap) assert.Equal(t, tt.want, r) }) } } func TestSplitAny(t *testing.T) { type args struct { s string seps string } tests := []struct { name string args args want []string }{ { name: "test_1", args: args{ s: "Tag1 / Tag2 / Tag3", seps: "/ ", }, want: []string{"Tag1", "Tag2", "Tag3"}, }, { name: "test_2", args: args{ s: "Tag1 | Tag2 | Tag3", seps: "| ", }, want: []string{"Tag1", "Tag2", "Tag3"}, }, { name: "test_3", args: args{ s: "Tag1 | Tag2 / Tag3", seps: "| /", }, want: []string{"Tag1", "Tag2", "Tag3"}, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { assert.Equalf(t, tt.want, SplitAny(tt.args.s, tt.args.seps), "SplitAny(%v, %v)", tt.args.s, tt.args.seps) }) } } func TestRelease_ParseString(t *testing.T) { type fields struct { Release } type args struct { title string } tests := []struct { name string fields fields args args wantErr bool }{ {name: "parse_1", fields: fields{}, args: args{title: "Phenomena 1985 International Cut UHD BluRay 2160p DTS-HD MA 5.1 DV HEVC HYBRID REMUX-FraMeSToR"}, wantErr: false}, {name: "parse_2", fields: fields{}, args: args{title: "Justice League: Dark 2017 UHD BluRay 2160p DTS-HD MA 5.1 DV HEVC HYBRID REMUX-FraMeSToR"}, wantErr: false}, {name: "parse_3", fields: fields{}, args: args{title: "Outer.Range.S01E02.The.Land.1080p.AMZN.WEB-DL.DDP5.1.H.264-TOMMY"}, wantErr: false}, {name: "parse_4", fields: fields{}, args: args{title: "WeCrashed S01E07 The Power of We 2160p ATVP WEB-DL DDP 5.1 Atmos HDR H.265-NOSiViD"}, wantErr: false}, {name: "parse_5", fields: fields{}, args: args{title: "WeCrashed.S01E07.The.Power.of.We.DV.2160p.ATVP.WEB-DL.DDPA5.1.H.265-NOSiViD"}, wantErr: false}, {name: "parse_6", fields: fields{}, args: args{title: "WeCrashed.S01E07.The.Power.of.We.DV.2160p.ATVP.WEB-DL.DDPA5.1.H265-NOSiViD"}, wantErr: false}, {name: "parse_7", fields: fields{}, args: args{title: "WeCrashed.S01E07.The.Power.of.We.DV.2160p.ATVP.WEB-DL.DDPA5.1.x265-NOSiViD"}, wantErr: false}, {name: "parse_8", fields: fields{}, args: args{title: "WeCrashed.S01E07.The.Power.of.We.HDR.2160p.ATVP.WEB-DL.DDPA5.1.x265-NOSiViD"}, wantErr: false}, {name: "parse_9", fields: fields{}, args: args{title: "WeCrashed.S01.HDR.2160p.ATVP.WEB-DL.DDPA5.1.x265-NOSiViD"}, wantErr: false}, {name: "parse_10", fields: fields{}, args: args{title: "WeCrashed.S01.DV.HDR+.2160p.ATVP.WEB-DL.DDPA5.1.x265-NOSiViD"}, wantErr: false}, {name: "parse_11", fields: fields{}, args: args{title: "WeCrashed.S01.DoVi.HDR10+.2160p.ATVP.WEB-DL.DDPA5.1.x265-NOSiViD"}, wantErr: false}, {name: "parse_12", fields: fields{}, args: args{title: "WeCrashed.S01.Dolby.Vision.2160p.ATVP.WEB-DL.DDPA5.1.x265-NOSiViD"}, wantErr: false}, {name: "parse_13", fields: fields{}, args: args{title: "WeCrashed.S01.Dolby.Vision.1080p.ATVP.WEB-DL.DDPA5.1.x264-NOSiViD"}, wantErr: false}, {name: "parse_14", fields: fields{}, args: args{title: "Without Remorse 2021 1080p Blu-ray AVC DTS-HD MA 5.1-MTeam"}, wantErr: false}, {name: "parse_15", fields: fields{}, args: args{title: "Annette 2021 2160p GER UHD Blu-ray SDR HEVC DTS-HD MA 5.1-UNTOUCHED"}, wantErr: false}, {name: "parse_16", fields: fields{}, args: args{title: "Sing 2 2021 MULTi COMPLETE UHD Blu-ray TrueHD Atmos 7.1-MMCLX"}, wantErr: false}, {name: "parse_17", fields: fields{}, args: args{title: "NBC.Nightly.News.2022.04.12.1080p.NBC.WEB-DL.AAC2.0.H.264-TEPES"}, wantErr: false}, {name: "parse_18", fields: fields{}, args: args{title: "[SubsPlease] Heroine Tarumono! Kiraware Heroine to Naisho no Oshigoto - 04 (1080p) [17083ED9]"}, wantErr: false}, {name: "parse_19", fields: fields{}, args: args{title: "The World is Not Enough 1999 2160p WEB-DL HEVC DTS-HD MA 5.1 H.265-DEFLATE"}, wantErr: false}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { r := &Release{ ID: tt.fields.ID, FilterStatus: tt.fields.FilterStatus, Rejections: tt.fields.Rejections, Indexer: tt.fields.Indexer, FilterName: tt.fields.FilterName, Protocol: tt.fields.Protocol, Implementation: tt.fields.Implementation, Timestamp: tt.fields.Timestamp, GroupID: tt.fields.GroupID, TorrentID: tt.fields.TorrentID, TorrentURL: tt.fields.TorrentURL, TorrentTmpFile: tt.fields.TorrentTmpFile, TorrentHash: tt.fields.TorrentHash, TorrentName: tt.fields.TorrentName, Size: tt.fields.Size, Title: tt.fields.Title, Category: tt.fields.Category, Season: tt.fields.Season, Episode: tt.fields.Episode, Year: tt.fields.Year, Resolution: tt.fields.Resolution, Source: tt.fields.Source, Codec: tt.fields.Codec, Container: tt.fields.Container, HDR: tt.fields.HDR, Audio: tt.fields.Audio, Group: tt.fields.Group, Region: tt.fields.Region, Language: tt.fields.Language, Proper: tt.fields.Proper, Repack: tt.fields.Repack, Website: tt.fields.Website, Artists: tt.fields.Artists, Type: tt.fields.Type, LogScore: tt.fields.LogScore, Origin: tt.fields.Origin, Tags: tt.fields.Tags, ReleaseTags: tt.fields.ReleaseTags, Freeleech: tt.fields.Freeleech, FreeleechPercent: tt.fields.FreeleechPercent, Uploader: tt.fields.Uploader, PreTime: tt.fields.PreTime, RawCookie: tt.fields.RawCookie, AdditionalSizeCheckRequired: tt.fields.AdditionalSizeCheckRequired, FilterID: tt.fields.FilterID, Filter: tt.fields.Filter, ActionStatus: tt.fields.ActionStatus, } r.ParseString(tt.args.title) }) } } var trackerLessTestTorrent = `d7:comment19:This is just a test10:created by12:Johnny Bravo13:creation datei1430648794e8:encoding5:UTF-84:infod6:lengthi1128e4:name12:testfile.bin12:piece lengthi32768e6:pieces20:Õˆë =‘UŒäiÎ^æ °Eâ?ÇÒe5:nodesl35:udp://tracker.openbittorrent.com:8035:udp://tracker.openbittorrent.com:80ee` func TestRelease_DownloadTorrentFile(t *testing.T) { // disable logger zerolog.SetGlobalLevel(zerolog.Disabled) mux := http.NewServeMux() ts := httptest.NewServer(mux) defer ts.Close() mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { if strings.Contains(r.RequestURI, "401") { w.WriteHeader(http.StatusUnauthorized) w.Write([]byte("unauthorized")) return } if strings.Contains(r.RequestURI, "403") { w.WriteHeader(http.StatusUnauthorized) w.Write([]byte("forbidden")) return } if strings.Contains(r.RequestURI, "404") { w.WriteHeader(http.StatusUnauthorized) w.Write([]byte("not found")) return } if strings.Contains(r.RequestURI, "405") { w.WriteHeader(http.StatusUnauthorized) w.Write([]byte("method not allowed")) return } if strings.Contains(r.RequestURI, "file.torrent") { w.WriteHeader(http.StatusOK) w.Header().Set("Content-Type", "application/x-bittorrent") payload, _ := os.ReadFile("testdata/archlinux-2011.08.19-netinstall-i686.iso.torrent") w.Write(payload) return } w.WriteHeader(http.StatusInternalServerError) w.Write([]byte("internal error")) }) type fields struct { ID int64 FilterStatus ReleaseFilterStatus Rejections []string Indexer string FilterName string Protocol ReleaseProtocol Implementation ReleaseImplementation Timestamp time.Time GroupID string TorrentID string TorrentURL string TorrentTmpFile string TorrentDataRawBytes []byte TorrentHash string TorrentName string Size uint64 Title string Category string Categories []string Season int Episode int Year int Resolution string Source string Codec []string Container string HDR []string Audio []string AudioChannels string Group string Region string Language []string Proper bool Repack bool Website string Artists string Type string LogScore int Origin string Tags []string ReleaseTags string Freeleech bool FreeleechPercent int Bonus []string Uploader string PreTime string Other []string RawCookie string AdditionalSizeCheckRequired bool FilterID int Filter *Filter ActionStatus []ReleaseActionStatus } tests := []struct { name string fields fields wantErr assert.ErrorAssertionFunc }{ { name: "401", fields: fields{ Indexer: "mock-indexer", TorrentName: "Test.Release-GROUP", TorrentURL: fmt.Sprintf("%v/%v", ts.URL, 401), }, wantErr: func(t assert.TestingT, err error, i ...interface{}) bool { return err != nil }, }, { name: "403", fields: fields{ Indexer: "mock-indexer", TorrentName: "Test.Release-GROUP", TorrentURL: fmt.Sprintf("%v/%v", ts.URL, 403), }, wantErr: func(t assert.TestingT, err error, i ...interface{}) bool { return err != nil }, }, { name: "ok", fields: fields{ Indexer: "mock-indexer", TorrentName: "Test.Release-GROUP", TorrentURL: fmt.Sprintf("%v/%v", ts.URL, "file.torrent"), }, wantErr: func(t assert.TestingT, err error, i ...interface{}) bool { return err != nil }, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { r := &Release{ ID: tt.fields.ID, FilterStatus: tt.fields.FilterStatus, Rejections: tt.fields.Rejections, Indexer: tt.fields.Indexer, FilterName: tt.fields.FilterName, Protocol: tt.fields.Protocol, Implementation: tt.fields.Implementation, Timestamp: tt.fields.Timestamp, GroupID: tt.fields.GroupID, TorrentID: tt.fields.TorrentID, TorrentURL: tt.fields.TorrentURL, TorrentTmpFile: tt.fields.TorrentTmpFile, TorrentDataRawBytes: tt.fields.TorrentDataRawBytes, TorrentHash: tt.fields.TorrentHash, TorrentName: tt.fields.TorrentName, Size: tt.fields.Size, Title: tt.fields.Title, Category: tt.fields.Category, Categories: tt.fields.Categories, Season: tt.fields.Season, Episode: tt.fields.Episode, Year: tt.fields.Year, Resolution: tt.fields.Resolution, Source: tt.fields.Source, Codec: tt.fields.Codec, Container: tt.fields.Container, HDR: tt.fields.HDR, Audio: tt.fields.Audio, AudioChannels: tt.fields.AudioChannels, Group: tt.fields.Group, Region: tt.fields.Region, Language: tt.fields.Language, Proper: tt.fields.Proper, Repack: tt.fields.Repack, Website: tt.fields.Website, Artists: tt.fields.Artists, Type: tt.fields.Type, LogScore: tt.fields.LogScore, Origin: tt.fields.Origin, Tags: tt.fields.Tags, ReleaseTags: tt.fields.ReleaseTags, Freeleech: tt.fields.Freeleech, FreeleechPercent: tt.fields.FreeleechPercent, Bonus: tt.fields.Bonus, Uploader: tt.fields.Uploader, PreTime: tt.fields.PreTime, Other: tt.fields.Other, RawCookie: tt.fields.RawCookie, AdditionalSizeCheckRequired: tt.fields.AdditionalSizeCheckRequired, FilterID: tt.fields.FilterID, Filter: tt.fields.Filter, ActionStatus: tt.fields.ActionStatus, } tt.wantErr(t, r.DownloadTorrentFile(), "DownloadTorrentFile()") }) } } func Test_getUniqueTags(t *testing.T) { type args struct { target []string source []string } tests := []struct { name string args args want []string }{ { name: "1", args: args{ target: []string{}, source: []string{"mp4"}, }, want: []string{"mp4"}, }, { name: "2", args: args{ target: []string{"mp4"}, source: []string{"mp4"}, }, want: []string{"mp4"}, }, { name: "3", args: args{ target: []string{"mp4"}, source: []string{"mp4", "dv"}, }, want: []string{"mp4", "dv"}, }, { name: "4", args: args{ target: []string{"dv"}, source: []string{"mp4", "dv"}, }, want: []string{"dv", "mp4"}, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { assert.Equalf(t, tt.want, getUniqueTags(tt.args.target, tt.args.source), "getUniqueTags(%v, %v)", tt.args.target, tt.args.source) }) } }