crawler_go/test/crawler_test.go

package test

import (
	"testing"
	"github.com/stretchr/testify/assert"
	"encoding/json"
	"git.loafle.net/overflow/crawler_go/health_crawler/redis_protocol_crawler_go"
	"git.loafle.net/overflow/crawler_go"

	"io/ioutil"
)

func initConfig() crawler.CrawlerImpl{
	c := crawler.CrawlerImpl{}
	c.SetInternal(redis_protocol_crawler_go.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))

}