12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091 |
- // Copyright (C) 2014 The Syncthing Authors.
- //
- // This Source Code Form is subject to the terms of the Mozilla Public
- // License, v. 2.0. If a copy of the MPL was not distributed with this file,
- // You can obtain one at https://mozilla.org/MPL/2.0/.
- package protocol
- import (
- "time"
- )
- type TestModel struct {
- data []byte
- folder string
- name string
- offset int64
- size int32
- hash []byte
- fromTemporary bool
- indexFn func(string, []FileInfo)
- ccFn func(*ClusterConfig)
- closedCh chan struct{}
- closedErr error
- }
- func newTestModel() *TestModel {
- return &TestModel{
- closedCh: make(chan struct{}),
- }
- }
- func (t *TestModel) Index(_ Connection, idx *Index) error {
- if t.indexFn != nil {
- t.indexFn(idx.Folder, idx.Files)
- }
- return nil
- }
- func (*TestModel) IndexUpdate(Connection, *IndexUpdate) error {
- return nil
- }
- func (t *TestModel) Request(_ Connection, req *Request) (RequestResponse, error) {
- t.folder = req.Folder
- t.name = req.Name
- t.offset = req.Offset
- t.size = int32(req.Size)
- t.hash = req.Hash
- t.fromTemporary = req.FromTemporary
- buf := make([]byte, len(t.data))
- copy(buf, t.data)
- return &fakeRequestResponse{buf}, nil
- }
- func (t *TestModel) Closed(_ Connection, err error) {
- t.closedErr = err
- close(t.closedCh)
- }
- func (t *TestModel) ClusterConfig(_ Connection, config *ClusterConfig) error {
- if t.ccFn != nil {
- t.ccFn(config)
- }
- return nil
- }
- func (*TestModel) DownloadProgress(Connection, *DownloadProgress) error {
- return nil
- }
- func (t *TestModel) closedError() error {
- select {
- case <-t.closedCh:
- return t.closedErr
- case <-time.After(1 * time.Second):
- return nil // Timeout
- }
- }
- type fakeRequestResponse struct {
- data []byte
- }
- func (r *fakeRequestResponse) Data() []byte {
- return r.data
- }
- func (*fakeRequestResponse) Close() {}
- func (*fakeRequestResponse) Wait() {}
|