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  }