def test_on_companies_list_with_elements(self, urlopen): prepare_url_open( urlopen, {COMPANIES_URL: [{"permalink": "foo"}, {"permalink": "bar"}], FOO_URL: ["some_foo"], BAR_URL: ["some_bar"]}, ) self.assertEqual(self._call_cmd("-q", self.tmpd), 0) try: urlopen.assert_called_with(BAR_URL) except AssertionError: # pragma: no cover urlopen.assert_called_with(FOO_URL) listing = os.listdir(self.tmpd) listing.sort() self.assertEqual(listing, ["bar.json", "foo.json"]) with open(os.path.join(self.tmpd, "bar.json")) as fp: self.assertEqual(json.loads(fp.read()), ["some_bar"]) with open(os.path.join(self.tmpd, "foo.json")) as fp: self.assertEqual(json.loads(fp.read()), ["some_foo"])
def test_on_empty_companies_list(self, urlopen): url_return = {COMPANIES_URL: []} prepare_url_open(urlopen, url_return) self.assertEqual(self._call_cmd(self.tmpd), 0) urlopen.assert_called_once_with(COMPANIES_URL) self.assertEqual(os.listdir(self.tmpd), [])
def test_update(self, urlopen): prepare_url_open(urlopen, {COMPANIES_URL: [{"permalink": "foo"}]}) cl = self._make_one(None) cl.update() urlopen.assert_called_once_with(COMPANIES_URL)
def test_update(self, urlopen): prepare_url_open(urlopen, {FOO_URL: {"foo": "bar"}}) company = self._make_update() self.assertEqual(company.data, {"foo": "bar"}) urlopen.assert_called_once_with(FOO_URL)