Skip to content

Commit

Permalink
fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
janschumann committed Jan 2, 2019
1 parent 195bf2c commit 723f280
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
1 change: 1 addition & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ NEW FEATURES:
IMPROVEMENTS:

* Add account and environment as model properties
* fix tests

## 0.61.0

Expand Down
8 changes: 4 additions & 4 deletions test/test_lifecycle_handler.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,7 @@ def setUp(self):
repositories.add('node', NodeRepository)
repositories.add('command', CommandRepository)

self.model = MockModel(mock.Mock(), repositories, logging)
self.model = MockModel(mock.Mock(), repositories, logging, 'test', 'test')


def get_default_tansition_config(self):
Expand Down Expand Up @@ -1151,7 +1151,7 @@ def test_new_nodes_wait_for_cloud_init(self):
clients = mock.Mock()
clients.get.return_value = client
client.wait_for_scan_count_is = mock.MagicMock()
model = Model(clients, repositories, mock.Mock())
model = Model(clients, repositories, mock.Mock(), 'test', 'test')

event = get_event('autoscaling_event.json')
model.initialize(event)
Expand All @@ -1160,7 +1160,7 @@ def test_new_nodes_wait_for_cloud_init(self):


def test_scheduled_event_does_not_load_a_node_by_default(self):
model = MockModel(mock.Mock(), mock.Mock(), mock.Mock())
model = MockModel(mock.Mock(), mock.Mock(), mock.Mock(), 'test', 'test')
event = get_event('scheduled_event.json')
model.initialize(event)
model.transitions = self.get_before_node_load_tansition_config()
Expand All @@ -1173,7 +1173,7 @@ def test_scheduled_event_does_not_load_a_node_by_default(self):


def test_scheduled_event_can_load_node_during_transitions(self):
model = MockModel(mock.Mock(), mock.Mock(), mock.Mock())
model = MockModel(mock.Mock(), mock.Mock(), mock.Mock(), 'test', 'test')
event = get_event('scheduled_event.json')
model.initialize(event)
model.transitions = self.get_backup_tansition_config()
Expand Down

0 comments on commit 723f280

Please sign in to comment.