94 lines
1.5 KiB
Go
94 lines
1.5 KiB
Go
package test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"git.loafle.net/overflow/overflow_probe/crawler"
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
|
|
"io/ioutil"
|
|
"git.loafle.net/overflow/overflow_probe/crawler/health_crawler/redis_protocol_crawler_go"
|
|
)
|
|
|
|
func initConfig() crawler.CrawlerImpl {
|
|
c := crawler.CrawlerImpl{}
|
|
c.SetInternal(redis_protocol_crawler.NewRedisHeahthCrawler())
|
|
|
|
b, _ := ioutil.ReadFile("./example.json")
|
|
|
|
c.Init(b)
|
|
return c
|
|
}
|
|
|
|
func TestCrawlerInit(t *testing.T) {
|
|
|
|
c := initConfig()
|
|
m := c.GetConfig("test_id_001")
|
|
assert.Equal(t, m.Id, "test_id_001")
|
|
|
|
}
|
|
|
|
func TestCrawlerAdd(t *testing.T) {
|
|
|
|
c := initConfig()
|
|
c.Remove("test_id_001")
|
|
bb, _ := ioutil.ReadFile("./example.json")
|
|
b, _ := c.Add(bb)
|
|
|
|
var ck bool
|
|
json.Unmarshal(b, &ck)
|
|
|
|
assert.Equal(t, true, ck)
|
|
|
|
}
|
|
|
|
func TestCrawlerRemove(t *testing.T) {
|
|
c := initConfig()
|
|
b, _ := c.Remove("test_id_001")
|
|
|
|
var ck bool
|
|
json.Unmarshal(b, &ck)
|
|
|
|
assert.Equal(t, true, ck)
|
|
}
|
|
|
|
func TestCrawlerGet(t *testing.T) {
|
|
c := initConfig()
|
|
b, _ := c.Get("test_id_001")
|
|
|
|
var ck bool
|
|
json.Unmarshal(b, &ck)
|
|
|
|
assert.Equal(t, true, ck)
|
|
}
|
|
|
|
func TestFolder(t *testing.T) {
|
|
|
|
configPath := "./"
|
|
files, err := ioutil.ReadDir(configPath)
|
|
if err != nil {
|
|
t.Log(err)
|
|
}
|
|
|
|
for _, file := range files {
|
|
if file.IsDir() == true {
|
|
continue
|
|
}
|
|
|
|
t.Log(file.Name())
|
|
//_,err := c.Add(file.Name())
|
|
if err != nil {
|
|
t.Log(err)
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
func TestFile(t *testing.T) {
|
|
|
|
b, _ := ioutil.ReadFile("./example.json")
|
|
|
|
t.Log(string(b))
|
|
|
|
}
|