143 lines
4.3 KiB
Go
143 lines
4.3 KiB
Go
package task_logs
|
|
|
|
/* ***** BEGIN GPL LICENSE BLOCK *****
|
|
*
|
|
* Original Code Copyright (C) 2022 Blender Foundation.
|
|
*
|
|
* This file is part of Flamenco.
|
|
*
|
|
* Flamenco is free software: you can redistribute it and/or modify it under
|
|
* the terms of the GNU General Public License as published by the Free Software
|
|
* Foundation, either version 3 of the License, or (at your option) any later
|
|
* version.
|
|
*
|
|
* Flamenco is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
|
|
* A PARTICULAR PURPOSE. See the GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along with
|
|
* Flamenco. If not, see <https://www.gnu.org/licenses/>.
|
|
*
|
|
* ***** END GPL LICENSE BLOCK ***** */
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func setUpTest(t *testing.T) string {
|
|
temppath, err := ioutil.TempDir("", "testlogs")
|
|
assert.NoError(t, err)
|
|
return temppath
|
|
}
|
|
|
|
func tearDownTest(temppath string) {
|
|
os.RemoveAll(temppath)
|
|
}
|
|
|
|
func TestCreateNumberedPath(t *testing.T) {
|
|
temppath := setUpTest(t)
|
|
defer tearDownTest(temppath)
|
|
|
|
numtest := func(path string, number int, basepath string) {
|
|
result := createNumberedPath(path)
|
|
assert.Equal(t, numberedPath{path, number, basepath}, result)
|
|
}
|
|
|
|
numtest("", -1, "")
|
|
numtest(" ", -1, " ")
|
|
numtest("jemoeder.1", 1, "jemoeder")
|
|
numtest("jemoeder.", -1, "jemoeder.")
|
|
numtest("jemoeder", -1, "jemoeder")
|
|
numtest("jemoeder.abc", -1, "jemoeder.abc")
|
|
numtest("jemoeder.-4", -4, "jemoeder")
|
|
numtest("jemoeder.1.2.3", 3, "jemoeder.1.2")
|
|
numtest("jemoeder.001", 1, "jemoeder")
|
|
numtest("jemoeder.01", 1, "jemoeder")
|
|
numtest("jemoeder.010", 10, "jemoeder")
|
|
numtest("jemoeder 47 42.327", 327, "jemoeder 47 42")
|
|
numtest("/path/üničøde.327/.47", 47, "/path/üničøde.327/")
|
|
numtest("üničøde.327.what?", -1, "üničøde.327.what?")
|
|
}
|
|
|
|
func TestNoFiles(t *testing.T) {
|
|
temppath := setUpTest(t)
|
|
defer tearDownTest(temppath)
|
|
|
|
filepath := filepath.Join(temppath, "nonexisting.txt")
|
|
err := rotateLogFile(zerolog.Nop(), filepath)
|
|
assert.NoError(t, err)
|
|
assert.False(t, fileExists(filepath))
|
|
}
|
|
|
|
func TestOneFile(t *testing.T) {
|
|
temppath := setUpTest(t)
|
|
defer tearDownTest(temppath)
|
|
|
|
filepath := filepath.Join(temppath, "existing.txt")
|
|
fileTouch(filepath)
|
|
|
|
err := rotateLogFile(zerolog.Nop(), filepath)
|
|
assert.NoError(t, err)
|
|
assert.False(t, fileExists(filepath))
|
|
assert.True(t, fileExists(filepath+".1"))
|
|
}
|
|
|
|
func TestMultipleFilesWithHoles(t *testing.T) {
|
|
temppath := setUpTest(t)
|
|
defer tearDownTest(temppath)
|
|
|
|
filepath := filepath.Join(temppath, "existing.txt")
|
|
assert.NoError(t, ioutil.WriteFile(filepath, []byte("thefile"), 0666))
|
|
assert.NoError(t, ioutil.WriteFile(filepath+".1", []byte("file .1"), 0666))
|
|
assert.NoError(t, ioutil.WriteFile(filepath+".2", []byte("file .2"), 0666))
|
|
assert.NoError(t, ioutil.WriteFile(filepath+".3", []byte("file .3"), 0666))
|
|
assert.NoError(t, ioutil.WriteFile(filepath+".5", []byte("file .5"), 0666))
|
|
assert.NoError(t, ioutil.WriteFile(filepath+".7", []byte("file .7"), 0666))
|
|
|
|
err := rotateLogFile(zerolog.Nop(), filepath)
|
|
|
|
assert.NoError(t, err)
|
|
assert.False(t, fileExists(filepath))
|
|
assert.True(t, fileExists(filepath+".1"))
|
|
assert.True(t, fileExists(filepath+".2"))
|
|
assert.True(t, fileExists(filepath+".3"))
|
|
assert.True(t, fileExists(filepath+".4"))
|
|
assert.False(t, fileExists(filepath+".5"))
|
|
assert.True(t, fileExists(filepath+".6"))
|
|
assert.False(t, fileExists(filepath+".7"))
|
|
assert.True(t, fileExists(filepath+".8"))
|
|
assert.False(t, fileExists(filepath+".9"))
|
|
|
|
read := func(filename string) string {
|
|
content, err := ioutil.ReadFile(filename)
|
|
assert.NoError(t, err)
|
|
return string(content)
|
|
}
|
|
|
|
assert.Equal(t, "thefile", read(filepath+".1"))
|
|
assert.Equal(t, "file .1", read(filepath+".2"))
|
|
assert.Equal(t, "file .2", read(filepath+".3"))
|
|
assert.Equal(t, "file .3", read(filepath+".4"))
|
|
assert.Equal(t, "file .5", read(filepath+".6"))
|
|
assert.Equal(t, "file .7", read(filepath+".8"))
|
|
}
|
|
|
|
func fileExists(filename string) bool {
|
|
_, err := os.Stat(filename)
|
|
return !os.IsNotExist(err)
|
|
}
|
|
|
|
func fileTouch(filename string) {
|
|
file, err := os.OpenFile(filename, os.O_CREATE|os.O_APPEND|os.O_RDONLY, 0666)
|
|
if err != nil {
|
|
panic(err.Error())
|
|
}
|
|
file.Close()
|
|
}
|