diff --git a/fake_delayer.go b/fake_delayer.go new file mode 100644 index 0000000..1b181fb --- /dev/null +++ b/fake_delayer.go @@ -0,0 +1,39 @@ +// Copyright 2021 Google LLC +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// https://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +package main + +import ( + "context" + "log" +) + +type FakeDelayer struct { + DelayOnChan bool + StopDelay chan bool +} + +func (f *FakeDelayer) Delay() { + f.DelayContext(context.Background()) +} + +func (f *FakeDelayer) DelayContext(ctx context.Context) bool { + log.Printf("Faking Backoff") + if f.DelayOnChan { + log.Printf("Waiting StopDelay signal") + <-f.StopDelay + log.Printf("Received StopDelay signal") + } + return true +} diff --git a/irc_test.go b/irc_test.go index 970aeaf..bba4cd8 100644 --- a/irc_test.go +++ b/irc_test.go @@ -195,25 +195,6 @@ func makeTestServer(t *testing.T) (*testServer, int) { return server, addr.Port } -type FakeDelayer struct { - DelayOnChan bool - StopDelay chan bool -} - -func (f *FakeDelayer) Delay() { - f.DelayContext(context.Background()) -} - -func (f *FakeDelayer) DelayContext(ctx context.Context) bool { - log.Printf("Faking Backoff") - if f.DelayOnChan { - log.Printf("Waiting StopDelay signal") - <-f.StopDelay - log.Printf("Received StopDelay signal") - } - return true -} - func makeTestIRCConfig(IRCPort int) *Config { return &Config{ IRCNick: "foo",