git.r.bdr.sh - rbdr/mobius/blob - hotline/file_store.go

   1 package hotline
   2
   3 import (
   4         "github.com/stretchr/testify/mock"
   5         "os"
   6 )
   7
   8 var FS FileStore
   9
  10 type FileStore interface {
  11         Mkdir(name string, perm os.FileMode) error
  12         Stat(name string) (os.FileInfo, error)
  13         Open(name string) (*os.File, error)
  14         // TODO: implement these
  15         //Rename(oldpath string, newpath string) error
  16         //RemoveAll(path string) error
  17 }
  18
  19 type OSFileStore struct{}
  20
  21 func (fs OSFileStore) Mkdir(name string, perm os.FileMode) error {
  22         return os.Mkdir(name, perm)
  23 }
  24
  25 func (fs OSFileStore) Stat(name string) (os.FileInfo, error) {
  26         return os.Stat(name)
  27 }
  28
  29 func (fs OSFileStore) Open(name string) (*os.File, error) {
  30         return os.Open(name)
  31 }
  32
  33 type MockFileStore struct {
  34         mock.Mock
  35 }
  36
  37 func (mfs MockFileStore) Mkdir(name string, perm os.FileMode) error {
  38         args := mfs.Called(name, perm)
  39         return args.Error(0)
  40 }
  41
  42 func (mfs MockFileStore) Stat(name string) (os.FileInfo, error) {
  43         args := mfs.Called(name)
  44         if args.Get(0) == nil {
  45                 return nil, args.Error(1)
  46
  47         }
  48         return args.Get(0).(os.FileInfo), args.Error(1)
  49 }
  50
  51 func (mfs MockFileStore) Open(name string) (*os.File, error) {
  52         args := mfs.Called(name)
  53         return args.Get(0).(*os.File), args.Error(1)
  54 }