github.com/Prakhar-Agarwal-byte/moby@v0.0.0-20231027092010-a14e3e8ab87e/libnetwork/drivers/overlay/overlay_test.go (about)

     1  //go:build linux
     2  
     3  package overlay
     4  
     5  import (
     6  	"testing"
     7  
     8  	"github.com/Prakhar-Agarwal-byte/moby/libnetwork/driverapi"
     9  	"github.com/Prakhar-Agarwal-byte/moby/pkg/plugingetter"
    10  )
    11  
    12  type driverTester struct {
    13  	t *testing.T
    14  	d *driver
    15  }
    16  
    17  const testNetworkType = "overlay"
    18  
    19  func (dt *driverTester) GetPluginGetter() plugingetter.PluginGetter {
    20  	return nil
    21  }
    22  
    23  func (dt *driverTester) RegisterDriver(name string, drv driverapi.Driver, cap driverapi.Capability) error {
    24  	if name != testNetworkType {
    25  		dt.t.Fatalf("Expected driver register name to be %q. Instead got %q",
    26  			testNetworkType, name)
    27  	}
    28  
    29  	if _, ok := drv.(*driver); !ok {
    30  		dt.t.Fatalf("Expected driver type to be %T. Instead got %T",
    31  			&driver{}, drv)
    32  	}
    33  
    34  	dt.d = drv.(*driver)
    35  	return nil
    36  }
    37  
    38  func TestOverlayInit(t *testing.T) {
    39  	if err := Register(&driverTester{t: t}, nil); err != nil {
    40  		t.Fatal(err)
    41  	}
    42  }
    43  
    44  func TestOverlayType(t *testing.T) {
    45  	dt := &driverTester{t: t}
    46  	if err := Register(dt, nil); err != nil {
    47  		t.Fatal(err)
    48  	}
    49  
    50  	if dt.d.Type() != testNetworkType {
    51  		t.Fatalf("Expected Type() to return %q. Instead got %q", testNetworkType,
    52  			dt.d.Type())
    53  	}
    54  }