
from typing import Optional

from stoq.data_classes import ArchiverResponse, Payload, RequestMeta
from stoq.plugins import ArchiverPlugin


class DummyArchiver(ArchiverPlugin):
    def archive(
        self, payload: Payload, request_meta: RequestMeta
    ) -> Optional[ArchiverResponse]:
        return None

    def get(self, task: str) -> Optional[Payload]:
        return None


from typing import Optional

from stoq.plugins import ArchiverPlugin
from stoq.data_classes import ArchiverResponse, Payload, RequestMeta


class DummyArchiver(ArchiverPlugin):
    def archive(
        self, payload: Payload, request_meta: RequestMeta
    ) -> Optional[ArchiverResponse]:
        return None

    def get(self, task: ArchiverResponse) -> Optional[Payload]:
        return None

Fix test signature value type for task
Fix test signature value type for task

  
  from typing import Optional
  
+ from stoq.plugins import ArchiverPlugin
  from stoq.data_classes import ArchiverResponse, Payload, RequestMeta
- from stoq.plugins import ArchiverPlugin
  
  
  class DummyArchiver(ArchiverPlugin):
      def archive(
          self, payload: Payload, request_meta: RequestMeta
      ) -> Optional[ArchiverResponse]:
          return None
  
-     def get(self, task: str) -> Optional[Payload]:
+     def get(self, task: ArchiverResponse) -> Optional[Payload]:
          return None
  
Fix test signature value type for task
## Code Before:

from typing import Optional

from stoq.data_classes import ArchiverResponse, Payload, RequestMeta
from stoq.plugins import ArchiverPlugin


class DummyArchiver(ArchiverPlugin):
    def archive(
        self, payload: Payload, request_meta: RequestMeta
    ) -> Optional[ArchiverResponse]:
        return None

    def get(self, task: str) -> Optional[Payload]:
        return None

## Instruction:
Fix test signature value type for task
## Code After:

from typing import Optional

from stoq.plugins import ArchiverPlugin
from stoq.data_classes import ArchiverResponse, Payload, RequestMeta


class DummyArchiver(ArchiverPlugin):
    def archive(
        self, payload: Payload, request_meta: RequestMeta
    ) -> Optional[ArchiverResponse]:
        return None

    def get(self, task: ArchiverResponse) -> Optional[Payload]:
        return None
