How to use Check method of host Package

Best Syzkaller code snippet using host.Check

helpers.go

Source:helpers.go

...14	CurrentDb string `json:"CurrentDb"`15}16// CommandOrchestrator is the function that all commands call to be executed. It orchestrates17// validation and configuration for each command.18func CommandOrchestrator(initCommand string, args []string, prevArgCheckMap map[string]bool, checkDbRequired bool, validators ...map[string]bool) {19	// get user's $HOME dir20	home, err := homedir.Dir()21	CheckErr("Could not get $HOME dir:", err)22	// get currently set host23	currentHost, err := GetHostState(home)24	CheckErr("Could not get currently set host:", err)25	// check host config exists26	CheckErr(fmt.Sprintf("Could not get config for host %s", currentHost.Host), CheckHostConfig(currentHost.Host))27	// if config exists, get settings for current host28	hostSettings := viper.GetStringMapString(currentHost.Host)29	// mysql30	if hostSettings["platform"] == "mysql" {31		// find 'mysql' executable path32		mysqlPath, err := exec.LookPath("mysql")33		if err != nil {34			fmt.Println(err)35			os.Exit(1)36		}37		// isClauseNeedingDb checks if args includes item that requires 'use db'38		var isClauseNeedingDb bool39		if checkDbRequired {40			// if true, the option to not use db exists..41			isClauseNeedingDb = CheckArgsInclude(args, validators...)42			if isClauseNeedingDb {43				if currentHost.CurrentDb == "" {44					fmt.Println("Please use 'ssql use [dbName]' first")45					os.Exit(1)46				}47			}48		} else {49			// else, there is no option, db is needed50			// assert that a db is set51			isClauseNeedingDb = true52			if currentHost.CurrentDb == "" {53				fmt.Println("Please use 'ssql use [dbName]' first")54				os.Exit(1)55			}56		}57		// build query and query command58		dbQueryBase := initCommand59		// select does not need to wrap any values in quotes60		//prevArgCheck := map[string]bool{"placeholder": false}61		dbQueryBase = BuildDbQuery(args, prevArgCheckMap, dbQueryBase)62		var dbQuery string63		if !isClauseNeedingDb {64			dbQuery = dbQueryBase65		} else {66			dbQuery = fmt.Sprintf(`USE %s; %s;`, currentHost.CurrentDb, dbQueryBase)67		}68		query := CreateMysqlQueryCommand(dbQuery, hostSettings, mysqlPath)69		// run the command70		if err := query.Run(); err != nil {71			CheckErr(fmt.Sprintf("Could not run mysql command for query %s", dbQuery), err)72		}73		// postgres74	} else {75		// TODO: else do postgres stuff(coming soon)76	}77}78// GetLongDescription creates a long description for the passed command.79func GetLongDescription(commandName string) string {80	return fmt.Sprintf(`The %s command has most of the usual functionality81provided by the sql %s statement.`, commandName, commandName)82}83// GetHostState reads the currently set host from the user's84// $HOME/.ssqlHostState file and returns the unmarshaled json85// string as a data structure.86func GetHostState(home string) (stateData hostState, err error) {87	hostStateFile, err := ioutil.ReadFile(home + "/.ssqlHostState")88	stateData = hostState{}89	_ = json.Unmarshal([]byte(hostStateFile), &stateData)90	return stateData, err91}92// SetHostState uses the user's $HOME path and the user's desired93// host retrieved from args to create a state containing which94// host from the user's config file should be used. The state is95// stored in the file $HOME/.ssqlHostState.96func SetHostState(home, host, platform, currentDb string) (err error) {97	hostData := hostState{98		Host:      host,99		Platform:  platform,100		CurrentDb: currentDb, // current db will always default to empty101	}102	json, err := json.MarshalIndent(hostData, "", "  ")103	if err != nil {104		return err105	}106	_ = ioutil.WriteFile(home+"/.ssqlHostState", json, 0644)107	return108}109// CheckHostConfig takes the user's desired host from args and asserts110// that it is available in the user's .ssql config file.111func CheckHostConfig(desiredHost string) error {112	if viper.IsSet(desiredHost) == false {113		return fmt.Errorf("Host '%s' is not configured. Please make sure the desired host is configured in your .ssql config file.", desiredHost)114	}115	return nil116}117// CheckErr is a general function that can be used to check for errors. If118// an error is present, it will output the error to stdout and terminate the119// program.120func CheckErr(fromMsg string, err error) {121	if e := err; err != nil {122		fmt.Println(fromMsg, e)123		os.Exit(1)124	}125}126// CreateMysqlQueryCommand creates the mysql command out of the user's host config127// and passed args.128func CreateMysqlQueryCommand(dbQuery string, hostSettings map[string]string, commandPath string) *exec.Cmd {129	if hostSettings["defaults_file"] != "" {130		dbUser := fmt.Sprintf(`-u%s`, hostSettings["user"])131		dbOptionFile := fmt.Sprintf(`--defaults-file=%s`, hostSettings["defaults_file"])132		dbCmd := &exec.Cmd{133			Path: commandPath,134			Args: []string{commandPath, dbOptionFile, `-h`, hostSettings["host"],135				`-P`, hostSettings["port"], dbUser, `-e`, dbQuery},136			Stdout: os.Stdout,137			Stderr: os.Stderr,138		}139		return dbCmd140	} else {141		dbUser := fmt.Sprintf(`-u%s`, hostSettings["user"])142		dbPassword := fmt.Sprintf(`-p%s`, hostSettings["password"])143		dbCmd := &exec.Cmd{144			Path: commandPath,145			Args: []string{commandPath, `-h`, hostSettings["host"],146				`-P`, hostSettings["port"], dbUser, dbPassword, `-e`, dbQuery},147			Stdout: os.Stdout,148			Stderr: os.Stderr,149		}150		return dbCmd151	}152}153// CheckArgsInclude is a generic variadic function that checks to see if an array(args)154// contains an element or elements contained in n maps. The maps are checked concurrently155// using goroutines.156func CheckArgsInclude(args []string, validators ...map[string]bool) bool {157	countGoroutine := len(validators)158	c := make(chan bool)159	// initiate n goroutines160	for i := 0; i < len(validators); i++ {161		go checkArgsIncludeChild(args, validators[i], c)162	}163	for countToComplete := 0; countToComplete < countGoroutine; countToComplete++ {164		result := <-c165		if result == true {166			return result167		}168	}169	return false170}171// checkArgsIncludeChild checks if each arg is present in a map and172// writes the results to a channel read by CheckArgsInclude.173func checkArgsIncludeChild(args []string, validator map[string]bool, c chan bool) {174	for iArg := 0; iArg < len(args); iArg++ {175		if validator[args[iArg]] == true {176			c <- true177		}178	}179	c <- false180}181// BuildDbQuery uses args passed by the user and builds a db182// query out of them.183func BuildDbQuery(args []string, prevArgCheck map[string]bool, dbQueryBase string) string {184	for i := 0; i < len(args); i++ {185		// if the previous arg is not less than 0(out of index range) and186		// the previous arg is LIKE, wrap pattern in quotes187		if i > 0 && prevArgCheck[args[i-1]] == true {188			dbQueryBase = fmt.Sprintf(`%s "%s"`, dbQueryBase, args[i])189		} else {190			dbQueryBase = dbQueryBase + " " + args[i]191		}192	}193	return dbQueryBase194}...

resource_compute_shared_vpc_test.go

Source:resource_compute_shared_vpc_test.go

...12	serviceProject := fmt.Sprintf("tf-test-s-%d", randInt(t))13	hostProjectResourceName := "google_compute_shared_vpc_host_project.host"14	serviceProjectResourceName := "google_compute_shared_vpc_service_project.service"15	vcrTest(t, resource.TestCase{16		PreCheck:  func() { testAccPreCheck(t) },17		Providers: testAccProviders,18		Steps: []resource.TestStep{19			{20				Config: testAccComputeSharedVpc_basic(hostProject, serviceProject, org, billingId),21				Check: resource.ComposeTestCheckFunc(22					testAccCheckComputeSharedVpcHostProject(t, hostProject, true),23					testAccCheckComputeSharedVpcServiceProject(t, hostProject, serviceProject, true),24				),25			},26			// Test import.27			{28				ResourceName:      hostProjectResourceName,29				ImportState:       true,30				ImportStateVerify: true,31			},32			{33				ResourceName:      serviceProjectResourceName,34				ImportState:       true,35				ImportStateVerify: true,36			},37			// Use a separate TestStep rather than a CheckDestroy because we need the project to still exist.38			{39				Config: testAccComputeSharedVpc_disabled(hostProject, serviceProject, org, billingId),40				Check: resource.ComposeTestCheckFunc(41					testAccCheckComputeSharedVpcHostProject(t, hostProject, false),42					testAccCheckComputeSharedVpcServiceProject(t, hostProject, serviceProject, false),43				),44			},45		},46	})47}48func testAccCheckComputeSharedVpcHostProject(t *testing.T, hostProject string, enabled bool) resource.TestCheckFunc {49	return func(s *terraform.State) error {50		config := googleProviderConfig(t)51		found, err := config.NewComputeClient(config.userAgent).Projects.Get(hostProject).Do()52		if err != nil {53			return fmt.Errorf("Error reading project %s: %s", hostProject, err)54		}55		if found.Name != hostProject {56			return fmt.Errorf("Project %s not found", hostProject)57		}58		if enabled != (found.XpnProjectStatus == "HOST") {59			return fmt.Errorf("Project %q shared VPC status was not expected, got %q", hostProject, found.XpnProjectStatus)60		}61		return nil62	}63}64func testAccCheckComputeSharedVpcServiceProject(t *testing.T, hostProject, serviceProject string, enabled bool) resource.TestCheckFunc {65	return func(s *terraform.State) error {66		config := googleProviderConfig(t)67		serviceHostProject, err := config.NewComputeClient(config.userAgent).Projects.GetXpnHost(serviceProject).Do()68		if err != nil {69			if enabled {70				return fmt.Errorf("Expected service project to be enabled.")71			}72			return nil73		}74		if enabled != (serviceHostProject.Name == hostProject) {75			return fmt.Errorf("Wrong host project for the given service project. Expected '%s', got '%s'", hostProject, serviceHostProject.Name)76		}77		return nil78	}...

discovery_test.go

Source:discovery_test.go

1package discovery2import (3	"testing"4	"github.com/go-check/check"5)6// Hook up gocheck into the "go test" runner.7func Test(t *testing.T) { check.TestingT(t) }8type DiscoverySuite struct{}9var _ = check.Suite(&DiscoverySuite{})10func (s *DiscoverySuite) TestNewEntry(c *check.C) {11	entry, err := NewEntry("127.0.0.1:2375")12	c.Assert(err, check.IsNil)13	c.Assert(entry.Equals(&Entry{Host: "127.0.0.1", Port: "2375"}), check.Equals, true)14	c.Assert(entry.String(), check.Equals, "127.0.0.1:2375")15	entry, err = NewEntry("[2001:db8:0:f101::2]:2375")16	c.Assert(err, check.IsNil)17	c.Assert(entry.Equals(&Entry{Host: "2001:db8:0:f101::2", Port: "2375"}), check.Equals, true)18	c.Assert(entry.String(), check.Equals, "[2001:db8:0:f101::2]:2375")19	_, err = NewEntry("127.0.0.1")20	c.Assert(err, check.NotNil)21}22func (s *DiscoverySuite) TestParse(c *check.C) {23	scheme, uri := parse("127.0.0.1:2375")24	c.Assert(scheme, check.Equals, "nodes")25	c.Assert(uri, check.Equals, "127.0.0.1:2375")26	scheme, uri = parse("localhost:2375")27	c.Assert(scheme, check.Equals, "nodes")28	c.Assert(uri, check.Equals, "localhost:2375")29	scheme, uri = parse("scheme://127.0.0.1:2375")30	c.Assert(scheme, check.Equals, "scheme")31	c.Assert(uri, check.Equals, "127.0.0.1:2375")32	scheme, uri = parse("scheme://localhost:2375")33	c.Assert(scheme, check.Equals, "scheme")34	c.Assert(uri, check.Equals, "localhost:2375")35	scheme, uri = parse("")36	c.Assert(scheme, check.Equals, "nodes")37	c.Assert(uri, check.Equals, "")38}39func (s *DiscoverySuite) TestCreateEntries(c *check.C) {40	entries, err := CreateEntries(nil)41	c.Assert(entries, check.DeepEquals, Entries{})42	c.Assert(err, check.IsNil)43	entries, err = CreateEntries([]string{"127.0.0.1:2375", "127.0.0.2:2375", "[2001:db8:0:f101::2]:2375", ""})44	c.Assert(err, check.IsNil)45	expected := Entries{46		&Entry{Host: "127.0.0.1", Port: "2375"},47		&Entry{Host: "127.0.0.2", Port: "2375"},48		&Entry{Host: "2001:db8:0:f101::2", Port: "2375"},49	}50	c.Assert(entries.Equals(expected), check.Equals, true)51	_, err = CreateEntries([]string{"127.0.0.1", "127.0.0.2"})52	c.Assert(err, check.NotNil)53}54func (s *DiscoverySuite) TestContainsEntry(c *check.C) {55	entries, err := CreateEntries([]string{"127.0.0.1:2375", "127.0.0.2:2375", ""})56	c.Assert(err, check.IsNil)57	c.Assert(entries.Contains(&Entry{Host: "127.0.0.1", Port: "2375"}), check.Equals, true)58	c.Assert(entries.Contains(&Entry{Host: "127.0.0.3", Port: "2375"}), check.Equals, false)59}60func (s *DiscoverySuite) TestEntriesEquality(c *check.C) {61	entries := Entries{62		&Entry{Host: "127.0.0.1", Port: "2375"},63		&Entry{Host: "127.0.0.2", Port: "2375"},64	}65	// Same66	c.Assert(entries.Equals(Entries{67		&Entry{Host: "127.0.0.1", Port: "2375"},68		&Entry{Host: "127.0.0.2", Port: "2375"},69	}), check.70		Equals, true)71	// Different size72	c.Assert(entries.Equals(Entries{73		&Entry{Host: "127.0.0.1", Port: "2375"},74		&Entry{Host: "127.0.0.2", Port: "2375"},75		&Entry{Host: "127.0.0.3", Port: "2375"},76	}), check.77		Equals, false)78	// Different content79	c.Assert(entries.Equals(Entries{80		&Entry{Host: "127.0.0.1", Port: "2375"},81		&Entry{Host: "127.0.0.42", Port: "2375"},82	}), check.83		Equals, false)84}85func (s *DiscoverySuite) TestEntriesDiff(c *check.C) {86	entry1 := &Entry{Host: "1.1.1.1", Port: "1111"}87	entry2 := &Entry{Host: "2.2.2.2", Port: "2222"}88	entry3 := &Entry{Host: "3.3.3.3", Port: "3333"}89	entries := Entries{entry1, entry2}90	// No diff91	added, removed := entries.Diff(Entries{entry2, entry1})92	c.Assert(added, check.HasLen, 0)93	c.Assert(removed, check.HasLen, 0)94	// Add95	added, removed = entries.Diff(Entries{entry2, entry3, entry1})96	c.Assert(added, check.HasLen, 1)97	c.Assert(added.Contains(entry3), check.Equals, true)98	c.Assert(removed, check.HasLen, 0)99	// Remove100	added, removed = entries.Diff(Entries{entry2})101	c.Assert(added, check.HasLen, 0)102	c.Assert(removed, check.HasLen, 1)103	c.Assert(removed.Contains(entry1), check.Equals, true)104	// Add and remove105	added, removed = entries.Diff(Entries{entry1, entry3})106	c.Assert(added, check.HasLen, 1)107	c.Assert(added.Contains(entry3), check.Equals, true)108	c.Assert(removed, check.HasLen, 1)109	c.Assert(removed.Contains(entry2), check.Equals, true)110}...

Check

Using AI Code Generation

1import (2func main() {3        addrs, err := net.LookupHost("www.google.com")4        if err != nil {5                fmt.Println("Error:", err.Error())6        }7        for i := 0; i < len(addrs); i++ {8                fmt.Println(addrs[i])9        }10        fmt.Println("Done")11}

Check

Using AI Code Generation

1import (2func main() {3	hosts := []string{4	}5	for _, host := range hosts {6		if Check(host) {7			fmt.Println(host, "is up!")8		} else {9			fmt.Println(host, "is down!")10		}11	}12}13func Check(host string) bool {14	_, err := net.Dial("tcp", host+":80")15	if err != nil {16	}17}

Check

Using AI Code Generation

1import (2func main() {3	hosts := []string{4	}5	c := make(chan string)6	for _, host := range hosts {7		go Check(host, c)8	}9	for i := 0; i < len(hosts); i++ {10		fmt.Println(<-c)11	}12}13func Check(host string, c chan string) {14	if _, err := net.Dial("tcp", host+":80"); err != nil {15	} else {16	}17}

Check

Using AI Code Generation

1import (2func main() {3        ips := []net.IPAddr{4                {IP: net.IP{127, 0, 0, 1}},5                {IP: net.IP{8, 8, 8, 8}},6                {IP: net.IP{8, 8, 4, 4}},7        }8        for _, v := range ips {9                fmt.Println(net.CheckIP(v.IP))10        }11}12GoLang net.LookupAddr() Method13func LookupAddr(addr string) (names []string, err error)14import (15func main() {16        ips := []string{

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.