8 "github.com/edwarnicke/exechelper"
9 "github.com/stretchr/testify/assert"
10 "github.com/stretchr/testify/suite"
15 defaultNamespaceName string = "default"
18 func IsPersistent() bool {
19 return os.Getenv("HST_PERSIST") == "1"
22 func IsVerbose() bool {
23 return os.Getenv("HST_VERBOSE") == "1"
26 type HstSuite struct {
28 containers map[string]*Container
30 netConfigs []NetConfig
31 netInterfaces map[string]NetInterface
35 func (s *HstSuite) TearDownSuite() {
36 s.unconfigureNetworkTopology()
39 func (s *HstSuite) TearDownTest() {
47 func (s *HstSuite) SetupTest() {
52 func (s *HstSuite) SetupVolumes() {
53 for _, volume := range s.volumes {
54 cmd := "docker volume create --name=" + volume
60 func (s *HstSuite) SetupContainers() {
61 for _, container := range s.containers {
62 if container.isOptional == false {
68 func (s *HstSuite) hstFail() {
72 func (s *HstSuite) assertNil(object interface{}, msgAndArgs ...interface{}) {
73 if !assert.Nil(s.T(), object, msgAndArgs...) {
78 func (s *HstSuite) assertNotNil(object interface{}, msgAndArgs ...interface{}) {
79 if !assert.NotNil(s.T(), object, msgAndArgs...) {
84 func (s *HstSuite) assertEqual(expected, actual interface{}, msgAndArgs ...interface{}) {
85 if !assert.Equal(s.T(), expected, actual, msgAndArgs...) {
90 func (s *HstSuite) assertNotEqual(expected, actual interface{}, msgAndArgs ...interface{}) {
91 if !assert.NotEqual(s.T(), expected, actual, msgAndArgs...) {
96 func (s *HstSuite) assertContains(testString, contains interface{}, msgAndArgs ...interface{}) {
97 if !assert.Contains(s.T(), testString, contains, msgAndArgs...) {
102 func (s *HstSuite) assertNotContains(testString, contains interface{}, msgAndArgs ...interface{}) {
103 if !assert.NotContains(s.T(), testString, contains, msgAndArgs...) {
108 func (s *HstSuite) assertNotEmpty(object interface{}, msgAndArgs ...interface{}) {
109 if !assert.NotEmpty(s.T(), object, msgAndArgs...) {
114 func (s *HstSuite) log(args ...any) {
120 func (s *HstSuite) skip(args ...any) {
125 func (s *HstSuite) ResetContainers() {
126 for _, container := range s.containers {
131 func (s *HstSuite) RemoveVolumes() {
132 for _, volumeName := range s.volumes {
133 cmd := "docker volume rm " + volumeName
135 os.RemoveAll(volumeName)
139 func (s *HstSuite) getContainerByName(name string) *Container {
140 return s.containers[name]
144 * Create a copy and return its address, so that individial tests which call this
145 * are not able to modify the original container and affect other tests by doing that
147 func (s *HstSuite) getTransientContainerByName(name string) *Container {
148 containerCopy := *s.containers[name]
149 return &containerCopy
152 func (s *HstSuite) loadContainerTopology(topologyName string) {
153 data, err := ioutil.ReadFile(ContainerTopologyDir + topologyName + ".yaml")
155 s.T().Fatalf("read error: %v", err)
157 var yamlTopo YamlTopology
158 err = yaml.Unmarshal(data, &yamlTopo)
160 s.T().Fatalf("unmarshal error: %v", err)
163 for _, elem := range yamlTopo.Volumes {
164 volumeMap := elem["volume"].(VolumeConfig)
165 hostDir := volumeMap["host-dir"].(string)
166 s.volumes = append(s.volumes, hostDir)
169 s.containers = make(map[string]*Container)
170 for _, elem := range yamlTopo.Containers {
171 newContainer, err := NewContainer(elem)
172 newContainer.suite = s
174 s.T().Fatalf("container config error: %v", err)
176 s.log(newContainer.getRunCommand())
177 s.containers[newContainer.name] = newContainer
181 func (s *HstSuite) loadNetworkTopology(topologyName string) {
182 data, err := ioutil.ReadFile(NetworkTopologyDir + topologyName + ".yaml")
184 s.T().Fatalf("read error: %v", err)
186 var yamlTopo YamlTopology
187 err = yaml.Unmarshal(data, &yamlTopo)
189 s.T().Fatalf("unmarshal error: %v", err)
192 s.addresser = NewAddresser(s)
193 s.netInterfaces = make(map[string]NetInterface)
194 for _, elem := range yamlTopo.Devices {
195 switch elem["type"].(string) {
198 if namespace, err := NewNetNamespace(elem); err == nil {
199 s.netConfigs = append(s.netConfigs, &namespace)
201 s.T().Fatalf("network config error: %v", err)
206 if veth, err := NewVeth(elem, s.addresser); err == nil {
207 s.netConfigs = append(s.netConfigs, &veth)
208 s.netInterfaces[veth.Name()] = &veth
210 s.T().Fatalf("network config error: %v", err)
215 if tap, err := NewTap(elem, s.addresser); err == nil {
216 s.netConfigs = append(s.netConfigs, &tap)
217 s.netInterfaces[tap.Name()] = &tap
219 s.T().Fatalf("network config error: %v", err)
224 if bridge, err := NewBridge(elem); err == nil {
225 s.netConfigs = append(s.netConfigs, &bridge)
227 s.T().Fatalf("network config error: %v", err)
234 func (s *HstSuite) configureNetworkTopology(topologyName string) {
235 s.loadNetworkTopology(topologyName)
237 for _, nc := range s.netConfigs {
238 if err := nc.Configure(); err != nil {
239 s.T().Fatalf("network config error: %v", err)
244 func (s *HstSuite) unconfigureNetworkTopology() {
248 for _, nc := range s.netConfigs {
253 type NamespaceAddresses struct {
255 numberOfAddresses int
258 type Addresser struct {
259 namespaces []*NamespaceAddresses
263 func (a *Addresser) AddNamespace(name string) {
264 var newNamespace = &NamespaceAddresses{
266 numberOfAddresses: 0,
268 a.namespaces = append(a.namespaces, newNamespace)
271 func (a *Addresser) NewIp4Address() (string, error) {
272 return a.NewIp4AddressWithNamespace(defaultNamespaceName)
275 func (a *Addresser) NewIp4AddressWithNamespace(namespace string) (string, error) {
276 for i, val := range a.namespaces {
277 if val.namespace != namespace {
280 if val.numberOfAddresses == 255 {
281 return "", fmt.Errorf("no available IPv4 addresses")
283 address := fmt.Sprintf("10.10.%v.%v/24", i, val.numberOfAddresses+1)
284 val.numberOfAddresses++
287 a.AddNamespace(namespace)
288 return a.NewIp4AddressWithNamespace(namespace)
291 func NewAddresser(suite *HstSuite) *Addresser {
292 var addresser = new(Addresser)
293 addresser.suite = suite
294 addresser.AddNamespace(defaultNamespaceName)