test_jira.py

import pytest
import json
from mock import patch, call
import requests
from botbot_plugins.base import DummyApp
from botbot_plugins.plugins import jira

class FakeProjectResponse(object):
    """Dummy response from JIRA"""
    status_code = 200
    text = json.dumps([{'key': 'TEST'}])    

class FakeUserResponse1(object):
    """Dummy response from JIRA"""
    status_code = 200
    text = json.dumps({'key': 'TEST-123', 'fields': {'summary': "Testing JIRA plugin"}})    

class FakeUserResponse2(object):
    """Dummy response from JIRA"""
    status_code = 200
    text = json.dumps({'key': 'TEST-234', 'fields': {'summary': "Testing JIRA plugin"}})    

@pytest.fixture
def app():
    dummy_app = DummyApp(test_plugin=jira.Plugin())
    dummy_app.set_config('jira', {'jira_url': 'https://tickets.test.org', 'bot_name': 'testbot'})
    return dummy_app


def test_jira(app):
    # patch requests.get so we don't need to make a real call to Jira

    # Test project retrival
    with patch.object(requests, 'get') as mock_get:
        mock_get.return_value = FakeProjectResponse()
        responses = app.respond("@UPDATE:JIRA")
        mock_get.assert_called_with(
            'https://tickets.test.org/rest/api/2/project')
        assert responses == ["Successfully updated projects list"]

    # Test appropriate response       
    with patch.object(requests, 'get') as mock_get:
        mock_get.return_value = FakeUserResponse1()
        responses = app.respond("I just assigned TEST-123 to testuser")
        mock_get.assert_called_with(
            'https://tickets.test.org/rest/api/2/issue/TEST-123')
        assert responses == ["TEST-123: Testing JIRA plugin https://tickets.test.org/browse/TEST-123"]

    # Test response when issue is mentioned as part of url
    with patch.object(requests, 'get') as mock_get:
        mock_get.return_value = FakeUserResponse1()
        responses = app.respond("Check out https://tickets.test.org/browse/TEST-123")
        mock_get.assert_called_with(
            'https://tickets.test.org/rest/api/2/issue/TEST-123')
        assert responses == ["TEST-123: Testing JIRA plugin"]