osel/processing_test.go

49 lines
2.6 KiB
Go

package main
import (
"fmt"
"os"
"testing"
"github.com/streadway/amqp"
"github.com/stretchr/testify/assert"
)
func TestProcessWaitingEvent(t *testing.T) {
var delivery amqp.Delivery
openstackActions := connectFakeOpenstack()
delivery.Body = []byte(securityGroupRuleCreateWithIcmpAndCider)
event, err := processWaitingEvent(delivery, openstackActions)
if err != nil {
t.Fatal(err)
}
_ = event
}
func TestLogEvents(t *testing.T) {
hostName, _ := os.Hostname()
IPList := []string{"10.0.0.1", "10.0.0.3"}
logLines := []string{fmt.Sprintf(`{"security_group_rule":{"remote_group_id":null,"direction":"ingress","protocol":"icmp","remote_ip_prefix":"192.168.1.0/24","port_range_max":null,"rule_direction":"","security_group_id":"46d46540-98ac-4c93-ae62-68dddab2282e","tenant_id":"ada3b9b0dbac429f9361e803b54f5f32","port_range_min":null,"ethertype":"IPv4","id":"66d7ac79-3551-4436-83c7-103b50760cfb"},"affected_ip_address":"10.0.0.1","change_type":"sg_rule_add","source_type":"osel","source_message_bus":"%s"}`, hostName), fmt.Sprintf(`{"security_group_rule":{"remote_group_id":null,"direction":"ingress","protocol":"icmp","remote_ip_prefix":"192.168.1.0/24","port_range_max":null,"rule_direction":"","security_group_id":"46d46540-98ac-4c93-ae62-68dddab2282e","tenant_id":"ada3b9b0dbac429f9361e803b54f5f32","port_range_min":null,"ethertype":"IPv4","id":"66d7ac79-3551-4436-83c7-103b50760cfb"},"affected_ip_address":"10.0.0.3","change_type":"sg_rule_add","source_type":"osel","source_message_bus":"%s"}`, hostName)}
logger := connectFakeSyslog()
qualys := connectFakeQualys()
IPs := make(map[string][]string)
IPs["46d46540-98ac-4c93-ae62-68dddab2282e"] = IPList
fakeEvent := Event{
RawData: []byte(securityGroupRuleCreateWithIcmpAndCider),
LogLines: logLines,
Processor: EventSecurityGroupRuleChange{ChangeType: "sg_rule_add"},
IPs: IPs,
}
events := []Event{fakeEvent}
logEvents(events, logger, qualys)
savedLogs := logger.GetLogs()
assert.Equal(t, 2, len(savedLogs))
logLine1 := fmt.Sprintf(`{"affected_ip_address":"10.0.0.1","change_type":"sg_rule_add","qualys_scan_id":"","qualys_scan_error":"Not scanned by Qualys","security_group_rule":{"remote_group_id":null,"direction":"ingress","protocol":"icmp","remote_ip_prefix":"192.168.1.0/24","port_range_max":null,"rule_direction":"","security_group_id":"46d46540-98ac-4c93-ae62-68dddab2282e","tenant_id":"ada3b9b0dbac429f9361e803b54f5f32","port_range_min":null,"ethertype":"IPv4","id":"66d7ac79-3551-4436-83c7-103b50760cfb"},"source_type":"osel1.1","source_message_bus":"%s"}`, hostName)
assert.Equal(t, logLine1, savedLogs[0])
}