ChineseSubFinder/internal/logic/task_queue/daily_download_limit_test.go

package task_queue

import (
	"testing"
)

const supplieName = "testtest"

func TestGetDailyDownloadCount(t *testing.T) {
	type args struct {
		supplierName string
		whichDay     []string
	}
	tests := []struct {
		name    string
		args    args
		want    int
		wantErr bool
	}{
		{name: "00", args: args{
			supplierName: supplieName,
			whichDay:     nil,
		}, want: 0, wantErr: false},
		{name: "01", args: args{
			supplierName: supplieName,
			whichDay:     []string{"2022-04-01"},
		}, want: 0, wantErr: false},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {

			got, err := GetDailyDownloadCount(tt.args.supplierName, "", tt.args.whichDay...)
			if (err != nil) != tt.wantErr {
				t.Errorf("GetDailyDownloadCount() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if got != tt.want {
				t.Errorf("GetDailyDownloadCount() got = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestAddDailyDownloadCount(t *testing.T) {
	type args struct {
		supplierName string
	}
	tests := []struct {
		name    string
		args    args
		want    int
		wantErr bool
	}{
		{name: "00", args: args{
			supplierName: supplieName,
		}, want: 1, wantErr: false},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {

			got, err := AddDailyDownloadCount(tt.args.supplierName, "")
			if (err != nil) != tt.wantErr {
				t.Errorf("AddDailyDownloadCount() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if got != tt.want {
				t.Errorf("AddDailyDownloadCount() got = %v, want %v", got, tt.want)
			}

			got, err = AddDailyDownloadCount(tt.args.supplierName, "")
			if (err != nil) != tt.wantErr {
				t.Errorf("AddDailyDownloadCount() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if got != tt.want+1 {
				t.Errorf("AddDailyDownloadCount() got = %v, want %v", got, tt.want)
			}
		})
	}

	err := DelDb()
	if err != nil {
		return
	}
}

func TestAddGetDailyDownloadCount(t *testing.T) {

	addCount, err := AddDailyDownloadCount(supplieName, "")
	if err != nil {
		t.Fatalf(err.Error())
	}

	getDailyDownloadCount, err := GetDailyDownloadCount(supplieName, "")
	if err != nil {
		t.Fatalf(err.Error())
	}

	if addCount != getDailyDownloadCount {
		t.Fatalf("not the same")
	}

	addCount, err = AddDailyDownloadCount(supplieName, "")
	if err != nil {
		t.Fatalf(err.Error())
	}

	getDailyDownloadCount, err = GetDailyDownloadCount(supplieName, "")
	if err != nil {
		t.Fatalf(err.Error())
	}

	if addCount != getDailyDownloadCount {
		t.Fatalf("not the same")
	}

	err = DelDb()
	if err != nil {
		return
	}
}