Exemplo n.º 1
0
func TestConnectionURL(t *testing.T) {

	c := ConnectionURL{}

	// Default connection string is only the protocol.
	if c.String() != "" {
		t.Fatal(`Expecting default connectiong string to be empty, got:`, c.String())
	}

	// Adding a database name.
	c.Database = "myfilename"

	if c.String() != "mongodb://myfilename" {
		t.Fatal(`Test failed, got:`, c.String())
	}

	// Adding an option.
	c.Options = map[string]string{
		"cache": "foobar",
		"mode":  "ro",
	}

	// Adding username and password
	c.User = "******"
	c.Password = "******"

	// Setting host.
	c.Address = db.Host("localhost")

	if c.String() != "mongodb://*****:*****@localhost/myfilename?cache=foobar&mode=ro" {
		t.Fatal(`Test failed, got:`, c.String())
	}

	// Setting host and port.
	c.Address = db.HostPort("localhost", 27017)

	if c.String() != "mongodb://*****:*****@localhost:27017/myfilename?cache=foobar&mode=ro" {
		t.Fatal(`Test failed, got:`, c.String())
	}

	// Setting cluster.
	c.Address = Cluster(db.Host("localhost"), db.Host("1.2.3.4"), db.HostPort("example.org", 1234))

	if c.String() != "mongodb://*****:*****@localhost,1.2.3.4,example.org:1234/myfilename?cache=foobar&mode=ro" {
		t.Fatal(`Test failed, got:`, c.String())
	}

	// Setting another database.
	c.Database = "another_database"

	if c.String() != "mongodb://*****:*****@localhost,1.2.3.4,example.org:1234/another_database?cache=foobar&mode=ro" {
		t.Fatal(`Test failed, got:`, c.String())
	}

}
Exemplo n.º 2
0
// ParseURL parses s into a ConnectionURL struct.
func ParseURL(s string) (u ConnectionURL, err error) {
	o := make(values)

	if strings.HasPrefix(s, "postgres://") {
		s, err = pq.ParseURL(s)
		if err != nil {
			return u, err
		}
	}

	if err := parseOpts(s, o); err != nil {
		return u, err
	}

	u.User = o.Get("user")
	u.Password = o.Get("password")

	h := o.Get("host")
	p, _ := strconv.Atoi(o.Get("port"))

	if p > 0 {
		u.Address = db.HostPort(h, uint(p))
	} else {
		u.Address = db.Host(h)
	}

	u.Database = o.Get("dbname")

	u.Options = map[string]string{
		"sslmode": o.Get("sslmode"),
	}

	return u, err
}
Exemplo n.º 3
0
func TestConnectionURL(t *testing.T) {

	c := ConnectionURL{}

	// Default connection string is empty.
	if c.String() != "" {
		t.Fatal(`Expecting default connectiong string to be empty, got:`, c.String())
	}

	// Adding a host.
	c.Address = db.Host("localhost")

	if c.String() != "host=localhost sslmode=disable" {
		t.Fatal(`Test failed, got:`, c.String())
	}

	// Adding a username.
	c.User = "******"

	if c.String() != "user=Anakin host=localhost sslmode=disable" {
		t.Fatal(`Test failed, got:`, c.String())
	}

	// Adding a password with special characters.
	c.Password = "******"

	if c.String() != `user=Anakin password=Some\ Sort\ of\ \'\ Password host=localhost sslmode=disable` {
		t.Fatal(`Test failed, got:`, c.String())
	}

	// Adding a port.
	c.Address = db.HostPort("localhost", 1234)

	if c.String() != `user=Anakin password=Some\ Sort\ of\ \'\ Password host=localhost port=1234 sslmode=disable` {
		t.Fatal(`Test failed, got:`, c.String())
	}

	// Adding a database.
	c.Database = "MyDatabase"

	if c.String() != `user=Anakin password=Some\ Sort\ of\ \'\ Password host=localhost port=1234 dbname=MyDatabase sslmode=disable` {
		t.Fatal(`Test failed, got:`, c.String())
	}

	// Adding options.
	c.Options = map[string]string{
		"sslmode": "verify-full",
	}

	if c.String() != `user=Anakin password=Some\ Sort\ of\ \'\ Password host=localhost port=1234 dbname=MyDatabase sslmode=verify-full` {
		t.Fatal(`Test failed, got:`, c.String())
	}

}
Exemplo n.º 4
0
func TestConnectionURL(t *testing.T) {

	c := ConnectionURL{}

	// Zero value equals to an empty string.
	if c.String() != "" {
		t.Fatal(`Expecting default connectiong string to be empty, got:`, c.String())
	}

	// Adding a database name.
	c.Database = "mydbname"

	if c.String() != "/mydbname?charset=utf8" {
		t.Fatal(`Test failed, got:`, c.String())
	}

	// Adding an option.
	c.Options = map[string]string{
		"charset": "utf8mb4,utf8",
		"sys_var": "esc@ped",
	}

	if c.String() != "/mydbname?charset=utf8mb4%2Cutf8&sys_var=esc%40ped" {
		t.Fatal(`Test failed, got:`, c.String())
	}

	// Setting default options
	c.Options = nil

	// Setting user and password.
	c.User = "******"
	c.Password = "******"

	if c.String() != `user:pass@/mydbname?charset=utf8` {
		t.Fatal(`Test failed, got:`, c.String())
	}

	// Setting host.
	c.Address = db.HostPort("1.2.3.4", 3306)

	if c.String() != `user:pass@tcp(1.2.3.4:3306)/mydbname?charset=utf8` {
		t.Fatal(`Test failed, got:`, c.String())
	}

	// Setting socket.
	c.Address = db.Socket("/path/to/socket")

	if c.String() != `user:pass@unix(/path/to/socket)/mydbname?charset=utf8` {
		t.Fatal(`Test failed, got:`, c.String())
	}

}
Exemplo n.º 5
0
// Attempts to connect to a database using the stored settings.
func (s *source) Open() error {
	var err error

	// Before db.ConnectionURL we used a unified db.Settings struct. This
	// condition checks for that type and provides backwards compatibility.
	if settings, ok := s.connURL.(db.Settings); ok {

		// User is providing a db.Settings struct, let's translate it into a
		// ConnectionURL{}.
		conn := ConnectionURL{
			User:     settings.User,
			Password: settings.Password,
			Database: settings.Database,
			Options: map[string]string{
				"charset": settings.Charset,
			},
		}

		// Connection charset, UTF-8 by default.
		if conn.Options["charset"] == "" {
			conn.Options["charset"] = "utf8"
		}

		if settings.Socket != "" {
			conn.Address = db.Socket(settings.Socket)
		} else {
			if settings.Host == "" {
				settings.Host = "127.0.0.1"
			}
			if settings.Port == 0 {
				settings.Port = defaultPort
			}
			conn.Address = db.HostPort(settings.Host, uint(settings.Port))
		}

		// Replace original s.connURL
		s.connURL = conn
	}

	if s.session, err = sql.Open(`mysql`, s.connURL.String()); err != nil {
		return err
	}

	if err = s.populateSchema(); err != nil {
		return err
	}

	return nil
}
Exemplo n.º 6
0
// Attempts to connect to a database using the stored settings.
func (s *source) Open() error {
	var err error

	// Before db.ConnectionURL we used a unified db.Settings struct. This
	// condition checks for that type and provides backwards compatibility.
	if settings, ok := s.connURL.(db.Settings); ok {

		// User is providing a db.Settings struct, let's translate it into a
		// ConnectionURL{}.
		conn := ConnectionURL{
			User:     settings.User,
			Password: settings.Password,
			Address:  db.HostPort(settings.Host, uint(settings.Port)),
			Database: settings.Database,
			Options: map[string]string{
				"sslmode": "disable",
			},
		}

		// Testing for SSLMode (deprecated)
		if SSLMode {
			conn.Options["sslmode"] = "verify-full"
		}

		// Replace original s.connURL
		s.connURL = conn
	}

	if s.session, err = sql.Open(`postgres`, s.connURL.String()); err != nil {
		return err
	}

	if err = s.populateSchema(); err != nil {
		return err
	}

	return nil
}