def test_add_note_to_xml(self, dt_stub, xml_mock):
# Arrange
value = 'This is a note!'
path = '/blah/temp.xml'
userName = '******'
dt_now_stub = MagicMock(name='dt_now_stub')
dt_now_stub.day = 9
dt_now_stub.month = 9
dt_now_stub.year = 9
fakeXmlFile = MagicMock(name='fakeXmlFile')
xml_mock().readXml.return_value = fakeXmlFile
dt_stub.now.return_value = dt_now_stub
# Act
inst = SetData.SetData()
result = inst._add_note_to_xml(value=value,
filePath=path,
userName=userName)
# Assert
self.assertTrue(result)
self.assertTrue(call().readXml(path) in xml_mock.mock_calls)
self.assertTrue(
call().saveXml(path, fakeXmlFile) in xml_mock.mock_calls)
def test_setValue_2(self, cmfe_stub, xml_mock, check_value_stub,
set_value_mock):
# Arrange
userName = '******'
typ = 'status'
value = 'ip'
path = '/blah/temp.xml'
fake_xml = MagicMock(name='xmlFile')
fake_xml.getElementsByTagName.return_value = ['']
xml_mock.readXml.return_value = fake_xml
xml_mock.getTagValue.return_value = 'msantos'
cmfe_stub.return_value = True
check_value_stub.return_value = True
set_value_mock.return_value = True
# Act
inst = SetData.SetData()
result = inst.setValue(typ=typ,
value=value,
filePath=path,
userName=userName)
# Assert
self.assertEqual(result, 'Added status:ip to meta file')
self.assertTrue(
call('status', 'ip', '/blah/temp.xml') in
set_value_mock.mock_calls)
self.assertTrue(
call().readXml('/blah/temp.xml') in xml_mock.mock_calls)
self.assertTrue(call().readXml().getElementsByTagName('status') in
xml_mock.mock_calls)
def test_check_existing_value(self):
currentValue = 'msantos'
value = 'jcalduch'
inst = SetData.SetData()
result = inst._check_existing_value(value=value,
currentValue=currentValue)
self.assertTrue(result)
#----
currentValue = 'msantos'
value = 'msantos'
inst = SetData.SetData()
result = inst._check_existing_value(value=value,
currentValue=currentValue)
self.assertFalse(result)
def _status_labels(self, name=None, statusGroupBox=None, hl=None, path=None):
""" """
#read xml if it exists and set status
# model_info.xml concept_info.xml texture_info.xml rig_info.xml
#assetInfoFiles = ['model_info.xml','concept_info.xml','texture_info.xml','rig_info.xml']
if 'animation' in path:
pathToXml = os.path.join(path, 'meta','anim_info.xml')
else:
pathToXml = os.path.join(path, 'meta', name + '_info.xml')
#setValue(typ = None, value = None, filePath = None, userName = None ):
#SetData.SetData().setValue('status', value, pathToXml, 'msantos' )
sd = SetData.SetData().setValue
scMethod = self._status_color
optList = [ ['wts',[sd,['status', 'wts', pathToXml, 'msantos',scMethod]]],
['ip',[sd,['status', 'ip', pathToXml, 'msantos',scMethod ]]],
['review',[sd,['status', 'review', pathToXml, 'msantos',scMethod ]]],
['final',[sd,['status', 'final', pathToXml, 'msantos',scMethod ]]],
['hold',[sd,['status', 'hold', pathToXml, 'msantos',scMethod ]]]
]
if os.path.exists(pathToXml):
xmlObj = tmXml.tmXml()
xmlFile = xmlObj.readXml(pathToXml)
status = xmlObj.getTagValueByTagName(xmlFile, 'status')
sColor = self._status_color(status)
statusLabel = ExtendedLabel.ExtendedLabel( statusGroupBox, optList )
statusLabel.setCursor(QtGui.QCursor(QtCore.Qt.PointingHandCursor))
statusLabel.setAutoFillBackground(False)
statusLabel.setStyleSheet("QLabel { background-color: rgb("+sColor+") } ")
if 'animation' in path:
statusLabel.setText(" ")
else:
statusLabel.setText(" %s"%name[0].title())
#statusLabel.setPixmap(QtGui.QPixmap("../graphics/png/status_open_small.png"))
#statusLabel.setScaledContents(True)
statusLabel.setObjectName(name)
hl.addWidget(statusLabel)
else:
# file could not be found
statusLabel = ExtendedLabel.ExtendedLabel( statusGroupBox, optList )
statusLabel.setCursor(QtGui.QCursor(QtCore.Qt.PointingHandCursor))
statusLabel.setAutoFillBackground(False)
statusLabel.setStyleSheet("background-color: rgb(10,10,10);")
statusLabel.setText("")
#statusLabel.setPixmap(QtGui.QPixmap("../graphics/png/status_open_small.png"))
#statusLabel.setScaledContents(True)
statusLabel.setObjectName(name)
hl.addWidget(statusLabel)
return False
def test_check_metaFile_exists(self, os_mock):
#------------------------ Arrange
path = '/temp/poo'
os_mock.return_value = True
# Act
inst = SetData.SetData()
result = inst._check_metaFile_exists(filePath=path)
# Assert
self.assertEqual(result, True)
self.assertTrue(call(path) in os_mock.mock_calls)
def get_data(self, user, password, table):
"""
Get the u-i rate matrix from database
:param user: user id
:param password: user password
:param table: the table id
:return:
"""
self.my_connect = SetData.GetData(self.host, self.database,
self.charset)
self.my_connect.connect(user, password)
self.my_connect.select("SELECT * FROM {}".format(table))
self.result = self.my_connect.result
def test_set_value_to_xml(self, xml_mock):
# Arrange
typ = 'assignee'
value = 'msantos'
path = '/blah/temp.xml'
fakeXmlFile = MagicMock(name='fakeXmlfile')
fakeXmlFile.toxml.return_value = 'Front<assignee></assignee>Back'
xml_mock().readXml.return_value = fakeXmlFile
# Act
inst = SetData.SetData()
result = inst._set_value_to_xml(typ=typ, value=value, filePath=path)
# Assert
self.assertTrue(result)
self.assertTrue(
call().saveXml(path, 'Front<assignee>msantos</assignee>Back') in
xml_mock.mock_calls)
self.assertTrue(call().readXml(path) in xml_mock.mock_calls)
def _writeXmlData(self, xmlFile=None):
''' Write data to xml files for asset '''
parent = self.parent().parent().parent().parent().parent().parent(
).parent().parent()
project = str(parent.projComboBox.currentText())
user = None #str(parent.userComboBox.currentText())
name = str(self.nameLineEdit.text())
typ = str(self.typeComboBox.currentText())
sd = SetData.SetData()
sd.setValue(typ='project',
value=project,
filePath=xmlFile,
userName=user)
sd.setValue(typ='assignee',
value=user,
filePath=xmlFile,
userName=user)
sd.setValue(typ='type', value=typ, filePath=xmlFile, userName=user)
sd.setValue(typ='name', value=name, filePath=xmlFile, userName=user)
sd.setValue(typ='status', value='wts', filePath=xmlFile, userName=user)
# from pydriller import *
from git import *
import os
import shutil
import stat
import csv
import re
import subprocess
import SetData as sd
import parser as pa
pa.run()
filename, file_dict, file_hash = sd.InitData()
cnt = 0
# File is cloned based on url.
with open(filename, 'r') as txtinput:
all_lines = txtinput.readlines()
all_num = len(all_lines)
for url_data in all_lines:
cnt += 1
try:
if cnt == all_num:
Git('clone_data').clone(url_data)
else:
url_data = url_data[:len(url_data) - 1]
Git('clone_data').clone(url_data)
except Exception as e: