microservice-test.go

// Copyright (c) 2018 ef-ds
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.

// Package benchmark contains benchmark tests targeted to test the performance
// and efficiency of data structures.
package benchmark

import (
	"strconv"
	"testing"
)

// Microservice tests the data structures performance by simulating the data structure being used by microservice
// and serverless systems when running in production environments.
func (t *Tests) Microservice(b *testing.B, initInstance func(), add func(v interface{}), remove func() (interface{}, bool), empty func() bool) {
	for _, test := range tests {
		b.Run(strconv.Itoa(test.count), func(b *testing.B) {
			for n := 0; n < b.N; n++ {
				initInstance()

				// Simulate stable traffic
				for i := 0; i < test.count; i++ {
					add(GetTestValue(i))
					remove()
				}

				// Simulate slowly increasing traffic
				for i := 0; i < test.count; i++ {
					add(GetTestValue(i))
					add(GetTestValue(i))
					remove()
				}

				// Simulate slowly decreasing traffic, bringing traffic back to normal
				for i := 0; i < test.count; i++ {
					remove()
					if !empty() {
						remove()
					}
					add(GetTestValue(i))
				}

				// Simulate quick traffic spike (DDOS attack, etc)
				for i := 0; i < test.count; i++ {
					add(GetTestValue(i))
				}

				// Simulate stable traffic while at high traffic
				for i := 0; i < test.count; i++ {
					add(GetTestValue(i))
					remove()
				}

				// Simulate going back to normal (DDOS attack fended off)
				for i := 0; i < test.count; i++ {
					remove()
				}

				// Simulate stable traffic (now that is back to normal)
				for i := 0; i < test.count; i++ {
					add(GetTestValue(i))
					remove()
				}
			}
		})
	}
}

// MicroserviceTestObject tests the data structures performance by simulating the data structure being used by microservice
// and serverless systems when running in production environments.
// MicroserviceTestObject is a copy of Microservice that operates on *TestValue object which allows data structures that suport
// generics to not need to perform any type cast in the benchmark tests.
func (t *Tests) MicroserviceTestObject(b *testing.B, initInstance func(), add func(v *TestValue), remove func() (*TestValue, bool), empty func() bool) {
	for _, test := range tests {
		b.Run(strconv.Itoa(test.count), func(b *testing.B) {
			for n := 0; n < b.N; n++ {
				initInstance()

				// Simulate stable traffic
				for i := 0; i < test.count; i++ {
					add(GetTestValue(i))
					remove()
				}

				// Simulate slowly increasing traffic
				for i := 0; i < test.count; i++ {
					add(GetTestValue(i))
					add(GetTestValue(i))
					remove()
				}

				// Simulate slowly decreasing traffic, bringing traffic back to normal
				for i := 0; i < test.count; i++ {
					remove()
					if !empty() {
						remove()
					}
					add(GetTestValue(i))
				}

				// Simulate quick traffic spike (DDOS attack, etc)
				for i := 0; i < test.count; i++ {
					add(GetTestValue(i))
				}

				// Simulate stable traffic while at high traffic
				for i := 0; i < test.count; i++ {
					add(GetTestValue(i))
					remove()
				}

				// Simulate going back to normal (DDOS attack fended off)
				for i := 0; i < test.count; i++ {
					remove()
				}

				// Simulate stable traffic (now that is back to normal)
				for i := 0; i < test.count; i++ {
					add(GetTestValue(i))
					remove()
				}
			}
		})
	}
}