MCPcopy Create free account
hub / github.com/cortexproject/cortex / TestCompileQueryPriorityRegex

Function TestCompileQueryPriorityRegex

pkg/util/validation/limits_test.go:846–916  ·  view source on GitHub ↗
(t *testing.T)

Source from the content-addressed store, hash-verified

844}
845
846func TestCompileQueryPriorityRegex(t *testing.T) {
847 l := Limits{
848 QueryPriority: QueryPriority{
849 Enabled: true,
850 Priorities: []PriorityDef{
851 {
852 Priority: 1,
853 QueryAttributes: []QueryAttribute{
854 {
855 Regex: "test",
856 },
857 },
858 },
859 },
860 },
861 QueryRejection: QueryRejection{
862 Enabled: false,
863 QueryAttributes: []QueryAttribute{
864 {
865 Regex: "testRejection",
866 },
867 },
868 },
869 }
870
871 require.Nil(t, l.QueryPriority.Priorities[0].QueryAttributes[0].CompiledRegex)
872
873 err := l.compileQueryAttributeRegex()
874 require.NoError(t, err)
875 require.Equal(t, regexp.MustCompile("test"), l.QueryPriority.Priorities[0].QueryAttributes[0].CompiledRegex)
876
877 l.QueryPriority.Priorities[0].QueryAttributes[0].Regex = "new"
878
879 err = l.compileQueryAttributeRegex()
880 require.NoError(t, err)
881 require.Equal(t, regexp.MustCompile("new"), l.QueryPriority.Priorities[0].QueryAttributes[0].CompiledRegex)
882
883 l.QueryPriority.Priorities[0].QueryAttributes[0].CompiledRegex = nil
884
885 err = l.compileQueryAttributeRegex()
886 require.NoError(t, err)
887 require.Equal(t, regexp.MustCompile("new"), l.QueryPriority.Priorities[0].QueryAttributes[0].CompiledRegex)
888
889 l.QueryPriority.Enabled = false
890 l.QueryPriority.Priorities[0].QueryAttributes[0].CompiledRegex = nil
891
892 err = l.compileQueryAttributeRegex()
893 require.NoError(t, err)
894 require.Nil(t, l.QueryPriority.Priorities[0].QueryAttributes[0].CompiledRegex)
895
896 require.Nil(t, l.QueryRejection.QueryAttributes[0].CompiledRegex)
897
898 l.QueryRejection.Enabled = true
899
900 err = l.compileQueryAttributeRegex()
901 require.NoError(t, err)
902 require.Equal(t, regexp.MustCompile("testRejection"), l.QueryRejection.QueryAttributes[0].CompiledRegex)
903 require.Equal(t, regexp.MustCompile(""), l.QueryPriority.Priorities[0].QueryAttributes[0].CompiledUserAgentRegex)

Callers

nothing calls this directly

Calls 2

EqualMethod · 0.65

Tested by

no test coverage detected