class AlertTestCase(unittest.TestCase): def setUp(self): self.client = Client(endpoint='http://api:8080', key='demo-key') def test_customer(self): customer = self.client.create_customer(customer='ACME Corp.', match='example.com') customer_id = customer.id self.assertEqual(customer.customer, 'ACME Corp.') self.assertEqual(customer.match, 'example.com') customer = self.client.update_customer(customer_id, customer='Foo Corp.', match='foo.com') self.assertEqual(customer.customer, 'Foo Corp.') self.assertEqual(customer.match, 'foo.com') customer = self.client.create_customer(customer='Quetzal Inc.', match='quetzal.io') customers = self.client.get_customers() self.assertEqual(len(customers), 2) self.client.delete_customer(customer_id) customers = self.client.get_customers() self.assertEqual(len(customers), 1)
class CustomerTestCase(unittest.TestCase): def setUp(self): self.client = Client() self.customer = """ { "customer": { "customer": "ACME Corp.", "href": "http://localhost:8080/customer/9bb97023-186e-4744-a59d-d18f641eee52", "id": "9bb97023-186e-4744-a59d-d18f641eee52", "match": "example.com" }, "id": "9bb97023-186e-4744-a59d-d18f641eee52", "status": "ok" } """ @requests_mock.mock() def test_customer(self, m): m.post('http://localhost:8080/customer', text=self.customer) customer = self.client.create_customer(customer='ACME Corp.', match='example.com') self.assertEqual(customer.customer, 'ACME Corp.') self.assertEqual(customer.match, 'example.com')