from django.test import TestCase
from ...models import StepBuild
from unittest import mock
import json


class TestStepBuild(TestCase):

    def test_step_build_passed(self):
        passed_dict = {"result": {"status": "passed"}}
        failed_dict = {"result": {"status": "failed"}}
        step_build = mock.Mock(spec=StepBuild)

        # passing step returns True
        step_build.json = json.dumps(passed_dict)
        self.assertEqual(StepBuild.passed(step_build), True)

        # failing step returns False
        step_build.json = json.dumps(failed_dict)
        self.assertEqual(StepBuild.passed(step_build), False)

    def test_step_build_duration(self):
        step_dict = {"result": {"duration": 123000000000}}
        step_build = mock.Mock(spec=StepBuild)

        # duration is 123
        step_build.json = json.dumps(step_dict)
        self.assertEqual(StepBuild.duration(step_build), 123)