import unittest import pickle from nagparser.Model.NagCommands import NagCommands from nagfactoryTests import nagfromtestdata from expectedresults import expectedresultsfolder #Fully hydrated nag object with test data nag = nagfromtestdata() class BaseTests(unittest.TestCase): def test_commands_property_is_NagCommands_instance(self): self.assertTrue(isinstance(nag.commands, NagCommands)) def test_attributes_property_is_correct(self): with open(expectedresultsfolder + '/nag_attributes.pickle') as f: self.assertEqual(nag.attributes, pickle.load(f)) def test_getbad(self): print nag.status if __name__ == '__main__': unittest.main()
import pickle from nagfactoryTests import nagfromtestdata expectedresultsfolder = 'ExpectedResults' def buildexpectedresults(nag): with open(expectedresultsfolder + '/nag_attributes.pickle', 'w') as f: pickle.dump(nag.attributes, f) if __name__ == '__main__': #Fully hydrated nag object with test data buildexpectedresults(nagfromtestdata()) print 'done'