287 lines
10 KiB
Go
287 lines
10 KiB
Go
|
package storage_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/go-flucky/flucky/pkg/storage"
|
||
|
|
"github.com/go-flucky/flucky/test/goldenfiles"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
type testCase struct {
|
||
|
|
source string
|
||
|
|
expected string
|
||
|
|
compression bool
|
||
|
|
round float64
|
||
|
|
}
|
||
|
|
|
||
|
|
var (
|
||
|
|
testCases = []*testCase{
|
||
|
|
// CSV - Uncompressed Not Rounded -> Uncompressed Rounded 0.5
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/csv/goldenMeasuredValuesUncompressedNotRounded.csv",
|
||
|
|
expected: "test/goldenfiles/csv/goldenMeasuredValuesUncompressedRounded.csv",
|
||
|
|
compression: false,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/csv/goldenHumiditiesUncompressedNotRounded.csv",
|
||
|
|
expected: "test/goldenfiles/csv/goldenHumiditiesUncompressedRounded.csv",
|
||
|
|
compression: false,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/csv/goldenPressuresUncompressedNotRounded.csv",
|
||
|
|
expected: "test/goldenfiles/csv/goldenPressuresUncompressedRounded.csv",
|
||
|
|
compression: false,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/csv/goldenTemperaturesUncompressedNotRounded.csv",
|
||
|
|
expected: "test/goldenfiles/csv/goldenTemperaturesUncompressedRounded.csv",
|
||
|
|
compression: false,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
|
||
|
|
// CSV - Uncompressed Not Rounded -> Compressed Not Rounded
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/csv/goldenMeasuredValuesUncompressedNotRounded.csv",
|
||
|
|
expected: "test/goldenfiles/csv/goldenMeasuredValuesCompressedNotRounded.csv",
|
||
|
|
compression: true,
|
||
|
|
round: 0,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/csv/goldenHumiditiesUncompressedNotRounded.csv",
|
||
|
|
expected: "test/goldenfiles/csv/goldenHumiditiesCompressedNotRounded.csv",
|
||
|
|
compression: true,
|
||
|
|
round: 0,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/csv/goldenPressuresUncompressedNotRounded.csv",
|
||
|
|
expected: "test/goldenfiles/csv/goldenPressuresCompressedNotRounded.csv",
|
||
|
|
compression: true,
|
||
|
|
round: 0,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/csv/goldenTemperaturesUncompressedNotRounded.csv",
|
||
|
|
expected: "test/goldenfiles/csv/goldenTemperaturesCompressedNotRounded.csv",
|
||
|
|
compression: true,
|
||
|
|
round: 0,
|
||
|
|
},
|
||
|
|
|
||
|
|
// CSV - Uncompressed Not Rounded -> Compressed Rounded 0.5
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/csv/goldenMeasuredValuesUncompressedNotRounded.csv",
|
||
|
|
expected: "test/goldenfiles/csv/goldenMeasuredValuesCompressedRounded.csv",
|
||
|
|
compression: true,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/csv/goldenHumiditiesUncompressedNotRounded.csv",
|
||
|
|
expected: "test/goldenfiles/csv/goldenHumiditiesCompressedRounded.csv",
|
||
|
|
compression: true,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/csv/goldenPressuresUncompressedNotRounded.csv",
|
||
|
|
expected: "test/goldenfiles/csv/goldenPressuresCompressedRounded.csv",
|
||
|
|
compression: true,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/csv/goldenTemperaturesUncompressedNotRounded.csv",
|
||
|
|
expected: "test/goldenfiles/csv/goldenTemperaturesCompressedRounded.csv",
|
||
|
|
compression: true,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
|
||
|
|
// JSON - Uncompressed Not Rounded -> Uncompressed Rounded 0.5
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/json/goldenMeasuredValuesUncompressedNotRounded.json",
|
||
|
|
expected: "test/goldenfiles/json/goldenMeasuredValuesUncompressedRounded.json",
|
||
|
|
compression: false,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/json/goldenHumiditiesUncompressedNotRounded.json",
|
||
|
|
expected: "test/goldenfiles/json/goldenHumiditiesUncompressedRounded.json",
|
||
|
|
compression: false,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/json/goldenPressuresUncompressedNotRounded.json",
|
||
|
|
expected: "test/goldenfiles/json/goldenPressuresUncompressedRounded.json",
|
||
|
|
compression: false,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/json/goldenTemperaturesUncompressedNotRounded.json",
|
||
|
|
expected: "test/goldenfiles/json/goldenTemperaturesUncompressedRounded.json",
|
||
|
|
compression: false,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
|
||
|
|
// JSON - Uncompressed Not Rounded -> Compressed Not Rounded
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/json/goldenMeasuredValuesUncompressedNotRounded.json",
|
||
|
|
expected: "test/goldenfiles/json/goldenMeasuredValuesCompressedNotRounded.json",
|
||
|
|
compression: true,
|
||
|
|
round: 0,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/json/goldenHumiditiesUncompressedNotRounded.json",
|
||
|
|
expected: "test/goldenfiles/json/goldenHumiditiesCompressedNotRounded.json",
|
||
|
|
compression: true,
|
||
|
|
round: 0,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/json/goldenPressuresUncompressedNotRounded.json",
|
||
|
|
expected: "test/goldenfiles/json/goldenPressuresCompressedNotRounded.json",
|
||
|
|
compression: true,
|
||
|
|
round: 0,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/json/goldenTemperaturesUncompressedNotRounded.json",
|
||
|
|
expected: "test/goldenfiles/json/goldenTemperaturesCompressedNotRounded.json",
|
||
|
|
compression: true,
|
||
|
|
round: 0,
|
||
|
|
},
|
||
|
|
|
||
|
|
// JSON - Uncompressed Not Rounded -> Compressed Rounded 0.5
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/json/goldenMeasuredValuesUncompressedNotRounded.json",
|
||
|
|
expected: "test/goldenfiles/json/goldenMeasuredValuesCompressedRounded.json",
|
||
|
|
compression: true,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/json/goldenHumiditiesUncompressedNotRounded.json",
|
||
|
|
expected: "test/goldenfiles/json/goldenHumiditiesCompressedRounded.json",
|
||
|
|
compression: true,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/json/goldenPressuresUncompressedNotRounded.json",
|
||
|
|
expected: "test/goldenfiles/json/goldenPressuresCompressedRounded.json",
|
||
|
|
compression: true,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/json/goldenTemperaturesUncompressedNotRounded.json",
|
||
|
|
expected: "test/goldenfiles/json/goldenTemperaturesCompressedRounded.json",
|
||
|
|
compression: true,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
|
||
|
|
// XML - Uncompressed Not Rounded -> Uncompressed Rounded 0.5
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/xml/goldenMeasuredValuesUncompressedNotRounded.xml",
|
||
|
|
expected: "test/goldenfiles/xml/goldenMeasuredValuesUncompressedRounded.xml",
|
||
|
|
compression: false,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/xml/goldenHumiditiesUncompressedNotRounded.xml",
|
||
|
|
expected: "test/goldenfiles/xml/goldenHumiditiesUncompressedRounded.xml",
|
||
|
|
compression: false,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/xml/goldenPressuresUncompressedNotRounded.xml",
|
||
|
|
expected: "test/goldenfiles/xml/goldenPressuresUncompressedRounded.xml",
|
||
|
|
compression: false,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/xml/goldenTemperaturesUncompressedNotRounded.xml",
|
||
|
|
expected: "test/goldenfiles/xml/goldenTemperaturesUncompressedRounded.xml",
|
||
|
|
compression: false,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
|
||
|
|
// XML - Uncompressed Not Rounded -> Compressed Not Rounded
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/xml/goldenMeasuredValuesUncompressedNotRounded.xml",
|
||
|
|
expected: "test/goldenfiles/xml/goldenMeasuredValuesCompressedNotRounded.xml",
|
||
|
|
compression: true,
|
||
|
|
round: 0,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/xml/goldenHumiditiesUncompressedNotRounded.xml",
|
||
|
|
expected: "test/goldenfiles/xml/goldenHumiditiesCompressedNotRounded.xml",
|
||
|
|
compression: true,
|
||
|
|
round: 0,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/xml/goldenPressuresUncompressedNotRounded.xml",
|
||
|
|
expected: "test/goldenfiles/xml/goldenPressuresCompressedNotRounded.xml",
|
||
|
|
compression: true,
|
||
|
|
round: 0,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/xml/goldenTemperaturesUncompressedNotRounded.xml",
|
||
|
|
expected: "test/goldenfiles/xml/goldenTemperaturesCompressedNotRounded.xml",
|
||
|
|
compression: true,
|
||
|
|
round: 0,
|
||
|
|
},
|
||
|
|
|
||
|
|
// XML - Uncompressed Not Rounded -> Compressed Rounded 0.5
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/xml/goldenMeasuredValuesUncompressedNotRounded.xml",
|
||
|
|
expected: "test/goldenfiles/xml/goldenMeasuredValuesCompressedRounded.xml",
|
||
|
|
compression: true,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/xml/goldenHumiditiesUncompressedNotRounded.xml",
|
||
|
|
expected: "test/goldenfiles/xml/goldenHumiditiesCompressedRounded.xml",
|
||
|
|
compression: true,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/xml/goldenPressuresUncompressedNotRounded.xml",
|
||
|
|
expected: "test/goldenfiles/xml/goldenPressuresCompressedRounded.xml",
|
||
|
|
compression: true,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
&testCase{
|
||
|
|
source: "test/goldenfiles/xml/goldenTemperaturesUncompressedNotRounded.xml",
|
||
|
|
expected: "test/goldenfiles/xml/goldenTemperaturesCompressedRounded.xml",
|
||
|
|
compression: true,
|
||
|
|
round: 0.5,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestCompressionAndRounding(t *testing.T) {
|
||
|
|
require := require.New(t)
|
||
|
|
|
||
|
|
for _, testCase := range testCases {
|
||
|
|
|
||
|
|
measuredValues, err := goldenfiles.GetGoldenMeasuredValues(testCase.source)
|
||
|
|
require.NoError(err)
|
||
|
|
|
||
|
|
expectedMeasuredValues, err := goldenfiles.GetGoldenMeasuredValues(testCase.expected)
|
||
|
|
require.NoError(err)
|
||
|
|
|
||
|
|
actualMeasuredValues := measuredValues
|
||
|
|
|
||
|
|
if testCase.round != 0 {
|
||
|
|
storage.Round(actualMeasuredValues, testCase.round)
|
||
|
|
}
|
||
|
|
|
||
|
|
if testCase.compression {
|
||
|
|
actualMeasuredValues = storage.Compression(actualMeasuredValues)
|
||
|
|
}
|
||
|
|
|
||
|
|
for i, _ := range expectedMeasuredValues {
|
||
|
|
require.Equal(expectedMeasuredValues[i].ID, actualMeasuredValues[i].ID, "ID of element %v is not equal between expected and actual", i)
|
||
|
|
require.Equal(expectedMeasuredValues[i].Value, actualMeasuredValues[i].Value, "Value of element %v is not equal between expected and actual", i)
|
||
|
|
require.Equal(expectedMeasuredValues[i].ValueType, actualMeasuredValues[i].ValueType, "ValueType of element %v is not equal between expected and actual", i)
|
||
|
|
require.Equal(expectedMeasuredValues[i].FromDate, actualMeasuredValues[i].FromDate, "FromDate of element %v is not equal between expected and actual", i)
|
||
|
|
require.Equal(expectedMeasuredValues[i].TillDate, actualMeasuredValues[i].TillDate, "TillDate of element %v is not equal between expected and actual", i)
|
||
|
|
require.Equal(expectedMeasuredValues[i].SensorID, actualMeasuredValues[i].SensorID, "SensorID of element %v is not equal between expected and actual", i)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|