easypost-go/endshipper_test.go at master · EasyPost/easypost-go · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
package easypost

import (
	"reflect"
	"strings"
)

func (c *ClientTests) TestEndShipperCreate() {
	client := c.TestClient()
	assert, require := c.Assert(), c.Require()

	endshipper, err := client.CreateEndShipper(c.fixture.CaAddress1())
	require.NoError(err)

	assert.Equal(reflect.TypeOf(&Address{}), reflect.TypeOf(endshipper))
	assert.True(strings.HasPrefix(endshipper.ID, "es"))
	assert.Equal("388 TOWNSEND ST APT 20", endshipper.Street1)
}

func (c *ClientTests) TestEndShipperRetrieve() {
	client := c.TestClient()
	assert, require := c.Assert(), c.Require()

	endshipper, err := client.CreateEndShipper(c.fixture.CaAddress1())
	require.NoError(err)

	retrievedEndShipper, err := client.GetEndShipper(endshipper.ID)
	require.NoError(err)

	assert.Equal(reflect.TypeOf(&Address{}), reflect.TypeOf(retrievedEndShipper))
	assert.True(strings.HasPrefix(retrievedEndShipper.ID, "es"))
	assert.Equal(endshipper.Street1, retrievedEndShipper.Street1)
}

func (c *ClientTests) TestEndShipperAll() {
	client := c.TestClient()
	assert, require := c.Assert(), c.Require()

	endshippers, err := client.ListEndShippers(
		&ListOptions{
			PageSize: c.fixture.pageSize(),
		},
	)
	require.NoError(err)

	assert.LessOrEqual(len(endshippers.EndShippers), c.fixture.pageSize())
	assert.NotNil(endshippers.HasMore)
	for _, endshipper := range endshippers.EndShippers {
		assert.Equal(reflect.TypeOf(&Address{}), reflect.TypeOf(endshipper))
	}
}

func (c *ClientTests) TestEndShipperUpdate() {
	client := c.TestClient()
	assert, require := c.Assert(), c.Require()

	endshipper, err := client.CreateEndShipper(c.fixture.CaAddress1())
	require.NoError(err)

	endshipper.Name = "Captain Sparrow"
	endshipper.Company = "EasyPost"
	endshipper.Street1 = "388 Townsend St"
	endshipper.Street2 = "Apt 20"
	endshipper.City = "San Francisco"
	endshipper.State = "CA"
	endshipper.Zip = "94107"
	endshipper.Country = "US"
	endshipper.Phone = "9999999999"
	endshipper.Email = "test@example.com"

	updatedEndShipper, err := client.UpdateEndShippers(endshipper)

	require.NoError(err)

	assert.Equal("CAPTAIN SPARROW", updatedEndShipper.Name)
}