def test_with_gear_and_two_non_unique_listings_on_different_gear_in_right_country(self): listing = EquipmentGenerators.equipment_item_listing() listing.retailer.countries = "us" listing.retailer.save() telescope = Generators.telescope() telescope.equipment_item_listings.add(listing) mount = Generators.mount() mount.equipment_item_listings.add(listing) image = Generators.image() image.imaging_telescopes.add(telescope) image.mounts.add(mount) self.assertEquals(1, unique_equipment_item_listings(image, 'us').count())
def test_with_gear_and_two_unique_listings_on_different_gear_from_same_retailer_in_right_country(self): listing1 = EquipmentGenerators.equipment_brand_listing() listing1.retailer.countries = "us" listing1.retailer.save() listing2 = EquipmentGenerators.equipment_brand_listing() listing2.retailer = listing1.retailer listing2.save() telescope = Generators.telescope() telescope.equipment_brand_listings.add(listing1) mount = Generators.mount() mount.equipment_brand_listings.add(listing2) image = Generators.image() image.imaging_telescopes.add(telescope) image.mounts.add(mount) self.assertEqual(2, unique_equipment_brand_listings(image, 'us').count())
def test_get_equipment_list_when_mix(self): image = Generators.image() image.imaging_telescopes.add( Generators.telescope(make='My make', name='My imaging telescope')) image.imaging_cameras.add( Generators.camera(make='My make', name='My imaging camera')) image.mounts.add(Generators.mount(make='My make', name='My mount')) image.filters.add(Generators.filter(make='My make', name='My filter')) image.accessories.add( Generators.accessory(make='My make', name='My accessory')) image.focal_reducers.add( Generators.focal_reducer(make='My make', name='My focal reducer')) image.software.add( Generators.software(make='My make', name='My software')) image.guiding_telescopes.add( Generators.telescope(make='My make', name='My guiding telescope')) image.guiding_cameras.add( Generators.camera(make='My make', name='My guiding camera')) brand = EquipmentGenerators.brand(name='My brand') image.imaging_telescopes_2.add( EquipmentGenerators.telescope(brand=brand, name='My new imaging telescope')) image.imaging_cameras_2.add( EquipmentGenerators.camera(brand=brand, name='My new imaging camera')) image.mounts_2.add( EquipmentGenerators.mount(brand=brand, name='My new mount')) image.filters_2.add( EquipmentGenerators.filter(brand=brand, name='My new filter')) image.accessories_2.add( EquipmentGenerators.accessory(brand=brand, name='My new accessory')) image.software_2.add( EquipmentGenerators.software(brand=brand, name='My new software')) image.guiding_telescopes_2.add( EquipmentGenerators.telescope(brand=brand, name='My new guiding telescope')) image.guiding_cameras_2.add( EquipmentGenerators.camera(brand=brand, name='My new guiding camera')) equipment_list = ImageService(image).get_equipment_list() self.assertEqual(2, len(equipment_list['imaging_telescopes'])) self.assertEqual(2, len(equipment_list['imaging_cameras'])) self.assertEqual(2, len(equipment_list['mounts'])) self.assertEqual(2, len(equipment_list['filters'])) self.assertEqual(3, len(equipment_list['accessories'])) self.assertEqual(2, len(equipment_list['software'])) self.assertEqual(2, len(equipment_list['guiding_telescopes'])) self.assertEqual(2, len(equipment_list['guiding_cameras'])) self.assertTrue('id' in equipment_list['imaging_telescopes'][0]) self.assertEqual('My brand My new imaging telescope', equipment_list['imaging_telescopes'][0]['label']) self.assertEqual('NEW', equipment_list['imaging_telescopes'][0]['version']) self.assertTrue('id' in equipment_list['imaging_cameras'][0]) self.assertEqual('My brand My new imaging camera', equipment_list['imaging_cameras'][0]['label']) self.assertEqual('NEW', equipment_list['imaging_cameras'][0]['version']) self.assertTrue('id' in equipment_list['mounts'][0]) self.assertEqual('My brand My new mount', equipment_list['mounts'][0]['label']) self.assertEqual('NEW', equipment_list['mounts'][0]['version']) self.assertTrue('id' in equipment_list['filters'][0]) self.assertEqual('My brand My new filter', equipment_list['filters'][0]['label']) self.assertEqual('NEW', equipment_list['filters'][0]['version']) self.assertTrue('id' in equipment_list['accessories'][0]) self.assertEqual('My brand My new accessory', equipment_list['accessories'][0]['label']) self.assertEqual('NEW', equipment_list['accessories'][0]['version']) self.assertTrue('id' in equipment_list['software'][0]) self.assertEqual('My brand My new software', equipment_list['software'][0]['label']) self.assertEqual('NEW', equipment_list['software'][0]['version']) self.assertTrue('id' in equipment_list['guiding_telescopes'][0]) self.assertEqual('My brand My new guiding telescope', equipment_list['guiding_telescopes'][0]['label']) self.assertEqual('NEW', equipment_list['guiding_telescopes'][0]['version']) self.assertTrue('id' in equipment_list['guiding_cameras'][0]) self.assertEqual('My brand My new guiding camera', equipment_list['guiding_cameras'][0]['label']) self.assertEqual('NEW', equipment_list['guiding_cameras'][0]['version']) self.assertTrue('id' in equipment_list['imaging_telescopes'][1]) self.assertEqual('My make My imaging telescope', equipment_list['imaging_telescopes'][1]['label']) self.assertEqual('LEGACY', equipment_list['imaging_telescopes'][1]['version']) self.assertTrue('id' in equipment_list['imaging_cameras'][1]) self.assertEqual('My make My imaging camera', equipment_list['imaging_cameras'][1]['label']) self.assertEqual('LEGACY', equipment_list['imaging_cameras'][1]['version']) self.assertTrue('id' in equipment_list['mounts'][1]) self.assertEqual('My make My mount', equipment_list['mounts'][1]['label']) self.assertEqual('LEGACY', equipment_list['mounts'][1]['version']) self.assertTrue('id' in equipment_list['filters'][1]) self.assertEqual('My make My filter', equipment_list['filters'][1]['label']) self.assertEqual('LEGACY', equipment_list['filters'][1]['version']) self.assertTrue('id' in equipment_list['accessories'][1]) self.assertEqual('My make My accessory', equipment_list['accessories'][1]['label']) self.assertEqual('LEGACY', equipment_list['accessories'][1]['version']) self.assertTrue('id' in equipment_list['accessories'][2]) self.assertEqual('My make My focal reducer', equipment_list['accessories'][2]['label']) self.assertEqual('LEGACY', equipment_list['accessories'][2]['version']) self.assertTrue('id' in equipment_list['software'][1]) self.assertEqual('My make My software', equipment_list['software'][1]['label']) self.assertEqual('LEGACY', equipment_list['software'][1]['version']) self.assertTrue('id' in equipment_list['guiding_telescopes'][1]) self.assertEqual('My make My guiding telescope', equipment_list['guiding_telescopes'][1]['label']) self.assertEqual('LEGACY', equipment_list['guiding_telescopes'][1]['version']) self.assertTrue('id' in equipment_list['guiding_cameras'][1]) self.assertEqual('My make My guiding camera', equipment_list['guiding_cameras'][1]['label']) self.assertEqual('LEGACY', equipment_list['guiding_cameras'][1]['version'])
def test_get_equipment_list_when_only_legacy_gear(self): image = Generators.image() image.imaging_telescopes.add( Generators.telescope(make='My make', name='My imaging telescope')) image.imaging_cameras.add( Generators.camera(make='My make', name='My imaging camera')) image.mounts.add(Generators.mount(make='My make', name='My mount')) image.filters.add(Generators.filter(make='My make', name='My filter')) image.accessories.add( Generators.accessory(make='My make', name='My accessory')) image.focal_reducers.add( Generators.focal_reducer(make='My make', name='My focal reducer')) image.software.add( Generators.software(make='My make', name='My software')) image.guiding_telescopes.add( Generators.telescope(make='My make', name='My guiding telescope')) image.guiding_cameras.add( Generators.camera(make='My make', name='My guiding camera')) equipment_list = ImageService(image).get_equipment_list() self.assertEqual(1, len(equipment_list['imaging_telescopes'])) self.assertEqual(1, len(equipment_list['imaging_cameras'])) self.assertEqual(1, len(equipment_list['mounts'])) self.assertEqual(1, len(equipment_list['filters'])) self.assertEqual(2, len(equipment_list['accessories'])) self.assertEqual(1, len(equipment_list['software'])) self.assertEqual(1, len(equipment_list['guiding_telescopes'])) self.assertEqual(1, len(equipment_list['guiding_cameras'])) self.assertTrue('id' in equipment_list['imaging_telescopes'][0]) self.assertEqual('My make My imaging telescope', equipment_list['imaging_telescopes'][0]['label']) self.assertEqual('LEGACY', equipment_list['imaging_telescopes'][0]['version']) self.assertTrue('id' in equipment_list['imaging_cameras'][0]) self.assertEqual('My make My imaging camera', equipment_list['imaging_cameras'][0]['label']) self.assertEqual('LEGACY', equipment_list['imaging_cameras'][0]['version']) self.assertTrue('id' in equipment_list['mounts'][0]) self.assertEqual('My make My mount', equipment_list['mounts'][0]['label']) self.assertEqual('LEGACY', equipment_list['mounts'][0]['version']) self.assertTrue('id' in equipment_list['filters'][0]) self.assertEqual('My make My filter', equipment_list['filters'][0]['label']) self.assertEqual('LEGACY', equipment_list['filters'][0]['version']) self.assertTrue('id' in equipment_list['accessories'][0]) self.assertEqual('My make My accessory', equipment_list['accessories'][0]['label']) self.assertEqual('LEGACY', equipment_list['accessories'][0]['version']) self.assertTrue('id' in equipment_list['accessories'][1]) self.assertEqual('My make My focal reducer', equipment_list['accessories'][1]['label']) self.assertEqual('LEGACY', equipment_list['accessories'][0]['version']) self.assertTrue('id' in equipment_list['software'][0]) self.assertEqual('My make My software', equipment_list['software'][0]['label']) self.assertEqual('LEGACY', equipment_list['software'][0]['version']) self.assertTrue('id' in equipment_list['guiding_telescopes'][0]) self.assertEqual('My make My guiding telescope', equipment_list['guiding_telescopes'][0]['label']) self.assertEqual('LEGACY', equipment_list['guiding_telescopes'][0]['version']) self.assertTrue('id' in equipment_list['guiding_cameras'][0]) self.assertEqual('My make My guiding camera', equipment_list['guiding_cameras'][0]['label']) self.assertEqual('LEGACY', equipment_list['guiding_cameras'][0]['version'])