Skip to content

Commit 27562cd

Browse files
committed
Update unit tests
1 parent 56fabd9 commit 27562cd

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

test/unit/test_upgrade.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
import pytest
2323
import mox
2424

25+
import copy
2526
import os
2627
import shutil
2728

@@ -44,6 +45,9 @@ def __init__(self, args):
4445
for k, v in args.iteritems():
4546
setattr(self, k, v)
4647

48+
def __eq__(self, o):
49+
return self.__dict__ == o.__dict__
50+
4751
class PartialMockUnixInstall(UnixInstall):
4852

4953
def __init__(self, args, ext):
@@ -90,7 +94,10 @@ def test_get_server_dir(self, server):
9094
).AndReturn('server')
9195
expected = 'server'
9296
else:
93-
omego.upgrade.Artifacts(args).AndReturn(self.MockArtifacts())
97+
artifact_args = copy.copy(args)
98+
artifact_args.sym = ''
99+
omego.upgrade.Artifacts(artifact_args).AndReturn(
100+
self.MockArtifacts())
94101
expected = 'server-dir'
95102

96103
self.mox.ReplayAll()

0 commit comments

Comments
 (0)