From 6a31a05e9a8440eaad91b2ac198059e9ec4a53fa Mon Sep 17 00:00:00 2001
From: l3uddz <l3uddz@gmail.com>
Date: Sun, 21 Nov 2021 13:33:44 +0000
Subject: [PATCH] test: newznab category tests

---
 util/newznab_test.go | 92 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 92 insertions(+)
 create mode 100644 util/newznab_test.go

diff --git a/util/newznab_test.go b/util/newznab_test.go
new file mode 100644
index 0000000..a4601f4
--- /dev/null
+++ b/util/newznab_test.go
@@ -0,0 +1,92 @@
+package util
+
+import "testing"
+
+func TestContainsMovieCategory(t *testing.T) {
+	type args struct {
+		cats []string
+	}
+	tests := []struct {
+		name string
+		args args
+		want bool
+	}{
+		{
+			name: "want true",
+			args: args{
+				cats: []string{
+					"2000",
+					"5000",
+				},
+			},
+			want: true,
+		}, {
+			name: "want false",
+			args: args{
+				cats: []string{
+					"3000",
+					"4000",
+				},
+			},
+			want: false,
+		}, {
+			name: "want false",
+			args: args{
+				cats: []string{},
+			},
+			want: false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := ContainsMovieCategory(tt.args.cats); got != tt.want {
+				t.Errorf("ContainsMovieCategory() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func TestContainsTvCategory(t *testing.T) {
+	type args struct {
+		cats []string
+	}
+	tests := []struct {
+		name string
+		args args
+		want bool
+	}{
+		{
+			name: "want true",
+			args: args{
+				cats: []string{
+					"5000",
+					"4000",
+					"2000",
+				},
+			},
+			want: true,
+		}, {
+			name: "want false",
+			args: args{
+				cats: []string{
+					"3000",
+					"4000",
+				},
+			},
+			want: false,
+		}, {
+			name: "want false",
+			args: args{
+				cats: []string{},
+			},
+			want: false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := ContainsTvCategory(tt.args.cats); got != tt.want {
+				t.Errorf("ContainsTvCategory() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}