def test_delete(self):
# copy "0" test troll box to temp dir, delete from it
file_dir = os.path.dirname(os.path.realpath(__file__))
data_path = os.path.join(file_dir, "data", "0")
temp_dir = tempfile.mkdtemp()
temp_model_dir = os.path.join(temp_dir, "0")
sample_path = os.path.join(temp_model_dir, "images", "book.jpg")
try:
print data_path, temp_dir, sample_path
subprocess.check_call(["cp", "-R", data_path, temp_dir])
model = ImageModel(troll_dir=temp_model_dir)
self.assertEqual(model.data(model.index(0, 0), Qt.DisplayRole),
"http://foo.bar")
print subprocess.check_output(["file", sample_path])
self.assertTrue(os.path.exists(sample_path))
# make sure image has been removed from model and file is removed from disk
model.deleteImage(model.index(0, 0))
self.assertEqual(model.data(model.index(0, 0), Qt.DisplayRole),
"http://foo.baz")
self.assertFalse(os.path.exists(sample_path))
finally:
subprocess.call(["rm", "-rf", temp_dir])
def test_addImage(self):
# create a troll box in a temp dir and add an image from the "0" test
# data set
file_dir = os.path.dirname(os.path.realpath(__file__))
sample_path = os.path.join(file_dir, "data", "0", "images", "book.jpg")
temp_dir = tempfile.mkdtemp()
try:
model = ImageModel(troll_dir=temp_dir)
model.addImage("http://foo.bar", ["tng"], sample_path)
index = model.index(0, 0)
self.assertEqual(model.data(index, Qt.DisplayRole), "http://foo.bar")
finally:
subprocess.call(["rm", "-rf", temp_dir])
def test_load(self):
file_dir = os.path.dirname(os.path.realpath(__file__))
sample_path = os.path.join(file_dir, "data", "0")
model = ImageModel(troll_dir=sample_path)
index = model.index(0, 0)
self.assertEqual(model.data(index, Qt.DisplayRole), "http://foo.bar")
self.assertEqual(model.data(index, ImageModel.TagRole),
["tng", "book"])
index = model.index(1, 0)
self.assertEqual(model.data(index, ImageModel.TagRole),
["koop", "postmaster"])
def test_save(self):
# create a troll box in a temp dir and add an image from the "0" test
# data set, save it, and verify another model picks up the changes
file_dir = os.path.dirname(os.path.realpath(__file__))
sample_path = os.path.join(file_dir, "data", "0", "images", "book.jpg")
temp_dir = tempfile.mkdtemp()
try:
model = ImageModel(troll_dir=temp_dir)
model.addImage("http://foo.bar", ["tng"], sample_path)
model_b = ImageModel(troll_dir=temp_dir)
self.assertEqual(model_b.data(model.index(0, 0), Qt.DisplayRole),
"http://foo.bar")
finally:
subprocess.call(["rm", "-rf", temp_dir])
def __init__(self, parent=None, model_path=None):
QListView.__init__(self, parent)
self.setViewMode(QListView.IconMode)
self.setDragEnabled(False)
self.setAcceptDrops(False)
self.setIconSize(QSize(150, 150))
self.setSelectionMode(self.SingleSelection)
imageModel = ImageModel(parent, model_path)
proxyModel = ImageSearcher(parent)
proxyModel.setSourceModel(imageModel)
self.setModel(proxyModel)
imageModel.imageAdded.connect(self.handleImageAdded)
imageModel.rowsAboutToBeRemoved.connect(self.handlePreDelete)
def test_load(self):
file_dir = os.path.dirname(os.path.realpath(__file__))
sample_path = os.path.join(file_dir, "data", "0")
model = ImageModel(troll_dir=sample_path)
index = model.index(0, 0)
self.assertEqual(model.data(index, Qt.DisplayRole), "http://foo.bar")
self.assertEqual(model.data(index, ImageModel.TagRole), ["tng", "book"])
index = model.index(1, 0)
self.assertEqual(model.data(index, ImageModel.TagRole), ["koop", "postmaster"])
def test_delete(self):
# copy "0" test troll box to temp dir, delete from it
file_dir = os.path.dirname(os.path.realpath(__file__))
data_path = os.path.join(file_dir, "data", "0")
temp_dir = tempfile.mkdtemp()
temp_model_dir = os.path.join(temp_dir, "0")
sample_path = os.path.join(temp_model_dir, "images", "book.jpg")
try:
print data_path, temp_dir, sample_path
subprocess.check_call(["cp", "-R", data_path, temp_dir])
model = ImageModel(troll_dir=temp_model_dir)
self.assertEqual(model.data(model.index(0, 0), Qt.DisplayRole), "http://foo.bar")
print subprocess.check_output(["file", sample_path])
self.assertTrue(os.path.exists(sample_path))
# make sure image has been removed from model and file is removed from disk
model.deleteImage(model.index(0,0))
self.assertEqual(model.data(model.index(0, 0), Qt.DisplayRole), "http://foo.baz")
self.assertFalse(os.path.exists(sample_path))
finally:
subprocess.call(["rm", "-rf", temp_dir])
def test_setData(self):
# create a troll box in a temp dir, add an image from the "0" test
# data set, and modify it via setData
file_dir = os.path.dirname(os.path.realpath(__file__))
sample_path = os.path.join(file_dir, "data", "0", "images", "book.jpg")
temp_dir = tempfile.mkdtemp()
try:
model = ImageModel(troll_dir=temp_dir)
model.addImage("http://foo.bar", ["tng"], sample_path)
self.assertEqual(model.data(model.index(0, 0), Qt.DisplayRole),
"http://foo.bar")
model.setData(model.index(0, 0), "http://bar.bar", Qt.DisplayRole)
self.assertEqual(model.data(model.index(0, 0), Qt.DisplayRole),
"http://bar.bar")
sample_path = os.path.join(file_dir, "data", "0", "images",
"koop.jpg")
model.setData(model.index(0, 0), sample_path, Qt.DecorationRole)
# no good way to check icon currently
finally:
subprocess.call(["rm", "-rf", temp_dir])
def test_instantiation(self):
temp_dir = tempfile.mkdtemp()
model = ImageModel(troll_dir=temp_dir)
self.assertEqual(model.rowCount(), 0)
subprocess.call(["rm", "-rf", temp_dir])
def test_setData(self):
# create a troll box in a temp dir, add an image from the "0" test
# data set, and modify it via setData
file_dir = os.path.dirname(os.path.realpath(__file__))
sample_path = os.path.join(file_dir, "data", "0", "images", "book.jpg")
temp_dir = tempfile.mkdtemp()
try:
model = ImageModel(troll_dir=temp_dir)
model.addImage("http://foo.bar", ["tng"], sample_path)
self.assertEqual(model.data(model.index(0, 0), Qt.DisplayRole), "http://foo.bar")
model.setData(model.index(0, 0), "http://bar.bar", Qt.DisplayRole)
self.assertEqual(model.data(model.index(0, 0), Qt.DisplayRole), "http://bar.bar")
sample_path = os.path.join(file_dir, "data", "0", "images", "koop.jpg")
model.setData(model.index(0, 0), sample_path, Qt.DecorationRole)
# no good way to check icon currently
finally:
subprocess.call(["rm", "-rf", temp_dir])
def test_instantiation(self):
temp_dir = tempfile.mkdtemp()
model = ImageModel(troll_dir=temp_dir)
self.assertEqual(model.rowCount(), 0)
subprocess.call(["rm", "-rf", temp_dir])
