def test_load_initial_data(self, mock_glob, mock_mgmt): fake_glob = { 'root/quark/foo/fixtures/*.yaml': ['file1'], 'root/thirdparty/bar/fixtures/*.yaml': ['file2', 'file3'], } mock_glob.side_effect = lambda x: fake_glob[x] mock_mgmt.execute.side_effect = lambda arg: self.assertEqual( arg, ['manage.py', 'loaddata', 'file1', 'file2', 'file3']) dev_utils.load_initial_data() self.assertEqual(mock_glob.call_count, 2)
def test_no_initial_data(self, mock_mgmt): """Don't run loaddata with no available fixtures""" dev_utils.load_initial_data() self.assertFalse(mock_mgmt.called)