def test_all(self, mock_requests):
mock_requests.register_uri(
'GET',
"https://api.chartmogul.com/v1/customers",
request_headers={'Authorization': 'Basic dG9rZW46c2VjcmV0'},
status_code=200,
json=docsListSample)
config = Config("token", "secret")
customers = Customer.all(config).get()
expected = Customer._many(entries=[deserializedCustomer],
has_more=True,
per_page=50,
page=1,
current_page=1,
total_pages=4)
self.assertEqual(mock_requests.call_count, 1, "expected call")
self.assertEqual(mock_requests.last_request.qs, {})
self.assertEqual(mock_requests.last_request.text, None)
# Complete comparing too complicated, would need to:
# 1) sort all dictionaries,
# 2) use special class/library for timezones (Python has no default)
# self.assertEqual(str(customers), str(expected))
# => check only first level fields are OK
self.assertEqual(sorted(dir(customers)), sorted(dir(expected)))
self.assertEqual(sorted(customers.entries[0].attributes.stripe),
sorted(expected.entries[0].attributes.stripe))
self.assertEqual(sorted(customers.entries[0].attributes.clearbit),
sorted(expected.entries[0].attributes.clearbit))
self.assertTrue(isinstance(customers.entries[0], Customer))
def test_create(self, mock_requests):
mock_requests.register_uri('POST',
"https://api.chartmogul.com/v1/customers",
status_code=200,
json=docsListSample["entries"][0])
config = Config("token", "secret")
Customer.create(config, data=createCustomer).get()
self.assertEqual(mock_requests.call_count, 1, "expected call")
self.assertEqual(mock_requests.last_request.qs, {})
self.assertEqual(mock_requests.last_request.json(), sentCreateExpected)
def test_connectSubscriptions(self, mock_requests):
mock_requests.register_uri(
'POST',
"https://api.chartmogul.com/v1/customers/cus_5915ee5a-babd-406b-b8ce-d207133fb4cb/connect_subscriptions",
status_code=202)
jsonRequest = {
'subscriptions': [{
"data_source_uuid":
"ds_ade45e52-47a4-231a-1ed2-eb6b9e541213",
"external_id":
"d1c0c885-add0-48db-8fa9-0bdf5017d6b0"
}, {
"data_source_uuid":
"ds_ade45e52-47a4-231a-1ed2-eb6b9e541213",
"external_id":
"9db5f4a1-1695-44c0-8bd4-de7ce4d0f1d4"
}]
}
config = Config("token", "secret")
result = Customer.connectSubscriptions(
config,
uuid='cus_5915ee5a-babd-406b-b8ce-d207133fb4cb',
data=jsonRequest).get()
self.assertEqual(mock_requests.call_count, 1, "expected call")
self.assertEqual(mock_requests.last_request.qs, {})
self.assertEqual(mock_requests.last_request.json(), jsonRequest)
self.assertEqual(result, None)
def test_search(self, mock_requests):
mock_requests.register_uri(
'GET',
"https://api.chartmogul.com/v1/customers/[email protected]",
status_code=200,
json=docsListSample)
config = Config("token", "secret")
result = Customer.search(config, email='*****@*****.**').get()
self.assertEqual(mock_requests.call_count, 1, "expected call")
self.assertEqual(mock_requests.last_request.qs,
{'email': ['*****@*****.**']})
self.assertEqual(mock_requests.last_request.text, None)
self.assertTrue(isinstance(result, Customer._many))
self.assertTrue(isinstance(result.entries[0], Customer))
def _create_plan(ds):
customer = Customer.create(config,
data={
"data_source_uuid": ds.uuid,
"external_id": "cus_0001",
"name": "Adam Smith",
"email": "*****@*****.**",
"country": "US",
"city": "New York"
})
plan = Plan.create(config,
data={
"data_source_uuid": ds.uuid,
"name": "Bronze Plan",
"interval_count": 1,
"interval_unit": "month",
"external_id": "plan_0001"
})
return Promise.all([ds.uuid, customer, plan])
def test_merge(self, mock_requests):
mock_requests.register_uri(
'POST',
"https://api.chartmogul.com/v1/customers/merges",
status_code=204)
jsonRequest = {
"from": {
"customer_uuid": "cus_de305d54-75b4-431b-adb2-eb6b9e546012"
},
"into": {
"customer_uuid": "cus_ab223d54-75b4-431b-adb2-eb6b9e234571"
}
}
config = Config("token", "secret")
result = Customer.merge(config, data=jsonRequest).get()
self.assertEqual(mock_requests.call_count, 1, "expected call")
self.assertEqual(mock_requests.last_request.qs, {})
self.assertEqual(mock_requests.last_request.json(), jsonRequest)
self.assertEqual(result, None)
deserializedCustomer = Customer(
id=25647,
uuid="cus_de305d54-75b4-431b-adb2-eb6b9e546012",
external_id="34916129",
external_ids=["34916129"],
data_source_uuid="ds_610b7a84-c50f-11e6-8aab-97d6db98913a",
data_source_uuids=["ds_610b7a84-c50f-11e6-8aab-97d6db98913a"],
name="Example Company",
company="",
email="*****@*****.**",
status="Active",
lead_created_at=datetime(2015, 1, 1, 10, 0),
free_trial_started_at=datetime(2015, 1, 9, 10, 0),
customer_since=datetime(2015, 6, 9, 13, 16),
city="Nowhereville",
state="Alaska",
country="US",
zip="0185128",
attributes=Attributes(tags=["engage", "unit loss", "discountable"],
stripe={
"uid": 7,
"coupon": True
},
clearbit={
"company": {
"name": "Example Company",
"legalName": "Example Company Inc.",
"domain": "examplecompany.com",
"url": "http://examplecompany.com",
"category": {
"sector": "Information Technology",
"industryGroup": "Software and Services",
"industry": "Software",
"subIndustry": "Application Software"
},
"metrics": {
"alexaGlobalRank": 2319,
"googleRank": 7,
"employees": 1000,
"marketCap": None,
"raised": 1502450000
},
},
"person": {
"name": {
"fullName": "Bob Kramer"
},
"employment": {
"name": "Example Company"
}
}
},
custom={
"CAC": 213,
"utmCampaign": "social media 1",
"convertedAt": "2015-09-08 00:00:00",
"pro": False,
"salesRep": "Gabi"
}),
address=Address(address_zip="0185128",
city="Nowhereville",
country="US",
state="Alaska"),
mrr=3000.0,
arr=36000.0,
billing_system_url=
"https:\/\/dashboard.stripe.com\/customers\/cus_4Z2ZpyJFuQ0XMb",
chartmogul_url=
"https:\/\/app.chartmogul.com\/#customers\/25647-Example_Company",
billing_system_type="Stripe",
currency="USD",
currency_sign="$")