Revert "Tidy up mocking code (#438)[First commit]"

This reverts commit 8ebb4d35fa.
This commit is contained in:
Sam Manzi 2017-09-23 07:50:48 +10:00
parent f31caf1ea4
commit 00f8df60ce
3 changed files with 17 additions and 14 deletions

View File

@ -29,7 +29,7 @@ Make an 'Unknown' primary object
# Python modules # Python modules
# #
#------------------------------------------------------------------------- #-------------------------------------------------------------------------
from time import strftime, time from time import strftime, localtime, time
import os import os
#------------------------------------------------------------------------- #-------------------------------------------------------------------------
@ -147,7 +147,7 @@ def make_unknown(class_arg, explanation, class_func, commit_func, transaction,
if not hasattr(make_unknown, 'count'): if not hasattr(make_unknown, 'count'):
make_unknown.count = 1 #primitive static variable make_unknown.count = 1 #primitive static variable
obj.set_name(_("Unknown, was missing %(time)s (%(count)d)") % { obj.set_name(_("Unknown, was missing %(time)s (%(count)d)") % {
'time': strftime('%x %X'), 'time': strftime('%x %X', localtime()),
'count': make_unknown.count}) 'count': make_unknown.count})
make_unknown.count += 1 make_unknown.count += 1
else: else:
@ -167,7 +167,7 @@ def create_explanation_note(dbase):
""" """
note = Note( _('Objects referenced by this note ' note = Note( _('Objects referenced by this note '
'were missing in a file imported on %s.') % 'were missing in a file imported on %s.') %
strftime('%x %X')) strftime('%x %X', localtime()))
note.set_handle(create_id()) note.set_handle(create_id())
note.set_gramps_id(dbase.find_next_note_gramps_id()) note.set_gramps_id(dbase.find_next_note_gramps_id())
# Use defaults for privacy, format and type. # Use defaults for privacy, format and type.

View File

@ -213,7 +213,7 @@ class ProgenOptions(ManagedWindow):
# initial values # initial values
text = "Pro-Gen Import" text = "Pro-Gen Import"
fname = os.path.basename(filename).split('\\')[-1] fname = os.path.basename(filename).split('\\')[-1]
date = time.strftime("%Y-%m-%d") date = time.strftime('%Y-%m-%d', time.localtime())
# add import source title/confidence # add import source title/confidence
# citation page/confidence/privacy/attribute # citation page/confidence/privacy/attribute

View File

@ -26,7 +26,7 @@ import os
import sys import sys
import re import re
import locale import locale
from time import strptime, strftime from time import localtime, strptime
from unittest.mock import patch from unittest.mock import patch
#import logging #import logging
@ -65,12 +65,6 @@ def mock_localtime(*args):
""" """
return strptime("25 Dec 1999", "%d %b %Y") return strptime("25 Dec 1999", "%d %b %Y")
def mock_strftime(*args):
"""
Mock up a dummy to replace the varying 'time string results'
"""
return strftime(args[0], (1999, 12, 25, 0, 0, 0, 5, 359, -1))
class TestImports(unittest.TestCase): class TestImports(unittest.TestCase):
"""The test class cases will be dynamically created at import time from """The test class cases will be dynamically created at import time from
files to be tested. The following defs are used by the test cases files to be tested. The following defs are used by the test cases
@ -199,12 +193,21 @@ def make_tst_function(tstfile, file_name):
""" This is here to support the dynamic function creation. This creates """ This is here to support the dynamic function creation. This creates
the test function (a method, to be precise). the test function (a method, to be precise).
""" """
@patch('gramps.gen.utils.unknown.strftime', side_effect=mock_strftime)
@patch('time.strftime', side_effect=mock_strftime) @patch('gramps.plugins.db.dbapi.dbapi.time')
def tst(self, mstrftime1, mstrftime2): @patch('gramps.plugins.db.bsddb.write.time')
@patch('gramps.gen.utils.unknown.localtime')
@patch('gramps.gen.utils.unknown.time')
@patch('time.localtime')
def tst(self, mockptime, mocktime, mockltime, mockwtime, mockdtime):
""" This compares the import file with the expected result '.gramps' """ This compares the import file with the expected result '.gramps'
file. file.
""" """
mockptime.side_effect = mock_localtime
mocktime.side_effect = mock_time
mockltime.side_effect = mock_localtime
mockwtime.side_effect = mock_time
mockdtime.side_effect = mock_time
fn1 = os.path.join(TEST_DIR, tstfile) fn1 = os.path.join(TEST_DIR, tstfile)
fn2 = os.path.join(TEST_DIR, (file_name + ".gramps")) fn2 = os.path.join(TEST_DIR, (file_name + ".gramps"))
fres = os.path.join(TEMP_DIR, (file_name + ".difs")) fres = os.path.join(TEMP_DIR, (file_name + ".difs"))