malz
/
ekuiper-new


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241
							package server

import (
	"fmt"
	"net/http"
	"net/http/httptest"
	"reflect"
	"strings"
	"testing"
)

func TestParseHtml(t1 *testing.T) {
	var tests = []struct {
		html    string
		plugins []string
		arch    string
		error   string
	}{
		{
			html: `<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 3.2 Final//EN"><html>
			<title>Directory listing for enterprise: /4.1.1/</title>
			<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
			<meta name="robots" content="noindex,nofollow">
			<body>
			<h2>Directory listing for enterprise: /4.1.1/</h2>
			<hr>
			<ul>
				<li><a href="file_386.zip">file_386.zip</a>
				<li><a href="file_amd64.zip">file_amd64.zip</a>
				<li><a href="file_arm.zip">file_arm.zip</a>
				<li><a href="file_arm64.zip">file_arm64.zip</a>
				<li><a href="file_ppc64le.zip">file_ppc64le.zip</a>

				<li><a href="influx_386.zip">influx_386.zip</a>
				<li><a href="influx_amd64.zip">influx_amd64.zip</a>
				<li><a href="influx_arm.zip">influx_arm.zip</a>
				<li><a href="influx_arm64.zip">influx_arm64.zip</a>
				<li><a href="influx_ppc64le.zip">influx_ppc64le.zip</a>
			</ul>
			<hr>
			</body>
			</html>
			`,
			arch:    "arm64",
			plugins: []string{"file", "influx"},
			error:   "",
		},

		{
			html: `<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 3.2 Final//EN"><html>
			<title>Directory listing for enterprise: /4.1.1/</title>
			<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
			<meta name="robots" content="noindex,nofollow">
			<body>
			<h2>Directory listing for enterprise: /4.1.1/</h2>
			<hr>
			<ul>
				<li><a href="file_386.zip">file_386.zip</a>
				<li><a href="file_amd64.zip">file_amd64.zip</a>
				<li><a href="file_arm.zip">file_arm.zip</a>
				<li><a href="file_arm64.zip">file_arm64.zip</a>
				<li><a href="file_ppc64le.zip">file_ppc64le.zip</a>

				<li><a href="influx_386.zip">influx_386.zip</a>
				<li><a href="influx_amd64.zip">influx_amd64.zip</a>
				<li><a href="influx_arm.zip">influx_arm.zip</a>
				<li><a href="influx_arm64.zip">influx_arm64.zip</a>
				<li><a href="influx_ppc64le.zip">influx_ppc64le.zip</a>
			</ul>
			<hr>
			</body>
			</html>
			`,
			arch:    "arm7",
			plugins: []string{},
			error:   "",
		},

		{
			html: `<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 3.2 Final//EN"><html>
			<title>Directory listing for enterprise: /4.1.1/</title>
			<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
			<meta name="robots" content="noindex,nofollow">
			<body>
			<h2>Directory listing for enterprise: /4.1.1/</h2>
			<hr>
			<ul>
				<li><a href="file_386.zip">file_386.zip</a>
				<li><a href="file_amd64.zip">file_amd64.zip</a>
				<li><a href="file_arm.zip">file_arm.zip</a>
				<li><a href="file_arm64.zip">file_arm64.zip</a>
				<li><a href="file_ppc64le.zip">file_ppc64le.zip</a>

				<li><a href="influx_arm.zip">influx_arm.zip</a>
				<li><a href="influx_arm64.zip">influx_arm64.zip</a>
				<li><a href="influx_ppc64le.zip">influx_ppc64le.zip</a>
			</ul>
			<hr>
			</body>
			</html>
			`,
			arch:    "amd64",
			plugins: []string{"file"},
			error:   "",
		},

		{
			html: `<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 3.2 Final//EN"><html>
			<title>Directory listing for enterprise: /4.1.1/</title>
			<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
			<meta name="robots" content="noindex,nofollow">
			<body>
			<h2>Directory listing for enterprise: /4.1.1/</h2>
			<hr>
			<ul>
				
			</ul>
			<hr>
			</body>
			</html>
			`,
			arch:    "amd64",
			plugins: []string{},
			error:   "",
		},

		{
			html:    ``,
			arch:    "amd64",
			plugins: []string{},
			error:   "",
		},
	}

	fmt.Printf("The test bucket size is %d.\n\n", len(tests))
	for i, t := range tests {
		result := extractFromHtml(t.html, t.arch)
		if t.error == "" && !reflect.DeepEqual(t.plugins, result) {
			t1.Errorf("%d. %q\n\nresult mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", i, t.html, t.plugins, result)
		}
	}
}

func TestFetchPluginList(t1 *testing.T) {
	version = "0.9.1"
	// Start a local HTTP server
	server1 := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
		// Send response to be tested
		if _, err := rw.Write([]byte(`<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 3.2 Final//EN"><html>
			<title>Directory listing for enterprise: /4.1.1/</title>
			<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
			<meta name="robots" content="noindex,nofollow">
			<body>
			<h2>Directory listing for enterprise: /4.1.1/</h2>
			<hr>
			<ul>
				<li><a href="file_386.zip">file_386.zip</a>
				<li><a href="file_amd64.zip">file_amd64.zip</a>
				<li><a href="file_arm.zip">file_arm.zip</a>
				<li><a href="file_arm64.zip">file_arm64.zip</a>
				<li><a href="file_ppc64le.zip">file_ppc64le.zip</a>

				<li><a href="influx_386.zip">influx_386.zip</a>
				<li><a href="influx_amd64.zip">influx_amd64.zip</a>
				<li><a href="influx_arm.zip">influx_arm.zip</a>
				<li><a href="influx_arm64.zip">influx_arm64.zip</a>
				<li><a href="influx_ppc64le.zip">influx_ppc64le.zip</a>
			</ul>
			<hr>
			</body>
			</html>
			`)); err != nil {
			fmt.Printf("%s", err)
		}

	}))

	server2 := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
		// Send response to be tested
		if _, err := rw.Write([]byte(`<!DOCTYPE html PUBLIC "-//W3C//DTD HTML 3.2 Final//EN"><html>
			<title>Directory listing for enterprise: /4.1.1/</title>
			<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
			<meta name="robots" content="noindex,nofollow">
			<body>
			<h2>Directory listing for enterprise: /4.1.1/</h2>
			<hr>
			<ul>
				<li><a href="file_arm64.zip">file_arm64.zip</a>

				<li><a href="zmq_386.zip">influx_386.zip</a>
				<li><a href="zmq_amd64.zip">influx_amd64.zip</a>
				<li><a href="zmq_arm.zip">influx_arm.zip</a>
				<li><a href="zmq_arm64.zip">influx_arm64.zip</a>
				<li><a href="zmq_ppc64le.zip">influx_ppc64le.zip</a>
			</ul>
			<hr>
			</body>
			</html>
			`)); err != nil {
			fmt.Printf("%s", err)
		}

	}))

	// Close the server when test finishes
	defer server2.Close()

	if e, r := fetchPluginList(strings.Join([]string{server1.URL, server2.URL}, ","), "sinks", "alpine", "arm64"); e != nil {
		t1.Errorf("Error: %v", e)
	} else {
		exp := map[string]string{
			"file":   server1.URL + "/kuiper-plugins/" + version + "/alpine/sinks/file_arm64.zip",
			"influx": server1.URL + "/kuiper-plugins/" + version + "/alpine/sinks/influx_arm64.zip",
			"zmq":    server2.URL + "/kuiper-plugins/" + version + "/alpine/sinks/zmq_arm64.zip",
		}
		if !reflect.DeepEqual(exp, r) {
			t1.Errorf("result mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", exp, r)
		}
	}

	if e, r := fetchPluginList(strings.Join([]string{server2.URL}, ","), "sinks", "alpine", "arm64"); e != nil {
		t1.Errorf("Error: %v", e)
	} else {
		exp := map[string]string{
			"zmq":  server2.URL + "/kuiper-plugins/" + version + "/alpine/sinks/zmq_arm64.zip",
			"file": server2.URL + "/kuiper-plugins/" + version + "/alpine/sinks/file_arm64.zip",
		}
		if !reflect.DeepEqual(exp, r) {
			t1.Errorf("result mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", exp, r)
		}
	}

	if e, r := fetchPluginList(strings.Join([]string{server1.URL, server2.URL}, ","), "sinks", "alpine", "armv7"); e != nil {
		t1.Errorf("Error: %v", e)
	} else {
		exp := map[string]string{}
		if !reflect.DeepEqual(exp, r) {
			t1.Errorf("result mismatch:\n\nexp=%#v\n\ngot=%#v\n\n", exp, r)
		}
	}
}