resources/tools/odltools/mdsal/tests/test_models.py

   1 import logging
   2 import os
   3 import shutil
   4 import unittest
   5 from mdsal import models
   6 from odltools import logg
   7
   8
   9 class TestModels(unittest.TestCase):
  10     def setUp(self):
  11         logg.Logger(logging.DEBUG, logging.DEBUG)
  12         self.path = "/tmp/testmodels"
  13         self.ip = "127.0.0.1"
  14         self.port = "8181"
  15         self.user = "admin"
  16         self.pw = "admin"
  17
  18     def test_get_all_dumps(self):
  19         shutil.rmtree(self.path)
  20         models.get_all_dumps(self.path, self.ip, self.port, self.user, self.pw, True)
  21
  22         # assert each model has been saved to a file
  23         for res in models.DSMAP.itervalues():
  24             store = res[models.DSM_DSTYPE]
  25             model_path = res[models.DSM_PATH]
  26             path_split = models.split_model_path(model_path)
  27             filename = models.make_filename(self.path, store, path_split.name, path_split.container)
  28             self.assertTrue(os.path.isfile(filename))
  29
  30
  31 if __name__ == '__main__':
  32     unittest.main()