def test_get_attributes_success(self, m_repository):
cfg = {'return_value.read.return_value': random_string()}
m_repository.return_value.get_file.configure_mock(**cfg)
controller = ztpserver.controller.NodesController()
(resp, state) = controller.get_attributes(dict(),
resource=random_string())
self.assertEqual(state, 'do_substitution')
self.assertIsInstance(resp, dict)
def test_get_attributes_success(self, m_repository):
cfg = {'return_value.read.return_value': random_string()}
m_repository.return_value.get_file.configure_mock(**cfg)
controller = ztpserver.controller.NodesController()
(resp, state) = controller.get_attributes(dict(),
resource=random_string())
self.assertEqual(state, 'do_substitution')
self.assertIsInstance(resp, dict)
def test_get_attributes_missing(self, m_repository):
node = create_node()
m_repository.return_value.exists.return_value = False
m_repository.return_value.get_file = \
Mock(side_effect=ztpserver.repository.FileObjectNotFound)
controller = ztpserver.controller.NodesController()
(resp, state) = controller.get_attributes(dict(),
resource=node.serialnumber)
self.assertEqual(state, 'do_substitution')
self.assertIsInstance(resp, dict)
self.assertEqual(resp['attributes'], dict())
def test_get_attributes_missing(self, m_repository):
node = create_node()
m_repository.return_value.exists.return_value = False
m_repository.return_value.get_file = \
Mock(side_effect=ztpserver.repository.FileObjectNotFound)
controller = ztpserver.controller.NodesController()
(resp, state) = controller.get_attributes(dict(),
resource=node.serialnumber)
self.assertEqual(state, 'do_substitution')
self.assertIsInstance(resp, dict)
self.assertEqual(resp['attributes'], dict())
def test_get_attributes_success(self):
systemmac = random_string()
contents = random_string()
filestore = Mock()
filestore.return_value.get_file.return_value = Mock(contents=contents)
ztpserver.controller.create_file_store = filestore
controller = ztpserver.controller.NodesController()
(resp, state) = controller.get_attributes(dict(), resource=systemmac)
filepath = '%s/%s' % (systemmac, ztpserver.controller.ATTRIBUTES_FN)
filestore.return_value.get_file.assert_called_with(filepath)
self.assertEqual(state, 'do_substitution')
self.assertIsInstance(resp, dict)
self.assertEqual(resp['attributes'], contents)