示例#1
0
    def testAll(self, client_mock):
        client_mock.get.return_value = {
            "plans": [response_sample, response_sample, response_sample],
            "page": 1,
            "limit": 100,
            "total": 3,
        }

        service = Service(client_mock)
        plans = service.all()

        self.assertIsInstance(plans, ResponseList)
        self.assertEqual(1, plans.page)
        self.assertEqual(3, plans.total)
        self.assertEqual(response_sample.get("plan_id"), str(plans[0].plan_id))
示例#2
0
def test_all(client_mock, plan_response_sample):
    client_mock.get.return_value = {
        "plans": [plan_response_sample, plan_response_sample, plan_response_sample],
        "page": 1,
        "limit": 100,
        "total": 3,
    }

    service = Service(client_mock)
    plans = service.all()

    assert isinstance(plans, ResponseList)
    assert 1 == plans.page
    assert 3 == plans.total
    assert plan_response_sample.get("plan_id") == str(plans[0].plan_id)
示例#3
0
class PlansIntegrationTest(VCRTestCase):
    def setUp(self) -> None:
        super(PlansIntegrationTest, self).setUp()
        self.client = getnet.Client(
            os.environ.get("GETNET_SELLER_ID"),
            os.environ.get("GETNET_CLIENT_ID"),
            os.environ.get("GETNET_CLIENT_SECRET"),
            getnet.client.HOMOLOG,
        )
        self.service = Service(self.client)

    def testCreate(self):
        data = sample.copy()

        plan = self.service.create(Plan(**data))
        self.assertIsInstance(plan, Plan)
        self.assertIsNotNone(plan.plan_id)

    def testGet(self):
        data = sample.copy()
        created_plan = self.service.create(Plan(**data))

        plan = self.service.get(created_plan.plan_id)

        self.assertIsInstance(plan, Plan)
        self.assertEqual(created_plan, plan)
        self.assertEqual(created_plan.plan_id, plan.plan_id)

    def testInvalidGet(self):
        with self.assertRaises(NotFound) as err:
            self.service.get("14a2ce5d-ebc3-49dc-a516-cb5239b02285")

        self.assertEqual("Not Found", err.exception.error_code)

    def testAll(self):
        plans = self.service.all()
        self.assertIsInstance(plans, ResponseList)
        self.assertEqual(1, plans.page)
        self.assertEqual(100, plans.limit)
        self.assertIsNotNone(plans.total)

    def testAllNotFound(self):
        plans = self.service.all(name="foobarTest123")
        self.assertEqual(0, plans.total)

    def testUpdate(self):
        data = sample.copy()
        created_plan = self.service.create(Plan(**data))

        plan1 = self.service.update(created_plan.plan_id, "FooBar #1",
                                    created_plan.description)
        self.assertEqual("FooBar #1", plan1.name)

        plan2 = self.service.update(created_plan, "FooBar #2")
        self.assertEqual("FooBar #2", plan2.name)

        created_plan.name = "FooBar #3"
        plan3 = self.service.update(created_plan)
        self.assertEqual("FooBar #3", plan3.name)

    def testUpdateStatus(self):
        data = sample.copy()
        created_plan = self.service.create(Plan(**data))
        self.assertTrue(created_plan.is_active)

        plan = self.service.update_status(created_plan.plan_id, False)
        self.assertFalse(plan.is_active)