blob: 1c3b47182a726afbfb1890c5119144bad1bcf8c9 [file] [log] [blame]
Jonathan Hart4b110f62020-03-13 17:36:19 -07001package assert
2
3import (
4 "fmt"
5 "reflect"
6)
7
David K. Bainbridgee05cf0c2021-08-19 03:16:50 +00008// isOrdered checks that collection contains orderable elements.
9func isOrdered(t TestingT, object interface{}, allowedComparesResults []CompareType, failMessage string, msgAndArgs ...interface{}) bool {
10 objKind := reflect.TypeOf(object).Kind()
11 if objKind != reflect.Slice && objKind != reflect.Array {
12 return false
Jonathan Hart4b110f62020-03-13 17:36:19 -070013 }
14
David K. Bainbridgee05cf0c2021-08-19 03:16:50 +000015 objValue := reflect.ValueOf(object)
16 objLen := objValue.Len()
Jonathan Hart4b110f62020-03-13 17:36:19 -070017
David K. Bainbridgee05cf0c2021-08-19 03:16:50 +000018 if objLen <= 1 {
19 return true
Jonathan Hart4b110f62020-03-13 17:36:19 -070020 }
21
David K. Bainbridgee05cf0c2021-08-19 03:16:50 +000022 value := objValue.Index(0)
23 valueInterface := value.Interface()
24 firstValueKind := value.Kind()
Jonathan Hart4b110f62020-03-13 17:36:19 -070025
David K. Bainbridgee05cf0c2021-08-19 03:16:50 +000026 for i := 1; i < objLen; i++ {
27 prevValue := value
28 prevValueInterface := valueInterface
Jonathan Hart4b110f62020-03-13 17:36:19 -070029
David K. Bainbridgee05cf0c2021-08-19 03:16:50 +000030 value = objValue.Index(i)
31 valueInterface = value.Interface()
32
33 compareResult, isComparable := compare(prevValueInterface, valueInterface, firstValueKind)
34
35 if !isComparable {
36 return Fail(t, fmt.Sprintf("Can not compare type \"%s\" and \"%s\"", reflect.TypeOf(value), reflect.TypeOf(prevValue)), msgAndArgs...)
37 }
38
39 if !containsValue(allowedComparesResults, compareResult) {
40 return Fail(t, fmt.Sprintf(failMessage, prevValue, value), msgAndArgs...)
41 }
Jonathan Hart4b110f62020-03-13 17:36:19 -070042 }
43
44 return true
45}
46
David K. Bainbridgee05cf0c2021-08-19 03:16:50 +000047// IsIncreasing asserts that the collection is increasing
Jonathan Hart4b110f62020-03-13 17:36:19 -070048//
David K. Bainbridgee05cf0c2021-08-19 03:16:50 +000049// assert.IsIncreasing(t, []int{1, 2, 3})
50// assert.IsIncreasing(t, []float{1, 2})
51// assert.IsIncreasing(t, []string{"a", "b"})
52func IsIncreasing(t TestingT, object interface{}, msgAndArgs ...interface{}) bool {
53 return isOrdered(t, object, []CompareType{compareLess}, "\"%v\" is not less than \"%v\"", msgAndArgs)
Jonathan Hart4b110f62020-03-13 17:36:19 -070054}
55
David K. Bainbridgee05cf0c2021-08-19 03:16:50 +000056// IsNonIncreasing asserts that the collection is not increasing
Jonathan Hart4b110f62020-03-13 17:36:19 -070057//
David K. Bainbridgee05cf0c2021-08-19 03:16:50 +000058// assert.IsNonIncreasing(t, []int{2, 1, 1})
59// assert.IsNonIncreasing(t, []float{2, 1})
60// assert.IsNonIncreasing(t, []string{"b", "a"})
61func IsNonIncreasing(t TestingT, object interface{}, msgAndArgs ...interface{}) bool {
62 return isOrdered(t, object, []CompareType{compareEqual, compareGreater}, "\"%v\" is not greater than or equal to \"%v\"", msgAndArgs)
Jonathan Hart4b110f62020-03-13 17:36:19 -070063}
64
David K. Bainbridgee05cf0c2021-08-19 03:16:50 +000065// IsDecreasing asserts that the collection is decreasing
Jonathan Hart4b110f62020-03-13 17:36:19 -070066//
David K. Bainbridgee05cf0c2021-08-19 03:16:50 +000067// assert.IsDecreasing(t, []int{2, 1, 0})
68// assert.IsDecreasing(t, []float{2, 1})
69// assert.IsDecreasing(t, []string{"b", "a"})
70func IsDecreasing(t TestingT, object interface{}, msgAndArgs ...interface{}) bool {
71 return isOrdered(t, object, []CompareType{compareGreater}, "\"%v\" is not greater than \"%v\"", msgAndArgs)
72}
Jonathan Hart4b110f62020-03-13 17:36:19 -070073
David K. Bainbridgee05cf0c2021-08-19 03:16:50 +000074// IsNonDecreasing asserts that the collection is not decreasing
75//
76// assert.IsNonDecreasing(t, []int{1, 1, 2})
77// assert.IsNonDecreasing(t, []float{1, 2})
78// assert.IsNonDecreasing(t, []string{"a", "b"})
79func IsNonDecreasing(t TestingT, object interface{}, msgAndArgs ...interface{}) bool {
80 return isOrdered(t, object, []CompareType{compareLess, compareEqual}, "\"%v\" is not less than or equal to \"%v\"", msgAndArgs)
Jonathan Hart4b110f62020-03-13 17:36:19 -070081}