custom_crawl_test.go

package core

import (
	"github.com/datatogether/sql_datastore"
	"github.com/ipfs/go-datastore"
	"testing"
)

func TestCustomCrawlStorage(t *testing.T) {

	store := datastore.NewMapDatastore()

	p := &CustomCrawl{Id: "eg", OriginalUrl: "http://test.com"}
	if err := p.Save(store); err != nil {
		t.Error(err.Error())
		return
	}

	p.Jwt = "eg_jwt"
	if err := p.Save(store); err != nil {
		t.Error(err.Error())
		return
	}

	p2 := &CustomCrawl{Id: p.Id}
	if err := p2.Read(store); err != nil {
		t.Error(err.Error())
		return
	}

	if !p2.Created.Equal(p.Created) {
		t.Errorf("created doesn't match: %s != %s", p2.Created.String(), p.Created.String())
	}

	if !p2.Updated.Equal(p.Updated) {
		t.Errorf("updated doesn't match: %s != %s", p2.Updated.String(), p.Updated.String())
	}

	if err := p.Delete(store); err != nil {
		t.Error(err.Error())
		return
	}
}

func TestCustomCrawlSQLStorage(t *testing.T) {
	defer resetTestData(appDB, "uncrawlables")

	store := sql_datastore.NewDatastore(appDB)
	if err := store.Register(&CustomCrawl{}); err != nil {
		t.Error(err)
		return
	}

	p := &CustomCrawl{SqliteChecksum: "eg"}
	if err := p.Save(store); err != nil {
		t.Error(err.Error())
		return
	}

	p.SqliteChecksum = "eg_2"
	if err := p.Save(store); err != nil {
		t.Error(err.Error())
		return
	}

	p2 := &CustomCrawl{Id: p.Id}
	if err := p2.Read(store); err != nil {
		t.Error(err.Error())
		return
	}

	if !p2.Created.Equal(p.Created) {
		t.Errorf("created doesn't match: %s != %s", p2.Created.String(), p.Created.String())
	}

	if !p2.Updated.Equal(p.Updated) {
		t.Errorf("updated doesn't match: %s != %s", p2.Updated.String(), p.Updated.String())
	}

	if err := p.Delete(store); err != nil {
		t.Error(err.Error())
		return
	}
}