[add] subscription_test

src/broker/modules/subscription_test.go

@@ -0,0 +1,145 @@
+package modules
+
+import (
+	"fmt"
+	_ "fmt"
+	"math/rand"
+	"strconv"
+	"testing"
+	"time"
+	"github.com/stretchr/testify/assert"
+)
+
+func dataMake(isalpha bool) MsgUnit{
+	rand.Seed(time.Now().UnixNano())
+
+	// Set Ipaddr
+	msg := Message{"", "1.0", "", SM}
+	for i := 0; i < 4; i++{
+		itoa := strconv.Itoa(rand.Int() % 256)
+
+		msg.from += itoa
+		if i != 3{
+			msg.from += "."
+		}
+	}
+
+	// Set Time
+	msg.time += strconv.Itoa(rand.Int()%24) + ":"
+	msg.time += strconv.Itoa(rand.Int()%60)
+
+	// Set Topic, Value, operator
+	topic := []int64{}
+	value := []int64{}
+	operator := []string{}
+	candOp := []string{">", ">=", "<=" ,"<", "=="}
+	logicalOp := []string{"&&", "||"}
+
+	topic = append(topic, rand.Int63())
+
+	if isalpha{
+		value = append(value, rand.Int63())
+		operator = append(operator, "==")
+	} else{
+		len := rand.Int() % 2 + 1
+		value = append(value, rand.Int63())
+
+		if len == 1{
+			operator = append(operator, candOp[rand.Int()%5])
+		} else{
+			value = append(value, rand.Int63())
+			op := rand.Int()%2
+			operator = append(operator, candOp[rand.Int()%2])
+			if op == 0 {
+				operator = append(operator, logicalOp[0])
+			} else{
+				operator = append(operator, logicalOp[1])
+			}
+			operator = append(operator, candOp[rand.Int()%2 + 2])
+		}
+	}
+
+	return &SubscriptionMsg{msg, topic, value, operator, isalpha}
+}
+
+func Test_addSubscription(t *testing.T) {
+	rand.Seed(time.Now().UnixNano())
+
+	// To Init sub_mng
+	mos := Moscato{sub_mng : sub_manager{}}
+
+	// Make Data set(Subscription)
+	msgList := []MsgUnit{}
+	dataLen:= 3
+	for i := 0 ; i < dataLen ; i++{
+		seed := rand.Int()%2
+		if seed == 0{
+			msgList = append(msgList, dataMake(false))
+		} else{
+			msgList = append(msgList, dataMake(true))
+		}
+	}
+
+	// Watch Data set
+	for i := 0; i < dataLen ; i++{
+		msg := msgList[i]
+		fmt.Println(
+			" from = ", msg.(*SubscriptionMsg).Message.from,
+			" time = ", msg.(*SubscriptionMsg).Message.time,
+			" topic = ", msg.(*SubscriptionMsg).topic,
+			" value = ", msg.(*SubscriptionMsg).value,
+			" operator = ", msg.(*SubscriptionMsg).operator,
+			" isAlpha ?= ", msg.(*SubscriptionMsg).isAlpha)
+	}
+
+	// Test addSubScription
+	for i := 0; i < dataLen ; i++{
+
+		msg := msgList[i]
+		ip := msg.(*SubscriptionMsg).Message.from
+		topic := msg.(*SubscriptionMsg).topic
+		value := msg.(*SubscriptionMsg).value
+		//operator := msg.(*SubscriptionMsg).operator
+		//isAlpha := msg.(*SubscriptionMsg).isAlpha
+		subnumber := mos.sub_mng.count_sub
+
+		mos.sub_mng.addSubscription(msg)
+
+
+		// 1. Check if ip mapping is correct
+		assert.Equal(t, subnumber, mos.sub_mng.ip2sub[ip][len(mos.sub_mng.ip2sub[ip]) - 1], "Ip mapping is failed")
+
+		// 2. Check topicNode
+		nameptr := mos.sub_mng.list.head
+		for nameptr != nil{
+			if nameptr.topic[0] == topic[0]{
+				break
+			}
+			nameptr = nameptr.next
+		}
+
+		for i := 0 ; i < len(topic); i++{
+			assert.Equal(t, topic[i], nameptr.topic[i], "topicNode Add is failed")
+		}
+
+		// 3. Check Value in ValueNode
+		valptr := nameptr.list.head
+		for valptr != nil{
+			for i:=0 ; i < len(value); i++{
+				find := true
+				if value[i] != valptr.val[i]{
+					find = false
+				}
+				if find{
+					break
+				}
+			}
+			valptr = valptr.next
+		}
+		fmt.Println("my value = ", valptr.val, " real value = ", value)
+
+		for i := 0; i < len(value); i++{
+			assert.Equal(t, value[i], valptr.val[i], "ValNode Add is failed")
+		}
+	}
+}