def setUp(self): self.fixture = fixtures.CustomerFixture() self.division_1 = factories.DivisionFactory() self.division_2 = factories.DivisionFactory() self.fixture.customer.division = self.division_1 self.fixture.customer.save() self.url = factories.CustomerFactory.get_url(self.fixture.customer)
def test_private_offerings_are_available_for_users_in_allowed_customers(self): fixture = fixtures.CustomerFixture() self.offering.allowed_customers.add(fixture.customer) self.client.force_authenticate(fixture.owner) response = self.client.get(self.url) self.assertEqual(len(response.data), 1)
def setUp(self): self.fixture = structure_fixtures.CustomerFixture() self.fixture.owner self.invoice = factories.InvoiceFactory(customer=self.fixture.customer) self.patcher = mock.patch('waldur_mastermind.invoices.utils.pdfkit') mock_pdfkit = self.patcher.start() mock_pdfkit.from_string.return_value = b'pdf content'
def test_user_full_name_in_booking_details_if_user_belongs_to_the_same_organization( self, ): creator = self.fixture.order.created_by new_fixture = structure_fixtures.CustomerFixture() new_fixture.customer.add_user( creator, role=structure_models.CustomerRole.SERVICE_MANAGER) self.client.force_authenticate(new_fixture.owner) response = self.client.get( f'/api/marketplace-bookings/{self.fixture.offering.uuid.hex}/') self.assertEqual(response.status_code, status.HTTP_200_OK) self.assertCountEqual( response.data, [ { 'created_by_full_name': creator.full_name, 'start': parse_datetime('2020-02-12T02:00:00+03:00'), 'end': parse_datetime('2020-02-15T02:00:00+03:00'), }, { 'created_by_full_name': creator.full_name, 'start': parse_datetime('2020-03-01T02:00:00+03:00'), 'end': parse_datetime('2020-03-05T02:00:00+03:00'), }, ], )
def setUp(self): self.fixture = structure_fixtures.CustomerFixture() self.offering = marketplace_factories.OfferingFactory( customer=self.fixture.customer, type=PLUGIN_NAME) self.schedules = [ { 'start': '2020-02-12T02:00:00+03:00', 'end': '2020-02-15T02:00:00+03:00', 'id': '123', }, { 'start': '2020-03-01T02:00:00+03:00', 'end': '2020-03-05T02:00:00+03:00', 'id': '456', }, ] self.resource_1 = marketplace_factories.ResourceFactory( offering=self.offering, state=marketplace_models.Resource.States.OK, attributes={'schedules': [self.schedules[0]]}, ) self.resource_2 = marketplace_factories.ResourceFactory( offering=self.offering, state=marketplace_models.Resource.States.OK, attributes={'schedules': [self.schedules[1]]}, )
def setUp(self): self.fixture = fixtures.CustomerFixture() self.fixture.owner self.service_provider = factories.ServiceProviderFactory(customer=self.fixture.customer) offering = factories.OfferingFactory(customer=self.fixture.customer) self.component = factories.OfferingComponentFactory(billing_type=models.OfferingComponent.BillingTypes.USAGE, offering=offering) self.resource = factories.ResourceFactory(offering=offering, state=models.Resource.States.OK, name='My resource')
def setUp(self): self.fixture = structure_fixtures.CustomerFixture() self.service_provider = marketplace_factories.ServiceProviderFactory( customer=self.fixture.customer) self.credentials = factories.GoogleCredentialsFactory( service_provider=self.service_provider, calendar_token='calendar_token', calendar_refresh_token='calendar_refresh_token', ) self.url = factories.GoogleCredentialsFactory.get_authorize_url( self.credentials)
def setUp(self): fixture_1 = structure_fixtures.CustomerFixture() fixture_1.owner fixture_2 = structure_fixtures.CustomerFixture() fixture_2.owner offering_1 = marketplace_factories.OfferingFactory( customer=fixture_1.customer, type=PLUGIN_NAME) offering_2 = marketplace_factories.OfferingFactory( customer=fixture_2.customer, type=PLUGIN_NAME) self.resource = marketplace_factories.ResourceFactory( offering=offering_1, state=marketplace_models.Resource.States.CREATING) self.order_item = marketplace_factories.OrderItemFactory( resource=self.resource, offering=offering_1, state=marketplace_models.OrderItem.States.EXECUTING) self.resource_2 = marketplace_factories.ResourceFactory( offering=offering_2) self.owner = self.resource.offering.customer.get_owners()[0] self.client.force_authenticate(self.owner)
def test_missing_users_are_added_to_customer_group(self, mock_client): fixture = structure_fixtures.CustomerFixture() customer = fixture.customer owner = fixture.owner models.Profile.objects.create(user=owner, username=owner.username) mock_client().group_find.return_value = {'result': []} FreeIPABackend().synchronize_groups() mock_client().group_add_member.assert_called_once_with( 'waldur_org_%s' % customer.uuid.hex, users=[owner.username], skip_errors=True, )
def execute_request(self, ordering_param=None): fixture = structure_fixtures.CustomerFixture() url = structure_factories.CustomerFactory.get_list_url() self.client.force_login(fixture.staff) params = {} if ordering_param: params[api_settings.ORDERING_PARAM] = ordering_param response = self.client.get(url, params) return [ int(customer['billing_price_estimate']['current']) for customer in response.data ]
def setUp(self): self.fixture = structure_fixtures.CustomerFixture() self.offering = marketplace_factories.OfferingFactory( customer=self.fixture.customer, type=PLUGIN_NAME ) self.schedules = [ {'start': '2020-02-12', 'end': '2020-02-15',}, {'start': '2020-03-01', 'end': '2020-03-05',}, ] for schedule in self.schedules: marketplace_factories.ResourceFactory( offering=self.offering, state=marketplace_models.Resource.States.OK, attributes={'schedules': [schedule],}, )
def setUp(self): self.fixture = structure_fixtures.CustomerFixture() self.customer = self.fixture.customer self.other = structure_factories.UserFactory() self.valid_request = { 'amount': decimal.Decimal('9.99'), 'customer': structure_factories.CustomerFactory.get_url(self.customer), 'return_url': 'http://example.com/return/', 'cancel_url': 'http://example.com/cancel/' } self.valid_response = { 'approval_url': 'https://www.paypal.com/webscr?cmd=_express-checkout&token=EC-60U79048BN7719609', 'payer_id': '7E7MGXCWTTKK2', 'token': 'EC-60U79048BN7719609' }
def setUp(self): self.fixture = structure_fixtures.CustomerFixture() schedules = [ { 'start': '2020-01-01T00:00:00.000Z', 'end': '2020-01-02T00:00:00.000Z' }, { 'start': '2020-03-01T00:00:00.000Z', 'end': '2020-03-02T00:00:00.000Z' }, ] self.offering = marketplace_factories.OfferingFactory( customer=self.fixture.customer, type=PLUGIN_NAME, attributes={ 'schedules': schedules, }, ) self.url = marketplace_factories.OfferingFactory.get_url(self.offering)
def test_logger_called_when_customer_role_is_revoked(self): fixture = fixtures.CustomerFixture() owner = fixture.owner with mock.patch( 'waldur_core.structure.handlers.event_logger.customer_role.info' ) as logger_mock: fixture.customer.remove_user(owner, structure_models.CustomerRole.OWNER, fixture.staff) logger_mock.assert_called_once_with( 'User {affected_user_username} has lost role of {role_name} in customer {customer_name}.', event_type='role_revoked', event_context={ 'customer': fixture.customer, 'user': fixture.staff, 'affected_user': fixture.owner, 'structure_type': 'customer', 'role_name': 'Owner', }, )
def setUp(self): super(CustomerAdminTest, self).setUp() self.fixture = fixtures.CustomerFixture() self.customer = self.fixture.customer self.created_by = factories.UserFactory()