first import

sites/all/modules/media/test/media.entity.test

@@ -0,0 +1,64 @@
+<?php
+
+/**
+ * @file
+ * Tests for media entity controllers.
+ */
+
+/**
+ * Test media type creation and management.
+ */
+class MediaEntityTest extends DrupalWebTestCase {
+
+  public static function getInfo() {
+    return array(
+      'name' => 'Media entity',
+      'description' => 'Tests media entity handling',
+      'group' => 'Media',
+      'dependencies' => array('ctools'),
+    );
+  }
+
+  function setUp() {
+    parent::setUp('media');
+    // Nice, TDD FTW.  #totalsarcasm
+    variable_set('simpletest_verbose', TRUE);
+  }
+
+  /**
+   * Test the ability to create and query media items.
+   */
+  function testQueryMedia() {
+    $text_files = $this->drupalGetTestFiles('text');
+    $images = $this->drupalGetTestFiles('image');
+    // Moves serveral images to public://
+    foreach (array_merge($text_files, $images) as $file) {
+      $file = file_uri_to_object($file->uri);
+      file_save($file);
+    }
+
+    $query = new EntityFieldQuery();
+    $query->entityCondition('entity_type', 'file');
+    $query->propertyCondition('uri', 'public://%', 'LIKE');
+    $result = $query->execute();
+    $this->assertEqual(count($result['file']), count($images) + count($text_files), "Returned results as expected for like % condition");
+
+    $query = new EntityFieldQuery();
+    $query->entityCondition('entity_type', 'file');
+    $query->propertyCondition('uri', '%.jpg', 'LIKE');
+    $result = $query->execute();
+    $this->assertEqual(count($result['file']), 2, "Returned 2 results as expected for jpgs");
+
+    $query = new EntityFieldQuery();
+    $query->entityCondition('entity_type', 'file');
+    $query->propertyCondition('type', 'image');
+    $result = $query->execute();
+    $this->assertEqual(count($result['file']), count($images), "Returned expected results for type query");
+
+    $query = new EntityFieldQuery();
+    $query->entityCondition('entity_type', 'file');
+    $query->propertyCondition('uri', 'http://%', 'LIKE');
+    $result = $query->execute();
+    $this->assertEqual(count($result), 0, "Got no results for http scheme uris");
+  }
+}

sites/all/modules/media/test/media.types.test

@@ -0,0 +1,84 @@
+<?php
+
+/**
+ * @file
+ * Tests for media types.
+ */
+
+/**
+ * Test media type creation and management.
+ */
+class MediaTypeTest extends DrupalWebTestCase {
+
+  public static function getInfo() {
+    return array(
+      'name' => 'Media types',
+      'description' => 'Tests media types',
+      'group' => 'Media',
+      'dependencies' => array('ctools'),
+    );
+  }
+
+  function setUp() {
+    parent::setUp('media');
+    // Nice, TDD FTW.  #totalsarcasm
+    variable_set('simpletest_verbose', TRUE);
+  }
+
+  private function createType($overrides) {
+    $type = new stdClass();
+    $type->name = 'test';
+    $type->label = "Test";
+    $type->base = TRUE;
+//    $type->view_mode_defaults = array(
+//      'media_preview' => 'styles_file_square_thumbnail',
+//      'media_original' => 'file_default',
+//    );
+
+    $type->type_callback_args =
+    array(
+      'match_type' => 'all',
+      'mimetypes' => array('/^test/'),
+      'extensions' => array('jpg', 'jpeg', 'gif', 'png', 'tiff'),
+      'streams' => array('public', 'private'),
+    );
+
+    foreach ($overrides as $k => $v) {
+      $type->$k = $v;
+    }
+
+    media_type_save($type);
+    return $type;
+  }
+
+  /**
+   * Test creating a new type. Basic CRUD.
+   */
+  function testCreate() {
+    $type_machine_name = 'foo';
+    $type = $this->createType(array('name' => $type_machine_name, 'label' => 'foobar'));
+    $loaded_type = media_type_load($type_machine_name);
+    $this->assertEqual($loaded_type->label, 'foobar', "Was able to create a type and retreive it");
+  }
+
+  /**
+   * Ensures that the weight is respected when types are created.
+   * @return unknown_type
+   */
+  function testOrder() {
+    $type = $this->createType(array('name' => 'last', 'label' => 'Last', 'weight' => 100));
+    $type = $this->createType(array('name' => 'first', 'label' => 'First'));
+    $types = media_type_get_types();
+    $keys = array_keys($types);
+    $this->assertTrue(isset($types['last']) && isset($types['first']), "Both types saved");
+    $this->assertTrue(array_search('last', $keys) > array_search('first', $keys), 'Type which was supposed to be first came first');
+  }
+
+  /**
+   * Test view mode assignment.  Currently fails, don't know why.
+   * @return unknown_type
+   */
+  function testViewModesAssigned() {
+  }
+
+}