github.com/jiasir/deis@v1.12.2/logger/storage/factory_test.go (about) 1 package storage 2 3 import ( 4 "fmt" 5 "io/ioutil" 6 "os" 7 "reflect" 8 "testing" 9 ) 10 11 func TestGetUsingInvalidValues(t *testing.T) { 12 adapterStrs := []string{"bogus", "memory:", "memory:foo"} 13 for _, adapterStr := range adapterStrs { 14 _, err := NewAdapter(adapterStr) 15 if err == nil || err.Error() != fmt.Sprintf("Unrecognized storage adapter type: '%s'", adapterStr) { 16 t.Error("Did not receive expected error message") 17 } 18 } 19 } 20 21 func TestGetUsingEmptyString(t *testing.T) { 22 a, err := NewAdapter("") 23 if err != nil { 24 t.Error(err) 25 } 26 // Should use file-based log storage by default 27 expected := "*file.adapter" 28 aType := reflect.TypeOf(a).String() 29 if aType != expected { 30 t.Errorf("Expected a %s, but got a %s", expected, aType) 31 } 32 } 33 34 func TestGetFileBasedAdapter(t *testing.T) { 35 a, err := NewAdapter("file") 36 if err != nil { 37 t.Error(err) 38 } 39 expected := "*file.adapter" 40 aType := reflect.TypeOf(a).String() 41 if aType != expected { 42 t.Errorf("Expected a %s, but got a %s", expected, aType) 43 } 44 } 45 46 func TestGetMemoryBasedAdapter(t *testing.T) { 47 a, err := NewAdapter("memory") 48 if err != nil { 49 t.Error(err) 50 } 51 expected := "*ringbuffer.adapter" 52 aType := reflect.TypeOf(a).String() 53 if aType != expected { 54 t.Errorf("Expected a %s, but got a %s", expected, aType) 55 } 56 } 57 58 func TestGetMemoryBasedAdapterWithBufferSize(t *testing.T) { 59 a, err := NewAdapter("memory:1000") 60 if err != nil { 61 t.Error(err) 62 } 63 expected := "*ringbuffer.adapter" 64 aType := reflect.TypeOf(a).String() 65 if aType != expected { 66 t.Errorf("Expected a %s, but got a %s", expected, aType) 67 } 68 } 69 70 func TestMain(m *testing.M) { 71 LogRoot, _ = ioutil.TempDir("", "log-tests") 72 defer os.Remove(LogRoot) 73 os.Exit(m.Run()) 74 }