[fin] matchingTest

src/broker/modules/matching_test.go

@@ -1,37 +1,169 @@
 package modules
-//
-//import (
-//	"fmt"
-//	"testing"
-//)
-//
-//func Test_simpleLList_pbNtraverse(t *testing.T) {
-//	mylist := new(LList)
-//	mylist.PushBack(30)
-//	mylist.PushBack(40)
-//	mylist.PushBack(20)
-//	mylist.PushBack(10)
-//	ptr := mylist.head
-//	for ptr != nil {
-//		fmt.Printf("data = %d, address = %p\n", ptr.val, ptr.next)
-//		ptr = ptr.next
-//	}
-//}
-//
-//func Test_simpleLList_mergeSort(t *testing.T) {
-//	mylist := new(LList)
-//	mylist.PushBack(30)
-//	mylist.PushBack(40)
-//	mylist.PushBack(20)
-//	mylist.PushBack(10)
-//	mylist.head = MergeSort(mylist.head, mylist.size / 2)
-//	ptr := mylist.head
-//	for ptr != nil {
-//		fmt.Printf("data = %d, address = %p\n", ptr.val, ptr.next)
-//		ptr = ptr.next
-//	}
-//}
-//
-//func Test_removeArray(t *testing.T){}
-//func Test_deleteList(t *testing.T){}
-//func Test_insertList(t *testing.T){}
+
+import (
+	"fmt"
+	"github.com/stretchr/testify/assert"
+	"math/rand"
+	"strconv"
+	"testing"
+	"time"
+)
+
+
+func makePubData(isAlpha bool) MsgUnit{
+	rand.Seed(time.Now().UnixNano())
+	// Set Ipaddr
+	msg := Message{"", "1.0", "", PM}
+	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)
+
+	topic := []int64{}
+	value := []int64{}
+	content := []int64{}
+
+	// Set Topic
+	topicLen := rand.Int() % 10 + 1
+	for i := 0 ; i < topicLen; i++{
+		topic = append(topic, rand.Int63())
+	}
+
+	// Set Value
+	if isAlpha {
+		valueLen := rand.Int() % 10 + 1
+		for i := 0 ; i < valueLen; i++{
+			value = append(value, rand.Int63())
+		}
+	} else {
+		topic = append(topic, rand.Int63())
+		value = append(value, rand.Int63())
+	}
+
+	// Set content
+	contentLen := rand.Int() % 10
+	for i := 0; i < contentLen; i++{
+		content = append(content, rand.Int63())
+	}
+
+	return &PublishMsg{msg, topic, value, content}
+}
+
+func makeSubData(isAlpha bool, topic []int64, value []int64) 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
+	operator := []string{}
+	candOp := []string{">", ">=", "<=" ,"<", "=="}
+	logicalOp := []string{"&&", "||"}
+
+	if isAlpha {
+		operator = append(operator, "==")
+	} else {
+		randSeed := rand.Int() % 2
+		if randSeed == 0 {
+			lop := rand.Int() % 2
+			operator = append(operator, candOp[rand.Int()%2])
+			if lop == 0 {
+				operator = append(operator, logicalOp[0])
+				for {
+					x := rand.Int63()
+					if x > value[0] {
+						value = append(value, x)
+						break
+					}
+				}
+			} else {
+				operator = append(operator, logicalOp[1])
+				value = append(value, rand.Int63())
+			}
+			operator = append(operator, candOp[rand.Int()%2+2])
+		} else {
+			op := candOp[rand.Int()%5]
+			operator = append(operator, op)
+		}
+	}
+
+	return &SubscriptionMsg{msg, topic, value, operator, isAlpha}
+}
+
+func Test_matching(t *testing.T) {
+	rand.Seed(time.Now().UnixNano())
+	assert.Equal(t, 1, 1)
+	mos := Moscato{ sub_mng: *newSubmng(),}
+	mos.queue.queue_init()
+
+	// 1. Make Publish Data
+	pubLen := rand.Int()%100 + 1 // Set dataLength
+	var pubDataList []MsgUnit
+  
+	for i := 0; i < pubLen; i++{
+		if i % 2 == 1 {
+			msg := makePubData(true)
+			mos.queue.push(msg)
+			pubDataList = append(pubDataList, msg)
+		} else{
+			msg := makePubData(false)
+			mos.queue.push(msg)
+			pubDataList = append(pubDataList, msg)
+		}
+	}
+
+	// 2. Creates subscription data with (the same <topic, value> or (the same <topic, difValue>
+	//    And Add Subscription
+	subLen := pubLen
+	var subDataList []MsgUnit
+	for i := 0; i < subLen; i++{
+		topic := pubDataList[i].(*PublishMsg).topic
+		value := pubDataList[i].(*PublishMsg).value
+		if i % 2 == 1{
+			msg := makeSubData(true, topic, value)
+			mos.sub_mng.addSubscription(msg)
+			subDataList = append(subDataList, msg)
+
+		} else{
+			msg := makeSubData(false, topic, value)
+			mos.sub_mng.addSubscription(msg)
+			subDataList = append(subDataList, msg)
+		}
+	}
+
+	// 3. Watch Data
+	
+  	// fmt.Println("PubData")
+	// watchData(pubDataList, pubLen, false)
+	// fmt.Println("\nSubData")
+	// watchData(subDataList, subLen, true)
+
+	// 4. Matching
+	for i := 0; i < pubLen; i++ {
+		matching, pubmsg, err := mos.match_mng.Matching(&mos.queue, &mos.sub_mng)
+
+		// Check if matching is correct
+		assert.Equal(t, nil, err)
+
+		fmt.Println("matching list = ", matching)
+		fmt.Println("pub msg = ", pubmsg)
+		fmt.Println("err ?= ", err)
+	}
+}