gogs/truckstop
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

states are explicit

Browse Source
This commit is contained in:
Bel LaPointe
2022-01-27 17:27:08 -07:00
parent e19cd7095d
commit 5406250af3
7 changed files with 25 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
broker/broker.go
View File

@@ -20,7 +20,7 @@ var authlimiter = rate.NewLimiter(rate.Limit(1.0/60.0), 1)
var limiter = rate.NewLimiter(rate.Limit(1.0/20.0), 1)
type Broker interface {
Search([]config.State) ([]Job, error)
SearchStates([]config.State) ([]Job, error)
}
func do(r *http.Request) (*http.Response, error) {

8
broker/fastexact.go
View File

@@ -38,13 +38,13 @@ func (fe FastExact) WithMock() FastExact {
return fe
}
func (fe FastExact) Search(states []config.State) ([]Job, error) {
jobs, err := fe.search(states)
func (fe FastExact) SearchStates(states []config.State) ([]Job, error) {
jobs, err := fe.searchStates(states)
if err == ErrNoAuth {
if err := fe.login(); err != nil {
return nil, err
}
jobs, err = fe.search(states)
jobs, err = fe.searchStates(states)
}
return jobs, err
}
@@ -91,7 +91,7 @@ func (fe FastExact) setHeaders(req *http.Request) {
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
}
func (fe FastExact) search(states []config.State) ([]Job, error) {
func (fe FastExact) searchStates(states []config.State) ([]Job, error) {
var jobs []Job
for _, state := range states {
subjobs, err := fe.searchOne(state)

6
broker/fastexact_test.go
View File

@@ -34,7 +34,7 @@ func TestFastExactReal(t *testing.T) {
if err := fe.login(); err != nil {
t.Fatal(err)
}
jobs, err := fe.search(states)
jobs, err := fe.searchStates(states)
if err != nil {
t.Fatal(err)
}
@@ -56,14 +56,14 @@ func TestFastExactLogin(t *testing.T) {
}
}
func TestFastExactSearch(t *testing.T) {
func TestFastExactSearchStates(t *testing.T) {
logtr.SetLevel(logtr.SOS)
limiter = rate.NewLimiter(rate.Limit(60.0), 1)
fe := NewFastExact().WithMock()
_ = fe
db := storage.NewMap()
_ = db
if jobs, err := fe.search([]config.State{config.State("NC"), config.State("SC")}); err != nil {
if jobs, err := fe.searchStates([]config.State{config.State("NC"), config.State("SC")}); err != nil {
t.Fatal(err)
} else if len(jobs) != 10 {
t.Fatal(len(jobs))

14
broker/ntgvision.go
View File

@@ -15,7 +15,7 @@ import (
type NTGVision struct {
searcher interface {
search(states []config.State) (io.ReadCloser, error)
searchStates(states []config.State) (io.ReadCloser, error)
searchJobReadCloser(id int64) (io.ReadCloser, error)
}
}
@@ -187,8 +187,8 @@ func (ntg NTGVision) searchJob(id int64) (ntgVisionJobInfo, error) {
return result, err
}
func (ntg NTGVision) Search(states []config.State) ([]Job, error) {
rc, err := ntg.searcher.search(states)
func (ntg NTGVision) SearchStates(states []config.State) ([]Job, error) {
rc, err := ntg.searcher.searchStates(states)
if err != nil {
return nil, err
}
@@ -226,20 +226,20 @@ func setNTGToken(token string) {
db.Set(getNTGTokenKey(), []byte(token))
}
func (ntg NTGVision) search(states []config.State) (io.ReadCloser, error) {
func (ntg NTGVision) searchStates(states []config.State) (io.ReadCloser, error) {
if getNTGToken() == "" {
logtr.Debugf("NTG token is empty, refreshing ntg auth")
if err := ntg.refreshAuth(); err != nil {
return nil, err
}
}
rc, err := ntg._search(states)
rc, err := ntg._searchStates(states)
if err == ErrNoAuth {
logtr.Debugf("err no auth on search, refreshing ntg auth")
if err := ntg.refreshAuth(); err != nil {
return nil, err
}
rc, err = ntg._search(states)
rc, err = ntg._searchStates(states)
}
return rc, err
}
@@ -285,7 +285,7 @@ func (ntg NTGVision) _refreshAuth() error {
return nil
}
func (ntg NTGVision) _search(states []config.State) (io.ReadCloser, error) {
func (ntg NTGVision) _searchStates(states []config.State) (io.ReadCloser, error) {
request, err := ntg.newRequest(states)
if err != nil {
return nil, err

2
broker/ntgvision_mock.go
View File

@@ -15,7 +15,7 @@ func NewNTGVisionMock() NTGVisionMock {
return NTGVisionMock{}
}
func (ntgm NTGVisionMock) search(states []config.State) (io.ReadCloser, error) {
func (ntgm NTGVisionMock) searchStates(states []config.State) (io.ReadCloser, error) {
path := path.Join(os.Getenv("GOPATH"), "src", "local", "truckstop", "broker", "testdata", "ntgvision_response.json")
b, err := ioutil.ReadFile(path)
return io.NopCloser(bytes.NewReader(b)), err