143 lines
3.4 KiB
Go
143 lines
3.4 KiB
Go
// Licensed to Elasticsearch B.V. under one or more contributor
|
|
// license agreements. See the NOTICE file distributed with
|
|
// this work for additional information regarding copyright
|
|
// ownership. Elasticsearch B.V. licenses this file to you under
|
|
// the Apache License, Version 2.0 (the "License"); you may
|
|
// not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing,
|
|
// software distributed under the License is distributed on an
|
|
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
// KIND, either express or implied. See the License for the
|
|
// specific language governing permissions and limitations
|
|
// under the License.
|
|
|
|
package conditions
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/elastic/beats/libbeat/beat"
|
|
"github.com/elastic/beats/libbeat/common"
|
|
"github.com/elastic/beats/libbeat/logp"
|
|
)
|
|
|
|
func TestCreateNilCondition(t *testing.T) {
|
|
cond, err := NewCondition(nil)
|
|
assert.Nil(t, cond)
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func GetCondition(t *testing.T, config Config) Condition {
|
|
cond, err := NewCondition(&config)
|
|
assert.Nil(t, err)
|
|
return cond
|
|
}
|
|
|
|
func GetConditions(t *testing.T, configs []Config) []Condition {
|
|
conds := []Condition{}
|
|
|
|
for _, config := range configs {
|
|
conds = append(conds, GetCondition(t, config))
|
|
}
|
|
assert.True(t, len(conds) == len(configs))
|
|
|
|
return conds
|
|
}
|
|
|
|
var secdTestEvent = &beat.Event{
|
|
Timestamp: time.Now(),
|
|
Fields: common.MapStr{
|
|
"proc": common.MapStr{
|
|
"cmdline": "/usr/libexec/secd",
|
|
"cpu": common.MapStr{
|
|
"start_time": "Apr10",
|
|
"system": 1988,
|
|
"total": 6029,
|
|
"total_p": 0.08,
|
|
"user": 4041,
|
|
},
|
|
"name": "secd",
|
|
"pid": 305,
|
|
"ppid": 1,
|
|
"state": "running",
|
|
"username": "monica",
|
|
"keywords": []string{"foo", "bar"},
|
|
},
|
|
"type": "process",
|
|
"final": false,
|
|
},
|
|
}
|
|
|
|
var httpResponseTestEvent = &beat.Event{
|
|
Timestamp: time.Now(),
|
|
Fields: common.MapStr{
|
|
"@timestamp": "2015-06-11T09:51:23.642Z",
|
|
"bytes_in": 126,
|
|
"bytes_out": 28033,
|
|
"client_ip": "127.0.0.1",
|
|
"client_port": 42840,
|
|
"client_proc": "",
|
|
"client_server": "mar.local",
|
|
"http": common.MapStr{
|
|
"code": 200,
|
|
"content_length": 76985,
|
|
"phrase": "OK",
|
|
},
|
|
"ip": "127.0.0.1",
|
|
"method": "GET",
|
|
"params": "",
|
|
"path": "/jszip.min.js",
|
|
"port": 8000,
|
|
"proc": "",
|
|
"query": "GET /jszip.min.js",
|
|
"responsetime": 30,
|
|
"server": "mar.local",
|
|
"status": "OK",
|
|
"type": "http",
|
|
},
|
|
}
|
|
|
|
func testConfig(t *testing.T, expected bool, event *beat.Event, config *Config) {
|
|
cond, err := NewCondition(config)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, expected, cond.Check(event))
|
|
}
|
|
|
|
func TestCombinedCondition(t *testing.T) {
|
|
logp.TestingSetup()
|
|
config := Config{
|
|
OR: []Config{
|
|
{
|
|
Range: &Fields{fields: map[string]interface{}{
|
|
"http.code.gte": 100,
|
|
"http.code.lt": 300,
|
|
}},
|
|
},
|
|
{
|
|
AND: []Config{
|
|
{
|
|
Equals: &Fields{fields: map[string]interface{}{
|
|
"status": 200,
|
|
}},
|
|
},
|
|
{
|
|
Equals: &Fields{fields: map[string]interface{}{
|
|
"type": "http",
|
|
}},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
cond := GetCondition(t, config)
|
|
|
|
assert.True(t, cond.Check(httpResponseTestEvent))
|
|
}
|