106 lines
2.5 KiB
Go
106 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/tardisx/discord-auto-upload/config"
|
|
"github.com/tardisx/discord-auto-upload/upload"
|
|
)
|
|
|
|
func TestWatchNewFiles(t *testing.T) {
|
|
dir := createFileTree()
|
|
defer os.RemoveAll(dir)
|
|
time.Sleep(time.Second)
|
|
|
|
w := watch{
|
|
config: config.Watcher{Path: dir},
|
|
uploader: upload.NewUploader(),
|
|
lastCheck: time.Now(),
|
|
newLastCheck: time.Now(),
|
|
}
|
|
files := w.ProcessNewFiles()
|
|
if len(files) != 0 {
|
|
t.Errorf("was not zero files (%d): %v", len(files), files)
|
|
}
|
|
|
|
// create a new file
|
|
time.Sleep(time.Second)
|
|
os.Create(fmt.Sprintf("%s%c%s", dir, os.PathSeparator, "b.gif"))
|
|
files = w.ProcessNewFiles()
|
|
if len(files) != 1 {
|
|
t.Errorf("was not one file - got: %v", files)
|
|
}
|
|
if files[0] != fmt.Sprintf("%s%c%s", dir, os.PathSeparator, "b.gif") {
|
|
t.Error("wrong file")
|
|
}
|
|
}
|
|
|
|
func TestExclsion(t *testing.T) {
|
|
dir := createFileTree()
|
|
defer os.RemoveAll(dir)
|
|
time.Sleep(time.Second)
|
|
|
|
w := watch{
|
|
config: config.Watcher{Path: dir, Exclude: []string{"thumb", "tiny"}},
|
|
uploader: upload.NewUploader(),
|
|
lastCheck: time.Now(),
|
|
newLastCheck: time.Now(),
|
|
}
|
|
files := w.ProcessNewFiles()
|
|
if len(files) != 0 {
|
|
t.Errorf("was not zero files (%d): %v", len(files), files)
|
|
}
|
|
// create a new file that would not hit exclusion, and two that would
|
|
time.Sleep(time.Second)
|
|
os.Create(fmt.Sprintf("%s%c%s", dir, os.PathSeparator, "b.gif"))
|
|
os.Create(fmt.Sprintf("%s%c%s", dir, os.PathSeparator, "b_thumb.gif"))
|
|
os.Create(fmt.Sprintf("%s%c%s", dir, os.PathSeparator, "tiny_b.jpg"))
|
|
files = w.ProcessNewFiles()
|
|
if len(files) != 1 {
|
|
t.Error("was not one new file")
|
|
}
|
|
|
|
}
|
|
|
|
func TestCheckPath(t *testing.T) {
|
|
dir := createFileTree()
|
|
defer os.RemoveAll(dir)
|
|
|
|
w := watch{
|
|
config: config.Watcher{Path: dir},
|
|
uploader: upload.NewUploader(),
|
|
lastCheck: time.Now(),
|
|
newLastCheck: time.Now(),
|
|
}
|
|
if !w.checkPath() {
|
|
t.Error("checkPath failed?")
|
|
}
|
|
|
|
err := os.RemoveAll(dir)
|
|
if err != nil {
|
|
t.Fatalf("could not remove test dir: %v", err)
|
|
}
|
|
if w.checkPath() {
|
|
t.Error("checkPath succeeded when shouldn't?")
|
|
}
|
|
}
|
|
|
|
func createFileTree() string {
|
|
dir, err := ioutil.TempDir("", "dau-test")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
f1, _ := os.Create(fmt.Sprintf("%s%c%s", dir, os.PathSeparator, "a.gif"))
|
|
f2, _ := os.Create(fmt.Sprintf("%s%c%s", dir, os.PathSeparator, "a.jpg"))
|
|
f3, _ := os.Create(fmt.Sprintf("%s%c%s", dir, os.PathSeparator, "a.png"))
|
|
f1.Close()
|
|
f2.Close()
|
|
f3.Close()
|
|
return dir
|
|
}
|