diff --git a/cli/client.go b/cli/client.go index aae57c46267..0aef6a19c65 100644 --- a/cli/client.go +++ b/cli/client.go @@ -333,7 +333,7 @@ func (c *viamClient) updateBillingServiceAction(cCtx *cli.Context, orgID, addres printf(cCtx.App.Writer, "City: %s", address.GetCity()) printf(cCtx.App.Writer, "State: %s", address.GetState()) printf(cCtx.App.Writer, "Postal Code: %s", address.GetZipcode()) - printf(cCtx.App.Writer, "Country: USA") + printf(cCtx.App.Writer, "Country: %s", address.GetCountry()) return nil } @@ -375,7 +375,10 @@ func (c *viamClient) getBillingConfig(cCtx *cli.Context, orgID string) error { printf(cCtx.App.Writer, "City: %s", resp.BillingAddress.GetCity()) printf(cCtx.App.Writer, "State: %s", resp.BillingAddress.GetState()) printf(cCtx.App.Writer, "Postal Code: %s", resp.BillingAddress.GetZipcode()) - printf(cCtx.App.Writer, "Country: %s", "USA") + if resp.BillingAddress.GetCountry() != "" { + printf(cCtx.App.Writer, "Country: %s", resp.BillingAddress.GetCountry()) + } + return nil } diff --git a/cli/client_test.go b/cli/client_test.go index 11819882327..930eecd5388 100644 --- a/cli/client_test.go +++ b/cli/client_test.go @@ -264,6 +264,7 @@ func TestGetBillingConfigAction(t *testing.T) { City: "San Francisco", State: "CA", Zipcode: "94105", + Country: "United States", }, LogoUrl: "https://logo.com", BillingDashboardUrl: "https://app.viam.dev/my-dashboard", @@ -289,7 +290,7 @@ func TestGetBillingConfigAction(t *testing.T) { test.That(t, out.messages[8], test.ShouldContainSubstring, "San Francisco") test.That(t, out.messages[9], test.ShouldContainSubstring, "CA") test.That(t, out.messages[10], test.ShouldContainSubstring, "94105") - test.That(t, out.messages[11], test.ShouldContainSubstring, "USA") + test.That(t, out.messages[11], test.ShouldContainSubstring, "United States") } func TestOrganizationSetLogoAction(t *testing.T) { @@ -435,7 +436,7 @@ func TestUpdateBillingServiceAction(t *testing.T) { } cCtx, ac, out, errOut := setup(asc, nil, nil, nil, "token") - address := "123 Main St, Suite 100, San Francisco, CA, 94105" + address := "123 Main St, Suite 100, San Francisco, CA, 94105, United States" test.That(t, ac.updateBillingServiceAction(cCtx, "test-org", address), test.ShouldBeNil) test.That(t, len(errOut.messages), test.ShouldEqual, 0) test.That(t, len(out.messages), test.ShouldEqual, 8) @@ -446,7 +447,7 @@ func TestUpdateBillingServiceAction(t *testing.T) { test.That(t, out.messages[4], test.ShouldContainSubstring, "San Francisco") test.That(t, out.messages[5], test.ShouldContainSubstring, "CA") test.That(t, out.messages[6], test.ShouldContainSubstring, "94105") - test.That(t, out.messages[7], test.ShouldContainSubstring, "USA") + test.That(t, out.messages[7], test.ShouldContainSubstring, "United States") } func TestOrganizationEnableBillingServiceAction(t *testing.T) { diff --git a/cli/utils.go b/cli/utils.go index 477e1c5ad9a..df787f7a452 100644 --- a/cli/utils.go +++ b/cli/utils.go @@ -72,16 +72,17 @@ func parseBillingAddress(address string) (*apppb.BillingAddress, error) { } splitAddress := strings.Split(address, ",") - if len(splitAddress) != 4 && len(splitAddress) != 5 { - return nil, errors.Errorf("address: %s does not follow the format: line1, line2 (optional), city, state, zipcode", address) + if len(splitAddress) != 5 && len(splitAddress) != 6 { + return nil, errors.Errorf("address: %s does not follow the format: line1, line2 (optional), city, state, zipcode, country", address) } - if len(splitAddress) == 4 { + if len(splitAddress) == 5 { return &apppb.BillingAddress{ AddressLine_1: strings.Trim(splitAddress[0], " "), City: strings.Trim(splitAddress[1], " "), State: strings.Trim(splitAddress[2], " "), Zipcode: strings.Trim(splitAddress[3], " "), + Country: strings.Trim(splitAddress[4], " "), }, nil } @@ -92,6 +93,7 @@ func parseBillingAddress(address string) (*apppb.BillingAddress, error) { City: strings.Trim(splitAddress[2], " "), State: strings.Trim(splitAddress[3], " "), Zipcode: strings.Trim(splitAddress[4], " "), + Country: strings.Trim(splitAddress[5], " "), }, nil } diff --git a/cli/utils_test.go b/cli/utils_test.go index 270dae823fa..94d573125a8 100644 --- a/cli/utils_test.go +++ b/cli/utils_test.go @@ -48,22 +48,24 @@ func TestParseBillingAddress(t *testing.T) { expectedErr error }{ { - input: "123 Main St, Apt 1, San Francisco, CA, 94105", + input: "123 Main St, Apt 1, San Francisco, CA, 94105, United States", expectedAddress: &apppb.BillingAddress{ AddressLine_1: "123 Main St", AddressLine_2: &addressLine2, City: "San Francisco", State: "CA", Zipcode: "94105", + Country: "United States", }, }, { - input: "123 Main St, San Francisco, CA, 94105", + input: "123 Main St, San Francisco, CA, 94105, United States", expectedAddress: &apppb.BillingAddress{ AddressLine_1: "123 Main St", City: "San Francisco", State: "CA", Zipcode: "94105", + Country: "United States", }, }, { diff --git a/go.mod b/go.mod index c76dc82c03d..2fe1b5f2c5a 100644 --- a/go.mod +++ b/go.mod @@ -75,7 +75,7 @@ require ( go.uber.org/atomic v1.11.0 go.uber.org/multierr v1.11.0 go.uber.org/zap v1.27.0 - go.viam.com/api v0.1.387 + go.viam.com/api v0.1.388 go.viam.com/test v1.2.4 go.viam.com/utils v0.1.128 goji.io v2.0.2+incompatible diff --git a/go.sum b/go.sum index ccfb2f67c81..df50461c268 100644 --- a/go.sum +++ b/go.sum @@ -1513,8 +1513,8 @@ go.uber.org/zap v1.18.1/go.mod h1:xg/QME4nWcxGxrpdeYfq7UvYrLh66cuVKdrbD1XF/NI= go.uber.org/zap v1.23.0/go.mod h1:D+nX8jyLsMHMYrln8A0rJjFt/T/9/bGgIhAqxv5URuY= go.uber.org/zap v1.27.0 h1:aJMhYGrd5QSmlpLMr2MftRKl7t8J8PTZPA732ud/XR8= go.uber.org/zap v1.27.0/go.mod h1:GB2qFLM7cTU87MWRP2mPIjqfIDnGu+VIO4V/SdhGo2E= -go.viam.com/api v0.1.387 h1:nxWF+dO+z2mwTpTkcs8mr8L0ic4ZwcqOAumBtHnvqsc= -go.viam.com/api v0.1.387/go.mod h1:g5eipXHNm0rQmW7DWya6avKcmzoypLmxnMlAaIsE5Ls= +go.viam.com/api v0.1.388 h1:6UW1+5DY1zIupzJnms9oGEKG8xHgu0T1hc6Zx5UhAeQ= +go.viam.com/api v0.1.388/go.mod h1:g5eipXHNm0rQmW7DWya6avKcmzoypLmxnMlAaIsE5Ls= go.viam.com/test v1.2.4 h1:JYgZhsuGAQ8sL9jWkziAXN9VJJiKbjoi9BsO33TW3ug= go.viam.com/test v1.2.4/go.mod h1:zI2xzosHdqXAJ/kFqcN+OIF78kQuTV2nIhGZ8EzvaJI= go.viam.com/utils v0.1.128 h1:ScO0pWiwoYzsILcJudL4axCAGhNPmzszJeMoHhdx23s=