Update tests for previous commit

This commit is contained in:
gandalf3
2017-07-06 22:12:43 -07:00
parent 0f96cd4852
commit 6db5a2d190
2 changed files with 16 additions and 16 deletions

View File

@@ -7,7 +7,7 @@ import ast
import json import json
import make_repo import make_repo
logging.basicConfig(level=logging.DEBUG, logging.basicConfig(level=logging.ERROR,
format='%(levelname)8s: %(message)s') format='%(levelname)8s: %(message)s')
class TestRepoGeneration(unittest.TestCase): class TestRepoGeneration(unittest.TestCase):

View File

@@ -1,7 +1,7 @@
import requests import requests
import unittest import unittest
from unittest import mock from unittest import mock
from blenderpack import fetch_repo, load_repo, write_repo # from blenderpack import Repositories, fetch_repo
from datetime import datetime from datetime import datetime
import json import json
@@ -39,29 +39,29 @@ def mocked_requests_get(*args, **kwargs):
return MockResponse(None, 404) return MockResponse(None, 404)
_mocked_repo_storage = {} class MockRepositories:
def mocked_load_repo(*args, **kwargs): storage = {}
global _mocked_repo_storage
if args[0] not in _mocked_repo_storage:
_mocked_repo_storage[args[0]] = {'url': args[0]}
return _mocked_repo_storage[args[0]] def load(self, *args, **kwargs):
if args[0] not in self.storage:
self.storage[args[0]] = {'url': args[0]}
def mocked_write_repo(*args, **kwargs): return self.storage[args[0]]
global _mocked_repo_storage
_mocked_repo_storage[args[0]['url']] = args[0] def write(self, *args, **kwargs):
self.storage[args[0]['url']] = args[0]
class fetch_url_twice(unittest.TestCase): class fetch_url_twice(unittest.TestCase):
@mock.patch('requests.get', side_effect=mocked_requests_get) @mock.patch('requests.get', side_effect=mocked_requests_get)
@mock.patch('blenderpack.load_repo', side_effect=mocked_load_repo) def test_fetch(self, mock_get):
@mock.patch('blenderpack.write_repo', side_effect=mocked_write_repo) self.fail('unfinished test')
def test_fetch(self, mock_write, mock_load, mock_get): repos = MockRepositories()
fetch_repo('http://someurl.tld/repo.json') fetch_repo('http://someurl.tld/repo.json', repos)
mock_get.assert_called_with('http://someurl.tld/repo.json', headers={}) mock_get.assert_called_with('http://someurl.tld/repo.json', headers={})
fetch_repo('http://someurl.tld/repo.json') fetch_repo('http://someurl.tld/repo.json', repos)
mock_get.assert_called_with('http://someurl.tld/repo.json', headers={ mock_get.assert_called_with('http://someurl.tld/repo.json', headers={
'If-None-Match': '"2a0094b-b74-55326ced274f3"', 'If-None-Match': '"2a0094b-b74-55326ced274f3"',
'If-Modified-Since': 'Sun, 13 Mar 2011 13:38:53 GMT' 'If-Modified-Since': 'Sun, 13 Mar 2011 13:38:53 GMT'