def test_basic_unregistering_from_market(self): # Build the related Mock self._market_mock() # Mock offering info mock_off1 = MagicMock() mock_off1.marketplaces = [] mock_market_off = MagicMock() mock_market_off2 = MagicMock() mock_market_off2.marketplace = self._mock_market mock_off2 = MagicMock() mock_off2.marketplaces = [mock_market_off] mock_off3 = MagicMock() mock_off3.marketplaces = [mock_market_off, mock_market_off2] markets_management.Offering = MagicMock() markets_management.Offering.objects.all.return_value = [mock_off1, mock_off2, mock_off3] markets_management.unregister_from_market(self._user, 'test_market') markets_management.Marketplace.objects.get.assert_called_with(name='test_market') markets_management.marketadaptor_factory.assert_called_once_with(self._mock_market, self._user) self.adaptor_object.delete_store.assert_called_with() # Check that marketplace info has been removed from the corresponding offering self.assertEquals(mock_off1.marketplaces, []) self.assertEquals(mock_off2.marketplaces, [mock_market_off]) self.assertEquals(mock_off3.marketplaces, [mock_market_off]) mock_off3.save.assert_called_once_with()
def test_basic_unregistering_from_market(self): # Build the related Mock self._market_mock() # Mock offering info mock_off1 = MagicMock() mock_off1.marketplaces = [] mock_market_off = MagicMock() mock_market_off2 = MagicMock() mock_market_off2.marketplace = self._mock_market mock_off2 = MagicMock() mock_off2.marketplaces = [mock_market_off] mock_off3 = MagicMock() mock_off3.marketplaces = [mock_market_off, mock_market_off2] markets_management.Offering = MagicMock() markets_management.Offering.objects.all.return_value = [mock_off1, mock_off2, mock_off3] markets_management.unregister_from_market(self._user, "test_market") markets_management.Marketplace.objects.get.assert_called_with(name="test_market") markets_management.marketadaptor_factory.assert_called_once_with(self._mock_market, self._user) self.adaptor_object.delete_store.assert_called_with() # Check that marketplace info has been removed from the corresponding offering self.assertEquals(mock_off1.marketplaces, []) self.assertEquals(mock_off2.marketplaces, [mock_market_off]) self.assertEquals(mock_off3.marketplaces, [mock_market_off]) mock_off3.save.assert_called_once_with()
def test_basic_unregistering_from_market(self, mock): # Build the related Mock mock(self) markets_management.unregister_from_market('test_market') markets_management.Marketplace.objects.get.assert_called_with(name='test_market') self.adaptor_object.delete_store.assert_called_with('test_store')
def test_unregistering_already_unregistered(self): self._market_mock1() markets_management.Marketplace.objects.get.side_effect = Exception('Not found') error = False try: markets_management.unregister_from_market('test_market1') except: error = True self.assertTrue(error)
def test_unregistering_already_unregistered(self): self._market_mock() markets_management.Marketplace.objects.get.side_effect = Exception('Not found') error = False try: markets_management.unregister_from_market(self._user, 'test_market1') except: error = True self.assertTrue(error)
def test_unregistering_bad_gateway(self): # Mock Marketplace self._market_mock1() self.adaptor_object.delete_store.side_effect = HTTPError('http://testmarket.org', 500, 'Server error', None, None) error = False msg = None try: markets_management.unregister_from_market('test_market') except Exception as e: error = True msg = e.message self.assertTrue(error) self.assertEquals(msg, 'Bad Gateway') self.adaptor_object.delete_store.assert_called_with('test_store')
def test_unregistering_bad_gateway(self): # Mock Marketplace self._market_mock() self.adaptor_object.delete_store.side_effect = HTTPError('http://testmarket.org', 500, 'Server error', None, None) error = False msg = None try: markets_management.unregister_from_market(self._user, 'test_market') except Exception as e: error = True msg = e.message self.assertTrue(error) self.assertEquals(msg, 'Bad Gateway') self.adaptor_object.delete_store.assert_called_with() markets_management.marketadaptor_factory.assert_called_once_with(self._mock_market, self._user)
def delete(self, request, market): if not request.user.is_staff: return build_response(request, 403, 'Forbidden') try: unregister_from_market(request.user, market) except Exception, e: if e.message == 'Bad Gateway': code = 502 msg = e.message else: if e.message == 'Not found': code = 404 msg = e.message else: code = 400 msg = 'Bad request' return build_response(request, code, msg)
def test_unregistering_bad_gateway(self): # Mock Marketplace self._market_mock() self.adaptor_object.delete_store.side_effect = HTTPError( "http://testmarket.org", 500, "Server error", None, None ) error = False msg = None try: markets_management.unregister_from_market(self._user, "test_market") except Exception as e: error = True msg = e.message self.assertTrue(error) self.assertEquals(msg, "Bad Gateway") self.adaptor_object.delete_store.assert_called_with() markets_management.marketadaptor_factory.assert_called_once_with(self._mock_market, self._user)