LITE-29342 Add disable action for feeds

connect_bi_reporter/feeds/api/views.py

@@ -15,10 +15,11 @@
     FeedUpdateSchema,
     map_to_feed_schema,
 )
+from connect_bi_reporter.feeds.enums import FeedStatusChoices
 from connect_bi_reporter.feeds.services import (
+    change_feed_status,
     create_feed,
     delete_feed,
-    enable_feed,
     get_feed_or_404,
     get_feeds,
     update_feed,
@@ -115,6 +116,7 @@ def delete_feed(
         summary='Enable a Feed',
         response_model=FeedSchema,
         status_code=status.HTTP_200_OK,
+        name=FeedStatusChoices.enabled,
     )
     def enable_feed(
         self,
@@ -123,6 +125,26 @@ def enable_feed(
         installation: dict = Depends(get_installation),
         request: Request = None,
     ):
+        return self.handle_feed_status_change(feed_id, db, installation, request)
+
+    @router.post(
+        '/feeds/{feed_id}/disable',
+        summary='Enable a Feed',
+        response_model=FeedSchema,
+        status_code=status.HTTP_200_OK,
+        name=FeedStatusChoices.disabled,
+    )
+    def disable_feed(
+        self,
+        feed_id: str,
+        db: VerboseBaseSession = Depends(get_db),
+        installation: dict = Depends(get_installation),
+        request: Request = None,
+    ):
+        return self.handle_feed_status_change(feed_id, db, installation, request)
+
+    def handle_feed_status_change(self, feed_id, db, installation, request):
         logged_user_data = get_user_data_from_auth_token(request.headers['connect-auth'])
-        feed = enable_feed(db, installation, feed_id, logged_user_data)
+        status_for_change = request.scope['route'].name
+        feed = change_feed_status(db, installation, feed_id, logged_user_data, status_for_change)
         return map_to_feed_schema(feed)

connect_bi_reporter/feeds/services.py

@@ -4,7 +4,6 @@
 from connect_extension_utils.api.views import get_object_or_404
 
 from connect_bi_reporter.feeds.api.schemas import FeedCreateSchema, FeedUpdateSchema
-from connect_bi_reporter.feeds.enums import FeedStatusChoices
 from connect_bi_reporter.feeds.errors import FeedError
 from connect_bi_reporter.feeds.models import Feed
 from connect_bi_reporter.feeds.validator import FeedValidator
@@ -79,21 +78,22 @@ def delete_feed(
     db.commit()
 
 
-def enable_feed(
+def change_feed_status(
     db,
     installation: Dict[str, Any],
     feed_id: str,
     user: Dict[str, str],
+    status: str,
 ):
     feed = get_feed_or_404(db, installation, feed_id)
-    if feed.status == FeedStatusChoices.enabled:
+    if feed.status == status:
         raise FeedError.RF_003(
             format_kwargs={
                 'feed_id': feed.id,
                 'status': feed.status,
             },
         )
-    feed.status = FeedStatusChoices.enabled
+    feed.status = status
     feed.updated_by = user['id']
     db.commit()
     return feed

tests/feeds/api/test_views.py

@@ -437,7 +437,7 @@ def test_enable_feed_404(installation, api_client, connect_auth_header):
     assert response_data['errors'][0] == 'Object `NOT-FOUND` not found.'
 
 
-def test_already_enable_feed(
+def test_already_enabled_feed(
     installation,
     feed_factory,
     api_client,
@@ -456,3 +456,71 @@ def test_already_enable_feed(
     assert response_data['errors'][0] == (
         f'Feed `{feed.id}` is already in status `{feed.status}`.'
     )
+
+
+def test_disable_feed(
+    installation,
+    feed_factory,
+    api_client,
+    connect_auth_header,
+):
+    feed = feed_factory(account_id=installation['owner']['id'])
+    updated_at = feed.updated_at
+    previous_status = feed.status
+
+    response = api_client.post(
+        f'/api/feeds/{feed.id}/disable',
+        installation=installation,
+        headers={'connect-auth': connect_auth_header},
+    )
+    user = get_user_data_from_auth_token(connect_auth_header)
+
+    assert response.status_code == 200
+
+    response_data = response.json()
+    assert response_data['id'] == feed.id
+    assert response_data['id'].startswith(feed_factory._meta.model.PREFIX)
+    assert response_data['file_name'] == feed.file_name
+    assert response_data['description'] == feed.description
+    assert response_data['owner']['id'] == installation['owner']['id']
+    events = response_data['events']
+    assert events['created']['at'] is not None
+    assert events['created']['by'] is not None
+    assert feed.updated_at > updated_at
+    assert events['updated']['by']['id'] == user['id']
+    assert response_data['status'] == 'disabled'
+    assert feed.status != previous_status and feed.status == 'disabled'
+
+
+def test_disable_feed_404(installation, api_client, connect_auth_header):
+    response = api_client.post(
+        '/api/feeds/NOT-FOUND/disable',
+        installation=installation,
+        headers={'connect-auth': connect_auth_header},
+    )
+
+    assert response.status_code == 404
+    response_data = response.json()
+    assert response_data['error_code'] == 'NFND_000'
+    assert response_data['errors'][0] == 'Object `NOT-FOUND` not found.'
+
+
+def test_already_disabled_feed(
+    installation,
+    feed_factory,
+    api_client,
+    connect_auth_header,
+):
+    feed = feed_factory(account_id=installation['owner']['id'], status='disabled')
+    response = api_client.post(
+        f'/api/feeds/{feed.id}/disable',
+        installation=installation,
+        headers={'connect-auth': connect_auth_header},
+    )
+
+    assert response.status_code == 400
+    response_data = response.json()
+    assert response_data['error_code'] == 'RF_003'
+    assert response_data['errors'][0] == (
+        f'Feed `{feed.id}` is already in status `{feed.status}`.'
+    )