Golang Must Examples

Example #1

File: bytesize_test.go Project: jinhy/rpc_proxy

func TestBytesizeError(t *testing.T) {
	var err error
	_, err = Parse("--1")
	assert.Must(errors.Equal(err, ErrBadBytesize))
	_, err = Parse("hello world")
	assert.Must(errors.Equal(err, ErrBadBytesize))
	_, err = Parse("123.132.32")
	assert.Must(errors.Equal(err, ErrBadBytesize))
}

Example #2

File: priority_queue_test.go Project: vinsia/rpc_proxy

func TestPriorityQueue(t *testing.T) {
	pq := make(PriorityQueue, 0)
	heap.Push(&pq, &Worker{Identity: "hello", priority: 10})
	heap.Push(&pq, &Worker{Identity: "h20", priority: 20})
	topItem := heap.Pop(&pq)

	t.Log("TopItem: ", topItem.(*Worker).Identity, topItem.(*Worker).priority)
	assert.Must(topItem.(*Worker).Identity == "h20")
	//	assert.Must(topItem.(*Item).value == "hello")

	topItem = heap.Pop(&pq)
	t.Log("TopItem: ", topItem.(*Worker).Identity, topItem.(*Worker).priority)

	assert.Must(true)

}

Example #3

File: topology_test.go Project: vinsia/rpc_proxy

func TestZooKeeper(t *testing.T) {

	ifaces, _ := net.Interfaces()
	// handle err
	for _, i := range ifaces {
		addrs, _ := i.Addrs()
		// handle err
		for _, addr := range addrs {
			var ip net.IP
			switch v := addr.(type) {
			case *net.IPNet:
				ip = v.IP
			case *net.IPAddr:
				ip = v.IP
			}
			fmt.Println("IP: ", ip)
			// process IP address
		}
	}

	// 创建一个Topology
	top := NewTopology("online_service", "127.0.0.1:2181")

	testPath := "/hello"
	testPath = top.FullPath(testPath)
	top.DeleteDir(testPath)
	path, err := top.CreateDir(testPath)
	fmt.Println("Full Path: ", path)
	fmt.Println("Error: ", err)
	assert.Must(true)

	var proxyInfo map[string]interface{} = make(map[string]interface{})

	proxyInfo["rpc_front"] = "tcp://127.0.0.1:5550"

	fmt.Println("ProxyInfo: ", proxyInfo)

	top.SetRpcProxyData(proxyInfo)
	data, err := top.GetRpcProxyData()
	fmt.Println("Data: ", data, ", error: ", err)

	//	var endpointInfo map[string]interface{} = make(map[string]interface{})

	//	endpointInfo["endpoint"] = "tcp://127.0.0.1:5555"
	//	top.AddServiceEndPoint("account", "server001", endpointInfo)
	top.DeleteServiceEndPoint("account", "server001")

	//	top.SetPathData(testPath, []byte("hello"))
	//	var wait chan int32 = make(chan int32, 4)

	//	go func() {
	//		var evtbus chan interface{} = make(chan interface{})
	//		var pathes []string
	//		for i := 0; i < 10; i++ {
	//			pathes, _ = top.WatchChildren(testPath, evtbus)
	//			fmt.Println("pathes: ", pathes)
	//			<-evtbus
	//		}

	//		wait <- 1

	//		pathes, _ = top.WatchChildren(testPath, evtbus)
	//		fmt.Println("pathes: ", pathes)
	//		wait <- 2
	//	}()

	//	go func() {
	//		for i := 0; i < 10; i++ {
	//			path := fmt.Sprintf("%s/node_%d", testPath, i)
	//			top.CreateDir(path)
	//		}

	//		wait <- 100
	//	}()

	//	go func() {
	//		var evtbus chan interface{} = make(chan interface{})
	//		content, _ := top.WatchNode(testPath, evtbus)
	//		fmt.Println("--------Content: ", string(content))
	//		fmt.Println("--------EvtBus: ", <-evtbus)

	//		content, _ = top.WatchNode(testPath, evtbus)
	//		fmt.Println("--------Content: ", string(content))
	//		wait <- 1234
	//	}()

	//	go func() {
	//		top.SetPathData(testPath, []byte("world"))
	//		wait <- 5678

	//		time.Sleep(100 * time.Millisecond)

	//		top.SetPathData(testPath, []byte("world"))
	//		wait <- 0

	//	}()

	//	for i := 0; i < 3; i++ {
	//		fmt.Println("Waiting: ", <-wait)
	//	}
}

Example #4

File: bytesize_test.go Project: jinhy/rpc_proxy

func TestBytesize(t *testing.T) {
	assert.Must(MustParse("1") == 1)
	assert.Must(MustParse("1B") == 1)
	assert.Must(MustParse("1K") == KB)
	assert.Must(MustParse("1M") == MB)
	assert.Must(MustParse("1G") == GB)
	assert.Must(MustParse("1T") == TB)
	assert.Must(MustParse("1P") == PB)

	assert.Must(MustParse(" -1") == -1)
	assert.Must(MustParse(" -1 b") == -1)
	assert.Must(MustParse(" -1 kb ") == -1*KB)
	assert.Must(MustParse(" -1 mb ") == -1*MB)
	assert.Must(MustParse(" -1 gb ") == -1*GB)
	assert.Must(MustParse(" -1 tb ") == -1*TB)
	assert.Must(MustParse(" -1 pb ") == -1*PB)

	assert.Must(MustParse(" 1.5") == 1)
	assert.Must(MustParse(" 1.5 kb ") == 1.5*KB)
	assert.Must(MustParse(" 1.5 mb ") == 1.5*MB)
	assert.Must(MustParse(" 1.5 gb ") == 1.5*GB)
	assert.Must(MustParse(" 1.5 tb ") == 1.5*TB)
	assert.Must(MustParse(" 1.5 pb ") == 1.5*PB)
}