import sys
import json

from testwizard.commands_core import CommandBase
from .SaveFileResult import SaveFileResult


class SaveReferenceBitmapCommand(CommandBase):
    def __init__(self, testObject):
        CommandBase.__init__(self, testObject, "SaveReferenceBitmap")

    def execute(self, filename):
        requestObj = [filename]

        result = self.executeCommand(requestObj)

        return SaveFileResult(result, "SaveReferenceBitmap was successful", "SaveReferenceBitmap failed")
